24 lines · 705 B · Go
RawDownload
// Copyright 2023 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package context import (	"net/http"	"net/http/httptest"	"testing" 	"forgejo.org/modules/setting" 	"github.com/stretchr/testify/assert") func TestRemoveSessionCookieHeader(t *testing.T) {	w := httptest.NewRecorder()	w.Header().Add("Set-Cookie", (&http.Cookie{Name: setting.SessionConfig.CookieName, Value: "foo"}).String())	w.Header().Add("Set-Cookie", (&http.Cookie{Name: "other", Value: "bar"}).String())	assert.Len(t, w.Header().Values("Set-Cookie"), 2)	removeSessionCookieHeader(w)	assert.Len(t, w.Header().Values("Set-Cookie"), 1)	assert.Contains(t, "other=bar", w.Header().Get("Set-Cookie"))}