mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-12-14 21:50:17 +00:00
25 lines
713 B
Go
25 lines
713 B
Go
|
// Copyright 2023 The Gitea Authors. All rights reserved.
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
package context
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"code.gitea.io/gitea/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"))
|
||
|
|
}
|