internal/cookie/flash_test.go

// Copyright 2020 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package cookie

import (
	"net/http"
	"net/http/httptest"
	"net/url"
	"testing"
)

const (
	testName = "testname"
	testVal  = "testvalue"
)

func TestFlashMessages(t *testing.T) {
	w := httptest.NewRecorder()

	Set(w, testName, testVal, "/foo")
	r := &http.Request{
		Header: http.Header{"Cookie": w.Header()["Set-Cookie"]},
		URL:    &url.URL{Path: "/foo"},
	}

	got, err := Extract(w, r, testName)
	if err != nil {
		t.Fatal(err)
	}
	if got != testVal {
		t.Errorf("got %q, want %q", got, testVal)
	}
}