forked from rbcervilla/redisstore
-
Notifications
You must be signed in to change notification settings - Fork 0
/
redisstore_test.go
111 lines (103 loc) · 3.37 KB
/
redisstore_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package redisstore
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/go-redis/redis"
"github.com/gorilla/sessions"
)
var (
redisAddr = "ubuntu.home:6379"
redisAddrs = []string{
"ubuntu.home:7000", "ubuntu.home:7001", "ubuntu.home:7002",
"ubuntu.home:7003", "ubuntu.home:7004", "ubuntu.home:7005",
}
clustered = true
keyPrefix = "a:" // hard coded in redisstore.go
client redis.UniversalClient
)
func createStore(t *testing.T, keyPrefix string, options sessions.Options) *RedisStore {
store := NewRedisStore(client)
store.KeyPrefix(keyPrefix)
store.SetOptions(options)
return store
}
func TestSuite_Basic(t *testing.T) {
if clustered { // setup
client = redis.NewClusterClient(&redis.ClusterOptions{Addrs: redisAddrs})
} else {
client = redis.NewClient(&redis.Options{Addr: redisAddr})
}
defer client.Close() // teardown
t.Run("create store then request then session", func(t *testing.T) {
store := createStore(t, keyPrefix, sessions.Options{Path: "/", Domain: "example.com", MaxAge: 60 * 5})
request, err := http.NewRequest("GET", "http://www.example.com", nil)
if err != nil {
t.Fatal("failed to create request", err)
}
session, err := store.New(request, "hello")
if err != nil {
t.Fatal("failed to create session", err)
}
if session.IsNew == false {
t.Fatal("session is not new")
}
})
t.Run("setting options", func(t *testing.T) {
store := createStore(t, keyPrefix, sessions.Options{Path: "/", Domain: "example.com", MaxAge: 60 * 5})
opts := sessions.Options{Path: "/path", MaxAge: 99999}
store.SetOptions(opts)
request, err := http.NewRequest("GET", "http://www.example.com", nil)
if err != nil {
t.Fatal("failed to create request", err)
}
session, _ := store.New(request, "hello")
if session.Options.Path != opts.Path || session.Options.MaxAge != opts.MaxAge {
t.Fatal("failed to set options")
}
})
t.Run("saving session", func(t *testing.T) {
store := createStore(t, keyPrefix, sessions.Options{Path: "/", Domain: "example.com", MaxAge: 60 * 5})
request, err := http.NewRequest("GET", "http://www.example.com", nil)
if err != nil {
t.Fatal("failed to create request", err)
}
w := httptest.NewRecorder()
session, err := store.New(request, "hello")
if err != nil {
t.Fatal("failed to create session", err)
}
session.Values["key"] = "value"
err = session.Save(request, w)
if err != nil {
t.Fatal("failed to save: ", err)
}
})
t.Run("deleting session", func(t *testing.T) {
store := createStore(t, keyPrefix, sessions.Options{Path: "/", Domain: "example.com", MaxAge: 60 * 5})
request, err := http.NewRequest("GET", "http://www.example.com", nil)
if err != nil {
t.Fatal("failed to create request", err)
}
w := httptest.NewRecorder()
session, err := store.New(request, "hello")
if err != nil {
t.Fatal("failed to create session", err)
}
session.Values["username"] = "henry"
err = session.Save(request, w)
if err != nil {
t.Fatal("failed to save session: ", err)
}
session.Options.MaxAge = -1 // comment this to see un-delete session still exists
err = session.Save(request, w)
if err != nil {
t.Fatal("failed to delete session: ", err)
}
target := keyPrefix + session.ID
// session.Save() doesn't always reflect the key existence, using redis client to check
if client.Exists(target).Val() == 1 {
t.Fatal("delete target still exists: ", session.ID)
}
})
}