moneygo/internal/handlers/sessions_test.go

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

	create(client, user, &u, "/session/", "user")

	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) {
	RunWith(t, &data[0], func(t *testing.T, d *TestData) {
		if err := sessionExistsOrError(d.clients[0]); err != nil {
			t.Fatal(err)
		}
	})
}

func TestGetSession(t *testing.T) {
	RunWith(t, &data[0], func(t *testing.T, d *TestData) {
		session, err := getSession(d.clients[0])
		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 != d.users[0].UserId {
			t.Errorf("session's UserId (%d) should equal user's UserID (%d)", session.UserId, d.users[0].UserId)
		}

		if session.SessionId == 0 {
			t.Error("session's SessionId should not be 0")
		}
	})
}