root
61 lines
1.8 KiB
Go
61 lines
1.8 KiB
Go
// Copyright 2026 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package v1_27
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gitea.dev/models/migrations/migrationtest"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_AddBranchProtectionBypassAllowlist(t *testing.T) {
|
|
type ProtectedBranch struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
RepoID int64 `xorm:"INDEX"`
|
|
BranchName string `xorm:"INDEX"`
|
|
EnableBypassAllowlist bool `xorm:"NOT NULL DEFAULT false"`
|
|
BypassAllowlistUserIDs []int64 `xorm:"JSON TEXT"`
|
|
BypassAllowlistTeamIDs []int64 `xorm:"JSON TEXT"`
|
|
}
|
|
|
|
x, deferable := migrationtest.PrepareTestEnv(t, 0, new(ProtectedBranch))
|
|
defer deferable()
|
|
|
|
// Test with default values
|
|
_, err := x.Insert(&ProtectedBranch{RepoID: 1, BranchName: "main"})
|
|
require.NoError(t, err)
|
|
|
|
// Test with populated allowlist
|
|
_, err = x.Insert(&ProtectedBranch{
|
|
RepoID: 1,
|
|
BranchName: "develop",
|
|
EnableBypassAllowlist: true,
|
|
BypassAllowlistUserIDs: []int64{1, 2, 3},
|
|
BypassAllowlistTeamIDs: []int64{10, 20},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, AddBranchProtectionBypassAllowlist(x))
|
|
|
|
// Verify the default values record
|
|
var pb ProtectedBranch
|
|
has, err := x.Where("repo_id = ? AND branch_name = ?", 1, "main").Get(&pb)
|
|
require.NoError(t, err)
|
|
require.True(t, has)
|
|
require.False(t, pb.EnableBypassAllowlist)
|
|
require.Nil(t, pb.BypassAllowlistUserIDs)
|
|
require.Nil(t, pb.BypassAllowlistTeamIDs)
|
|
|
|
// Verify the populated allowlist record
|
|
var pb2 ProtectedBranch
|
|
has, err = x.Where("repo_id = ? AND branch_name = ?", 1, "develop").Get(&pb2)
|
|
require.NoError(t, err)
|
|
require.True(t, has)
|
|
require.True(t, pb2.EnableBypassAllowlist)
|
|
require.Equal(t, []int64{1, 2, 3}, pb2.BypassAllowlistUserIDs)
|
|
require.Equal(t, []int64{10, 20}, pb2.BypassAllowlistTeamIDs)
|
|
}
|