upstream-gitea/services/context/base_test.go

// Copyright 2024 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package context

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

	"gitea.dev/modules/setting"

	"github.com/stretchr/testify/assert"
)

func TestRedirect(t *testing.T) {
	setting.IsInTesting = true
	req, _ := http.NewRequest(http.MethodGet, "/", nil)

	cases := []struct {
		url  string
		keep bool
	}{
		{"http://test", false},
		{"https://test", false},
		{"//test", false},
		{"/://test", true},
		{"/test", true},
	}
	for _, c := range cases {
		resp := httptest.NewRecorder()
		b := NewBaseContextForTest(t, resp, req)
		resp.Header().Add("Set-Cookie", (&http.Cookie{Name: setting.SessionConfig.CookieName, Value: "dummy"}).String())
		b.Redirect(c.url)
		has := resp.Header().Get("Set-Cookie") == "i_like_gitea=dummy"
		assert.Equal(t, c.keep, has, "url = %q", c.url)
	}

	req, _ = http.NewRequest(http.MethodGet, "/", nil)
	resp := httptest.NewRecorder()
	req.Header.Add("X-Gitea-Fetch-Action", "1")
	b := NewBaseContextForTest(t, resp, req)
	b.Redirect("/other")
	assert.Contains(t, resp.Header().Get("Content-Type"), "application/json")
	assert.JSONEq(t, `{"redirect":"/other"}`, resp.Body.String())
	assert.Equal(t, http.StatusOK, resp.Code)
}