transfer.sh/vendor/github.com/golang/gddo/gddo-server/template_test.go

// Copyright 2014 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 or at
// https://developers.google.com/open-source/licenses/bsd.

package main

import (
	"net/http"
	"net/http/httptest"
	"reflect"
	"strings"
	"testing"
)

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

	expectedMessages := []flashMessage{
		{ID: "a", Args: []string{"one"}},
		{ID: "b", Args: []string{"two", "three"}},
		{ID: "c", Args: []string{}},
	}

	setFlashMessages(resp, expectedMessages)
	req := &http.Request{Header: http.Header{"Cookie": {strings.Split(resp.Header().Get("Set-Cookie"), ";")[0]}}}

	actualMessages := getFlashMessages(resp, req)
	if !reflect.DeepEqual(actualMessages, expectedMessages) {
		t.Errorf("got messages %+v, want %+v", actualMessages, expectedMessages)
	}
}