gitea/models/git/protected_branch_list_test.go

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

package git

import (
	"fmt"
	"testing"

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

func TestBranchRuleMatchPriority(t *testing.T) {
	kases := []struct {
		Rules            []string
		BranchName       string
		ExpectedMatchIdx int
	}{
		{
			Rules:            []string{"release/*", "release/v1.17"},
			BranchName:       "release/v1.17",
			ExpectedMatchIdx: 1,
		},
		{
			Rules:            []string{"release/v1.17", "release/*"},
			BranchName:       "release/v1.17",
			ExpectedMatchIdx: 0,
		},
		{
			Rules:            []string{"release/**/v1.17", "release/test/v1.17"},
			BranchName:       "release/test/v1.17",
			ExpectedMatchIdx: 1,
		},
		{
			Rules:            []string{"release/test/v1.17", "release/**/v1.17"},
			BranchName:       "release/test/v1.17",
			ExpectedMatchIdx: 0,
		},
		{
			Rules:            []string{"release/**", "release/v1.0.0"},
			BranchName:       "release/v1.0.0",
			ExpectedMatchIdx: 1,
		},
		{
			Rules:            []string{"release/v1.0.0", "release/**"},
			BranchName:       "release/v1.0.0",
			ExpectedMatchIdx: 0,
		},
		{
			Rules:            []string{"release/**", "release/v1.0.0"},
			BranchName:       "release/v2.0.0",
			ExpectedMatchIdx: 0,
		},
		{
			Rules:            []string{"release/*", "release/v1.0.0"},
			BranchName:       "release/1/v2.0.0",
			ExpectedMatchIdx: -1,
		},
	}

	for _, kase := range kases {
		var pbs ProtectedBranchRules
		for _, rule := range kase.Rules {
			pbs = append(pbs, &ProtectedBranch{RuleName: rule})
		}
		pbs.sort()
		matchedPB := pbs.GetFirstMatched(kase.BranchName)
		if matchedPB == nil {
			if kase.ExpectedMatchIdx >= 0 {
				assert.Error(t, fmt.Errorf("no matched rules but expected %s[%d]", kase.Rules[kase.ExpectedMatchIdx], kase.ExpectedMatchIdx))
			}
		} else {
			assert.Equal(t, kase.Rules[kase.ExpectedMatchIdx], matchedPB.RuleName)
		}
	}
}

func TestBranchRuleSort(t *testing.T) {
	for _, tc := range []struct {
		name     string
		rules    ProtectedBranchRules
		expected []string
	}{
		{"Legacy", ProtectedBranchRules{
			{RuleName: "b", CreatedUnix: 1},
			{RuleName: "b/*", CreatedUnix: 3},
			{RuleName: "a/*", CreatedUnix: 2},
			{RuleName: "c", CreatedUnix: 0},
			{RuleName: "a", CreatedUnix: 4},
		}, []string{"c", "b", "a", "a/*", "b/*"}},
		{"Priority", ProtectedBranchRules{
			{RuleName: "b", CreatedUnix: 1, Priority: 4},
			{RuleName: "b/*", CreatedUnix: 3, Priority: 2},
			{RuleName: "a/*", CreatedUnix: 2, Priority: 1},
			{RuleName: "c", CreatedUnix: 0, Priority: 0},
			{RuleName: "a", CreatedUnix: 4, Priority: 3},
		}, []string{"c", "a/*", "b/*", "a", "b"}},
	} {
		t.Run(tc.name, func(t *testing.T) {
			tc.rules.sort()
			var got []string
			for i := range tc.rules {
				got = append(got, tc.rules[i].RuleName)
			}
			assert.Equal(t, tc.expected, got)
		})
	}
}