go-packages/iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-03-13 21:36:28 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'refs/remotes/origin/dev'

Browse Source
This commit is contained in:
Makis Maropoulos 2016-07-07 16:02:29 +02:00
commit 96734efedc
1 changed files with 47 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
context_test.go
View File

@ -16,6 +16,9 @@ import (
"strconv"
"strings"
"testing"
"time"
"github.com/valyala/fasthttp"
)
func TestContextReset(t *testing.T) {
@ -363,6 +366,50 @@ func TestContextUserValues(t *testing.T) {
}
func TestContextCookieSetGetRemove(t *testing.T) {
initDefault()
key := "mykey"
value := "myvalue"
Get("/set", func(ctx *Context) {
ctx.SetCookieKV(key, value) // should return non empty cookies
})
Get("/set_advanced", func(ctx *Context) {
c := fasthttp.AcquireCookie()
c.SetKey(key)
c.SetValue(value)
c.SetHTTPOnly(true)
c.SetExpire(time.Now().Add(time.Duration((60 * 60 * 24 * 7 * 4)) * time.Second))
ctx.SetCookie(c)
fasthttp.ReleaseCookie(c)
})
Get("/get", func(ctx *Context) {
ctx.Write(ctx.GetCookie(key)) // should return my value
})
Get("/remove", func(ctx *Context) {
ctx.RemoveCookie(key)
cookieFound := false
ctx.VisitAllCookies(func(k, v string) {
cookieFound = true
})
if cookieFound {
t.Fatalf("Cookie has been found, when it shouldn't!")
}
ctx.Write(ctx.GetCookie(key)) // should return ""
})
e := Tester(t)
e.GET("/set").Expect().Status(StatusOK).Cookies().NotEmpty()
e.GET("/get").Expect().Status(StatusOK).Body().Equal(value)
e.GET("/remove").Expect().Status(StatusOK).Body().Equal("")
// test again with advanced set
e.GET("/set_advanced").Expect().Status(StatusOK).Cookies().NotEmpty()
e.GET("/get").Expect().Status(StatusOK).Body().Equal(value)
e.GET("/remove").Expect().Status(StatusOK).Body().Equal("")
}
func TestContextFlashMessages(t *testing.T) {
initDefault()
firstKey := "name"