97 lines
2.0 KiB
Go
97 lines
2.0 KiB
Go
package redis
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/astaxie/beego/pkg/infrastructure/session"
|
|
)
|
|
|
|
func TestRedis(t *testing.T) {
|
|
sessionConfig := &session.ManagerConfig{
|
|
CookieName: "gosessionid",
|
|
EnableSetCookie: true,
|
|
Gclifetime: 3600,
|
|
Maxlifetime: 3600,
|
|
Secure: false,
|
|
CookieLifeTime: 3600,
|
|
}
|
|
|
|
redisAddr := os.Getenv("REDIS_ADDR")
|
|
if redisAddr == "" {
|
|
redisAddr = "127.0.0.1:6379"
|
|
}
|
|
|
|
sessionConfig.ProviderConfig = fmt.Sprintf("%s,100,,0,30", redisAddr)
|
|
globalSession, err := session.NewManager("redis", sessionConfig)
|
|
if err != nil {
|
|
t.Fatal("could not create manager:", err)
|
|
}
|
|
|
|
go globalSession.GC()
|
|
|
|
r, _ := http.NewRequest("GET", "/", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
sess, err := globalSession.SessionStart(w, r)
|
|
if err != nil {
|
|
t.Fatal("session start failed:", err)
|
|
}
|
|
defer sess.SessionRelease(nil, w)
|
|
|
|
// SET AND GET
|
|
err = sess.Set(nil, "username", "astaxie")
|
|
if err != nil {
|
|
t.Fatal("set username failed:", err)
|
|
}
|
|
username := sess.Get(nil, "username")
|
|
if username != "astaxie" {
|
|
t.Fatal("get username failed")
|
|
}
|
|
|
|
// DELETE
|
|
err = sess.Delete(nil, "username")
|
|
if err != nil {
|
|
t.Fatal("delete username failed:", err)
|
|
}
|
|
username = sess.Get(nil, "username")
|
|
if username != nil {
|
|
t.Fatal("delete username failed")
|
|
}
|
|
|
|
// FLUSH
|
|
err = sess.Set(nil, "username", "astaxie")
|
|
if err != nil {
|
|
t.Fatal("set failed:", err)
|
|
}
|
|
err = sess.Set(nil, "password", "1qaz2wsx")
|
|
if err != nil {
|
|
t.Fatal("set failed:", err)
|
|
}
|
|
username = sess.Get(nil, "username")
|
|
if username != "astaxie" {
|
|
t.Fatal("get username failed")
|
|
}
|
|
password := sess.Get(nil, "password")
|
|
if password != "1qaz2wsx" {
|
|
t.Fatal("get password failed")
|
|
}
|
|
err = sess.Flush(nil)
|
|
if err != nil {
|
|
t.Fatal("flush failed:", err)
|
|
}
|
|
username = sess.Get(nil, "username")
|
|
if username != nil {
|
|
t.Fatal("flush failed")
|
|
}
|
|
password = sess.Get(nil, "password")
|
|
if password != nil {
|
|
t.Fatal("flush failed")
|
|
}
|
|
|
|
sess.SessionRelease(nil, w)
|
|
}
|