authboss/expire_test.go

package authboss

import (
	"net/http"
	"net/http/httptest"
	"testing"
	"time"
)

func TestDudeIsExpired(t *testing.T) {
	Cfg = NewConfig()

	session := mockClientStore{SessionKey: "username"}
	refreshExpiry(session)
	nowTime = func() time.Time {
		return time.Now().UTC().Add(Cfg.ExpireAfter * 2)
	}
	Cfg.SessionStoreMaker = func(_ http.ResponseWriter, _ *http.Request) ClientStorer {
		return session
	}

	r, _ := http.NewRequest("GET", "tra/la/la", nil)
	w := httptest.NewRecorder()
	called := false

	m := ExpireMiddleware(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		called = true
	}))

	m.ServeHTTP(w, r)

	if !called {
		t.Error("Expected middleware to call handler")
	}

	if key, ok := session.Get(SessionKey); ok {
		t.Error("Unexpcted session key:", key)
	}

	if key, ok := session.Get(SessionLastAction); ok {
		t.Error("Unexpcted last action key:", key)
	}
}

func TestDudeIsNotExpired(t *testing.T) {
	Cfg = NewConfig()

	session := mockClientStore{SessionKey: "username"}
	refreshExpiry(session)
	nowTime = func() time.Time {
		return time.Now().UTC().Add(Cfg.ExpireAfter / 2)
	}
	Cfg.SessionStoreMaker = func(_ http.ResponseWriter, _ *http.Request) ClientStorer {
		return session
	}

	r, _ := http.NewRequest("GET", "tra/la/la", nil)
	w := httptest.NewRecorder()
	called := false

	m := ExpireMiddleware(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		called = true
	}))

	m.ServeHTTP(w, r)

	if !called {
		t.Error("Expected middleware to call handler")
	}

	if key, ok := session.Get(SessionKey); !ok {
		t.Error("Expected session key:", key)
	}
}