Function TestCookie

pkg/web/cookie/cookie_test.go:70–145

(t *testing.T)

}

func TestCookie(t *testing.T) {
	root := mux.NewRouter()

	a := assertions.New(t)
	blockKey := random.Bytes(32)
	hashKey := random.Bytes(64)

	root.Use(mux.MiddlewareFunc(webmiddleware.Cookies(hashKey, blockKey)))

	root.Path("/set").HandlerFunc(testSetCookie(t, "test_value")).Methods(http.MethodGet)
	root.Path("/get").HandlerFunc(testGetCookie(t, "test_value", true)).Methods(http.MethodGet)
	root.Path("/del").HandlerFunc(testDeleteCookie()).Methods(http.MethodGet)
	root.Path("/no_cookie").HandlerFunc(testGetCookie(t, "", false)).Methods(http.MethodGet)

	jar, err := cookiejar.New(&cookiejar.Options{PublicSuffixList: publicsuffix.List})
	if err != nil {
		panic(err)
	}

	doGET := func(path string) *http.Response {
		req := httptest.NewRequest(http.MethodGet, path, nil)
		req.URL.Scheme, req.URL.Host = "http", req.Host
		for _, c := range jar.Cookies(req.URL) {
			req.AddCookie(c)
		}

		rec := httptest.NewRecorder()

		root.ServeHTTP(rec, req)

		resp := rec.Result()
		resp.Request = req

		if cookies := resp.Cookies(); len(cookies) > 0 {
			jar.SetCookies(req.URL, cookies)
		}

		return resp
	}

	resp := doGET("/no_cookie")

	cookies := jar.Cookies(resp.Request.URL)
	a.So(cookies, should.BeEmpty)

	resp = doGET("/del")

	cookies = jar.Cookies(resp.Request.URL)
	a.So(cookies, should.BeEmpty)

	resp = doGET("/set")

	cookies = jar.Cookies(resp.Request.URL)
	if a.So(cookies, should.HaveLength, 1) {
		cookie := cookies[0]
		a.So(cookie.Name, should.Equal, "test_cookie")
	}