diff --git a/session_test.go b/session_test.go new file mode 100644 index 0000000..b32934b --- /dev/null +++ b/session_test.go @@ -0,0 +1,30 @@ +package main + +import ( + "net/http" + "net/http/httptest" + "testing" +) + +func TestGetSessionCookieMissingCookie(t *testing.T) { + r, _ := http.NewRequest("GET", "/", nil) + w := httptest.NewRecorder() + sessionKey, err := getSessionCookie(w, r) + + if err != nil { + t.Fail() + } + if sessionKey == "" { + t.Errorf("Session key is empty") + } + + cookies := w.Result().Cookies() + for _, c := range cookies { + if c.Name == "session" { + if c.Value != sessionKey { + t.Errorf("Wrong sessionKey %q != %q", c.Value, sessionKey) + } + break + } + } +}