MokoGitea-Fork/routers/common/qos_test.go

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

package common

import (
	"testing"

	user_model "code.mokoconsulting.tech/MokoConsulting/MokoGitea/models/user"
	"code.mokoconsulting.tech/MokoConsulting/MokoGitea/modules/web/middleware"
	"code.mokoconsulting.tech/MokoConsulting/MokoGitea/services/contexttest"

	"github.com/go-chi/chi/v5"
	"github.com/stretchr/testify/assert"
)

func TestRequestPriority(t *testing.T) {
	type test struct {
		Name         string
		User         *user_model.User
		RoutePattern string
		Expected     Priority
	}

	cases := []test{
		{
			Name:     "Logged In",
			User:     &user_model.User{},
			Expected: HighPriority,
		},
		{
			Name:         "Sign In",
			RoutePattern: "/user/login",
			Expected:     DefaultPriority,
		},
		{
			Name:         "Repo Home",
			RoutePattern: "/{username}/{reponame}",
			Expected:     DefaultPriority,
		},
		{
			Name:         "User Repo",
			RoutePattern: "/{username}/{reponame}/src/branch/main",
			Expected:     LowPriority,
		},
	}

	for _, tc := range cases {
		t.Run(tc.Name, func(t *testing.T) {
			ctx, _ := contexttest.MockContext(t, "")

			if tc.User != nil {
				data := middleware.GetContextData(ctx)
				data[middleware.ContextDataKeySignedUser] = tc.User
			}

			rctx := chi.RouteContext(ctx)
			rctx.RoutePatterns = []string{tc.RoutePattern}

			assert.Exactly(t, tc.Expected, requestPriority(ctx))
		})
	}
}