rememberme_test.go

package rememberme

import (
	"github.com/janekolszak/rememberme/store/sqlstore"

	"github.com/stretchr/testify/assert"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"

	_ "github.com/mattn/go-sqlite3"
)

const (
	testFileName = "/tmp/idp_cookie_test.db3"
)

var (
	users = []string{"user1", "user2", "user3", "user4"}
)

func TestSetUpdateCookie(t *testing.T) {
	assert := assert.New(t)

	store, err := sqlstore.New("sqlite3", testFileName)
	assert.Nil(err)
	defer store.Close()

	c := Rememberme{
		Store:  store,
		MaxAge: time.Minute * 1,
	}

	for _, user := range users {
		w := httptest.NewRecorder()
		r, err := http.NewRequest("GET", "/", nil)
		err = c.SetCookie(w, r, user)
		assert.Nil(err)

		// Should check true if this cookie appears
		requestToVerify := &http.Request{Header: http.Header{"Cookie": w.HeaderMap["Set-Cookie"]}}
		selector, readUser, err := c.Check(requestToVerify)
		assert.Nil(err)
		assert.Equal(user, readUser)
		assert.NotEqual(selector, "")

		w2 := httptest.NewRecorder()
		err = c.UpdateCookie(w2, r, selector, user)
		assert.Nil(err)
	}
}