2017-10-07 06:21:05 -04:00
|
|
|
package handlers_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"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
|
|
|
|
|
|
|
|
jar, err := cookiejar.New(&cookiejar.Options{PublicSuffixList: nil})
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
client := server.Client()
|
|
|
|
client.Jar = jar
|
|
|
|
|
2017-10-08 21:18:30 -04:00
|
|
|
create(client, user, &u, "/session/", "user")
|
2017-10-07 06:21:05 -04:00
|
|
|
|
|
|
|
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) {
|
2017-10-08 20:29:51 -04:00
|
|
|
RunWith(t, &data[0], func(t *testing.T, d *TestData) {
|
|
|
|
if err := sessionExistsOrError(d.clients[0]); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
})
|
2017-10-07 06:21:05 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestGetSession(t *testing.T) {
|
2017-10-08 20:29:51 -04:00
|
|
|
RunWith(t, &data[0], func(t *testing.T, d *TestData) {
|
|
|
|
session, err := getSession(d.clients[0])
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
2017-10-07 06:21:05 -04:00
|
|
|
|
2017-10-08 20:29:51 -04:00
|
|
|
if len(session.SessionSecret) != 0 {
|
|
|
|
t.Error("Session.SessionSecret should not be passed back in JSON")
|
|
|
|
}
|
2017-10-07 06:21:05 -04:00
|
|
|
|
2017-10-08 20:29:51 -04:00
|
|
|
if session.UserId != d.users[0].UserId {
|
|
|
|
t.Errorf("session's UserId (%d) should equal user's UserID (%d)", session.UserId, d.users[0].UserId)
|
|
|
|
}
|
2017-10-07 06:21:05 -04:00
|
|
|
|
2017-10-08 20:29:51 -04:00
|
|
|
if session.SessionId == 0 {
|
|
|
|
t.Error("session's SessionId should not be 0")
|
|
|
|
}
|
|
|
|
})
|
2017-10-07 06:21:05 -04:00
|
|
|
}
|