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
|
package revel
import (
"net/http"
"testing"
"time"
)
func TestSessionRestore(t *testing.T) {
expireAfterDuration = 0
originSession := make(Session)
originSession["foo"] = "foo"
originSession["bar"] = "bar"
cookie := originSession.Cookie()
if !cookie.Expires.IsZero() {
t.Error("incorrect cookie expire", cookie.Expires)
}
restoredSession := GetSessionFromCookie(cookie)
for k, v := range originSession {
if restoredSession[k] != v {
t.Errorf("session restore failed session[%s] != %s", k, v)
}
}
}
func TestSessionExpire(t *testing.T) {
expireAfterDuration = time.Hour
session := make(Session)
session["user"] = "Tom"
var cookie *http.Cookie
for i := 0; i < 3; i++ {
cookie = session.Cookie()
time.Sleep(time.Second)
session = GetSessionFromCookie(cookie)
}
expectExpire := time.Now().Add(expireAfterDuration)
if cookie.Expires.Unix() < expectExpire.Add(-time.Second).Unix() {
t.Error("expect expires", cookie.Expires, "after", expectExpire.Add(-time.Second))
}
if cookie.Expires.Unix() > expectExpire.Unix() {
t.Error("expect expires", cookie.Expires, "before", expectExpire)
}
session.SetNoExpiration()
for i := 0; i < 3; i++ {
cookie = session.Cookie()
session = GetSessionFromCookie(cookie)
}
cookie = session.Cookie()
if !cookie.Expires.IsZero() {
t.Error("expect cookie expires is zero")
}
session.SetDefaultExpiration()
cookie = session.Cookie()
expectExpire = time.Now().Add(expireAfterDuration)
if cookie.Expires.Unix() < expectExpire.Add(-time.Second).Unix() {
t.Error("expect expires", cookie.Expires, "after", expectExpire.Add(-time.Second))
}
if cookie.Expires.Unix() > expectExpire.Unix() {
t.Error("expect expires", cookie.Expires, "before", expectExpire)
}
}
|