[refactor] Deprecate Middleware

2025-07-15 01:04:36 +02:00 · 2017-08-31 17:20:58 +02:00
parent bd7dc7a5c9
commit 688d5d4cd0
17 changed files with 1028 additions and 2070 deletions
--- a/session_test.go
+++ b/session_test.go
@ -0,0 +1,195 @@
+package scs
+
+import (
+	"encoding/gob"
+	"fmt"
+	"net/http"
+	"regexp"
+	"strings"
+	"testing"
+)
+
+type testUser struct {
+	Name string
+	Age  int
+}
+
+func init() {
+	gob.Register(new(testUser))
+}
+
+func TestGenerateToken(t *testing.T) {
+	id, err := generateToken()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	match, err := regexp.MatchString("^[0-9a-zA-Z_\\-]{43}$", id)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if match == false {
+		t.Errorf("got %q: should match %q", id, "^[0-9a-zA-Z_\\-]{43}$")
+	}
+}
+
+func TestString(t *testing.T) {
+	manager := NewManager(newMockStore())
+
+	_, body, cookie := testRequest(t, testPutString(manager), "")
+	if body != "OK" {
+		t.Fatalf("got %q: expected %q", body, "OK")
+	}
+
+	_, body, _ = testRequest(t, testGetString(manager), cookie)
+	if body != "lorem ipsum" {
+		t.Fatalf("got %q: expected %q", body, "lorem ipsum")
+	}
+
+	_, body, cookie = testRequest(t, testPopString(manager), cookie)
+	if body != "lorem ipsum" {
+		t.Fatalf("got %q: expected %q", body, "lorem ipsum")
+	}
+
+	_, body, _ = testRequest(t, testGetString(manager), cookie)
+	if body != "" {
+		t.Fatalf("got %q: expected %q", body, "")
+	}
+}
+
+func TestObject(t *testing.T) {
+	manager := NewManager(newMockStore())
+
+	_, body, cookie := testRequest(t, testPutObject(manager), "")
+	if body != "OK" {
+		t.Fatalf("got %q: expected %q", body, "OK")
+	}
+
+	_, body, _ = testRequest(t, testGetObject(manager), cookie)
+	if body != "alice: 21" {
+		t.Fatalf("got %q: expected %q", body, "alice: 21")
+	}
+
+	_, body, cookie = testRequest(t, testPopObject(manager), cookie)
+	if body != "alice: 21" {
+		t.Fatalf("got %q: expected %q", body, "alice: 21")
+	}
+
+	_, body, _ = testRequest(t, testGetObject(manager), cookie)
+	if body != ": 0" {
+		t.Fatalf("got %q: expected %q", body, ": 0")
+	}
+}
+
+func TestDestroy(t *testing.T) {
+	store := newMockStore()
+	manager := NewManager(store)
+
+	_, _, cookie := testRequest(t, testPutString(manager), "")
+	oldToken := extractTokenFromCookie(cookie)
+
+	_, body, cookie := testRequest(t, testDestroy(manager), cookie)
+
+	if body != "OK" {
+		t.Fatalf("got %q: expected %q", body, "OK")
+	}
+	if strings.HasPrefix(cookie, fmt.Sprintf("%s=;", CookieName)) == false {
+		t.Fatalf("got %q: expected prefix %q", cookie, fmt.Sprintf("%s=;", CookieName))
+	}
+	if strings.Contains(cookie, "Expires=Thu, 01 Jan 1970 00:00:01 GMT") == false {
+		t.Fatalf("got %q: expected to contain %q", cookie, "Expires=Thu, 01 Jan 1970 00:00:01 GMT")
+	}
+	if strings.Contains(cookie, "Max-Age=0") == false {
+		t.Fatalf("got %q: expected to contain %q", cookie, "Max-Age=0")
+	}
+	_, found, _ := store.Find(oldToken)
+	if found != false {
+		t.Fatalf("got %v: expected %v", found, false)
+	}
+}
+
+func TestRenewToken(t *testing.T) {
+	store := newMockStore()
+	manager := NewManager(store)
+
+	_, _, cookie := testRequest(t, testPutString(manager), "")
+	oldToken := extractTokenFromCookie(cookie)
+
+	_, body, cookie := testRequest(t, testRenewToken(manager), cookie)
+	if body != "OK" {
+		t.Fatalf("got %q: expected %q", body, "OK")
+	}
+	newToken := extractTokenFromCookie(cookie)
+	if newToken == oldToken {
+		t.Fatal("expected a difference")
+	}
+	_, found, _ := store.Find(oldToken)
+	if found != false {
+		t.Fatalf("got %v: expected %v", found, false)
+	}
+
+	_, body, _ = testRequest(t, testGetString(manager), cookie)
+	if body != "lorem ipsum" {
+		t.Fatalf("got %q: expected %q", body, "lorem ipsum")
+	}
+}
+
+func TestClear(t *testing.T) {
+	manager := NewManager(newMockStore())
+
+	_, _, cookie := testRequest(t, testPutString(manager), "")
+	_, _, cookie = testRequest(t, testPutBool(manager), cookie)
+
+	_, body, cookie := testRequest(t, testClear(manager), cookie)
+	if body != "OK" {
+		t.Fatalf("got %q: expected %q", body, "OK")
+	}
+
+	_, body, _ = testRequest(t, testGetString(manager), cookie)
+	if body != "" {
+		t.Fatalf("got %q: expected %q", body, "")
+	}
+
+	_, body, _ = testRequest(t, testGetBool(manager), cookie)
+	if body != "false" {
+		t.Fatalf("got %q: expected %q", body, "false")
+	}
+
+	// Check that it's a no-op if there is no data in the session
+	_, _, cookie = testRequest(t, testClear(manager), cookie)
+	if cookie != "" {
+		t.Fatalf("got %q: expected %q", cookie, "")
+	}
+}
+
+func TestKeys(t *testing.T) {
+	manager := NewManager(newMockStore())
+
+	_, _, cookie := testRequest(t, testPutString(manager), "")
+	_, _, _ = testRequest(t, testPutBool(manager), cookie)
+
+	_, body, _ := testRequest(t, testKeys(manager), cookie)
+	if body != "[test_bool test_string]" {
+		t.Fatalf("got %q: expected %q", body, "[test_bool test_string]")
+	}
+
+	_, _, _ = testRequest(t, testClear(manager), cookie)
+	_, body, _ = testRequest(t, testKeys(manager), cookie)
+	if body != "[]" {
+		t.Fatalf("got %q: expected %q", body, "[]")
+	}
+}
+
+func TestLoadFailure(t *testing.T) {
+	manager := NewManager(newMockStore())
+
+	cookie := http.Cookie{
+		Name:  "session",
+		Value: "force-error",
+	}
+
+	_, body, _ := testRequest(t, testPutString(manager), cookie.String())
+	if body != "forced-error\n" {
+		t.Fatalf("got %q: expected %q", body, "forced-error\n")
+	}
+}