mirror of
https://github.com/aclindsa/moneygo.git
synced 2025-07-03 04:38:38 -04:00
Add tests for users and sessions
Split out common test infrastructure from security_templates_test, make tests HTTPS, use the http.Client provided by httptest
This commit is contained in:
144
internal/handlers/sessions_test.go
Normal file
144
internal/handlers/sessions_test.go
Normal file
@ -0,0 +1,144 @@
|
||||
package handlers_test
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/aclindsa/moneygo/internal/handlers"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/http/cookiejar"
|
||||
"net/url"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func newSession(user *User) (*http.Client, error) {
|
||||
var u User
|
||||
var e handlers.Error
|
||||
|
||||
jar, err := cookiejar.New(&cookiejar.Options{PublicSuffixList: nil})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
client := server.Client()
|
||||
client.Jar = jar
|
||||
|
||||
bytes, err := json.Marshal(user)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
response, err := client.PostForm(server.URL+"/session/", url.Values{"user": {string(bytes)}})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
body, err := ioutil.ReadAll(response.Body)
|
||||
response.Body.Close()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = (&u).Read(string(body))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = (&e).Read(string(body))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if e.ErrorId != 0 || len(e.ErrorString) != 0 {
|
||||
return nil, fmt.Errorf("Unexpected error when creating session %+v", e)
|
||||
}
|
||||
|
||||
return client, nil
|
||||
}
|
||||
|
||||
func getSession(client *http.Client) (*handlers.Session, error) {
|
||||
var s handlers.Session
|
||||
response, err := client.Get(server.URL + "/session/")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
body, err := ioutil.ReadAll(response.Body)
|
||||
response.Body.Close()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = (&s).Read(string(body))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &s, nil
|
||||
}
|
||||
|
||||
func sessionExistsOrError(c *http.Client) error {
|
||||
|
||||
url, err := url.Parse(server.URL)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cookies := c.Jar.Cookies(url)
|
||||
|
||||
var found_session bool = false
|
||||
for _, cookie := range cookies {
|
||||
if cookie.Name == "moneygo-session" {
|
||||
found_session = true
|
||||
}
|
||||
}
|
||||
if found_session {
|
||||
return nil
|
||||
}
|
||||
return fmt.Errorf("Didn't find 'moneygo-session' cookie in CookieJar")
|
||||
}
|
||||
|
||||
func TestCreateSession(t *testing.T) {
|
||||
u, err := createUser(&users[0])
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
u.Password = users[0].Password
|
||||
|
||||
client, err := newSession(u)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer deleteUser(client, u)
|
||||
if err := sessionExistsOrError(client); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetSession(t *testing.T) {
|
||||
u, err := createUser(&users[0])
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
u.Password = users[0].Password
|
||||
|
||||
client, err := newSession(u)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer deleteUser(client, u)
|
||||
session, err := getSession(client)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if len(session.SessionSecret) != 0 {
|
||||
t.Error("Session.SessionSecret should not be passed back in JSON")
|
||||
}
|
||||
|
||||
if session.UserId != u.UserId {
|
||||
t.Errorf("session's UserId (%d) should equal user's UserID (%d)", session.UserId, u.UserId)
|
||||
}
|
||||
|
||||
if session.SessionId == 0 {
|
||||
t.Error("session's SessionId should not be 0")
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user