mirror of
https://github.com/go-gitea/gitea.git
synced 2026-08-03 12:34:34 +00:00
wxiaoguang
fff32e9469
Migrations should never use model structs directly, because the model structs can be different in different releases. e.g. if one migration uses "User" model, it works in the early releases, then one day, when the User model changes, the migration breaks because it will use the new (incorrect) User model, it should only use the old User model. The same to "modules/structs". --------- Signed-off-by: wxiaoguang <wxiaoguang@gmail.com> Co-authored-by: delvh <dev.lh@web.de>
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/modelmigration/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)
|
|
}
|