85 lines
2.4 KiB
Go
85 lines
2.4 KiB
Go
package auth
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
const testSecret = "test-secret-at-least-32-bytes-long!"
|
|
|
|
func TestMintAndParseSessionCookie(t *testing.T) {
|
|
claims := Claims{Sub: "u1", Name: "Alice", Email: "alice@example.com"}
|
|
cookie, err := MintSessionCookie(claims, []byte(testSecret), time.Hour, true)
|
|
if err != nil {
|
|
t.Fatalf("mint: %v", err)
|
|
}
|
|
if cookie.Name != SessionCookieName || !cookie.HttpOnly || !cookie.Secure {
|
|
t.Fatalf("cookie = %+v, want name=%s HttpOnly+Secure", cookie, SessionCookieName)
|
|
}
|
|
|
|
got, err := ParseSessionCookie(cookie, []byte(testSecret))
|
|
if err != nil {
|
|
t.Fatalf("parse: %v", err)
|
|
}
|
|
if got != claims {
|
|
t.Fatalf("got %+v, want %+v", got, claims)
|
|
}
|
|
}
|
|
|
|
func TestParseSessionCookie_Expired(t *testing.T) {
|
|
cookie, err := MintSessionCookie(Claims{Sub: "u1"}, []byte(testSecret), -time.Hour, false)
|
|
if err != nil {
|
|
t.Fatalf("mint: %v", err)
|
|
}
|
|
if _, err := ParseSessionCookie(cookie, []byte(testSecret)); err == nil {
|
|
t.Fatal("expected error for expired cookie")
|
|
}
|
|
}
|
|
|
|
func TestParseSessionCookie_Tampered(t *testing.T) {
|
|
cookie, err := MintSessionCookie(Claims{Sub: "u1"}, []byte(testSecret), time.Hour, false)
|
|
if err != nil {
|
|
t.Fatalf("mint: %v", err)
|
|
}
|
|
cookie.Value = cookie.Value[:len(cookie.Value)-1] + "x"
|
|
if _, err := ParseSessionCookie(cookie, []byte(testSecret)); err == nil {
|
|
t.Fatal("expected error for tampered cookie")
|
|
}
|
|
}
|
|
|
|
func TestParseSessionCookie_WrongSecret(t *testing.T) {
|
|
cookie, err := MintSessionCookie(Claims{Sub: "u1"}, []byte(testSecret), time.Hour, false)
|
|
if err != nil {
|
|
t.Fatalf("mint: %v", err)
|
|
}
|
|
if _, err := ParseSessionCookie(cookie, []byte("a-completely-different-secret!!")); err == nil {
|
|
t.Fatal("expected error for wrong secret")
|
|
}
|
|
}
|
|
|
|
func TestMintAndParseTxnCookie(t *testing.T) {
|
|
txn := TxnState{State: "abc123", CodeVerifier: "verifier-xyz"}
|
|
cookie, err := MintTxnCookie(txn, []byte(testSecret), false)
|
|
if err != nil {
|
|
t.Fatalf("mint: %v", err)
|
|
}
|
|
if cookie.Name != TxnCookieName {
|
|
t.Fatalf("cookie name = %q, want %q", cookie.Name, TxnCookieName)
|
|
}
|
|
|
|
got, err := ParseTxnCookie(cookie, []byte(testSecret))
|
|
if err != nil {
|
|
t.Fatalf("parse: %v", err)
|
|
}
|
|
if got != txn {
|
|
t.Fatalf("got %+v, want %+v", got, txn)
|
|
}
|
|
}
|
|
|
|
func TestClearCookie(t *testing.T) {
|
|
c := ClearCookie(SessionCookieName, true)
|
|
if c.Value != "" || c.MaxAge >= 0 {
|
|
t.Fatalf("ClearCookie = %+v, want empty value and negative MaxAge", c)
|
|
}
|
|
}
|