diff --git a/models/repo/update.go b/models/repo/update.go index 0222d09de5..658b01501d 100644 --- a/models/repo/update.go +++ b/models/repo/update.go @@ -105,9 +105,9 @@ func (err ErrRepoFilesAlreadyExist) Unwrap() error { return util.ErrAlreadyExist } -// CheckCreateRepository check if could created a repository +// CheckCreateRepository check if a repository can be created func CheckCreateRepository(ctx context.Context, doer, u *user_model.User, name string, overwriteOrAdopt bool) error { - if !doer.CanCreateRepo() { + if !doer.IsAdmin && !u.CanCreateRepo() { return ErrReachLimitOfRepo{u.MaxRepoCreation} } diff --git a/models/repo/update_test.go b/models/repo/update_test.go new file mode 100644 index 0000000000..88a0c675f6 --- /dev/null +++ b/models/repo/update_test.go @@ -0,0 +1,74 @@ +// Copyright 2025 The Forgejo Authors. All rights reserved. +// SPDX-License-Identifier: MIT + +package repo_test + +import ( + "testing" + + "forgejo.org/models/db" + repo_model "forgejo.org/models/repo" + "forgejo.org/models/unittest" + user_model "forgejo.org/models/user" + "forgejo.org/modules/util" + + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" +) + +func TestCheckCreateRepository(t *testing.T) { + require.NoError(t, unittest.PrepareTestDatabase()) + + t.Run("Success", func(t *testing.T) { + user := &user_model.User{MaxRepoCreation: 1} + + err := repo_model.CheckCreateRepository(db.DefaultContext, user, user, "testName", true) + + require.NoError(t, err) + }) + + t.Run("AdminIgnoresRepoLimit", func(t *testing.T) { + user := &user_model.User{MaxRepoCreation: 0, IsAdmin: true} + + err := repo_model.CheckCreateRepository(db.DefaultContext, user, user, "testName", true) + + require.NoError(t, err) + }) + + t.Run("RepoLimitReached", func(t *testing.T) { + user := &user_model.User{MaxRepoCreation: 0} + + err := repo_model.CheckCreateRepository(db.DefaultContext, user, user, "testName", true) + + require.ErrorIs(t, err, repo_model.ErrReachLimitOfRepo{}) + }) + + t.Run("UnusableRepoName", func(t *testing.T) { + user := &user_model.User{MaxRepoCreation: 1} + + err := repo_model.CheckCreateRepository(db.DefaultContext, user, user, "testName/", true) + + require.ErrorIs(t, err, db.ErrNameCharsNotAllowed{Name: "testName/"}) + }) + + t.Run("RepoAlreadyExists", func(t *testing.T) { + unittest.AssertExistsIf(t, true, &repo_model.Repository{Name: "repo1"}) + user := &user_model.User{MaxRepoCreation: 2} + + err := repo_model.CheckCreateRepository(db.DefaultContext, user, user, "repo1", true) + + require.ErrorIs(t, err, repo_model.ErrRepoAlreadyExist{Name: "repo1"}) + }) + + t.Run("RepoDirAlreadyExists", func(t *testing.T) { + db.DeleteByBean(db.DefaultContext, &repo_model.Repository{Name: "repo1"}) + user := &user_model.User{MaxRepoCreation: 2, Name: "user2"} + + exists, _ := util.IsExist(repo_model.RepoPath("user2", "repo1")) + assert.True(t, exists) + + err := repo_model.CheckCreateRepository(db.DefaultContext, user, user, "repo1", false) + + require.ErrorIs(t, err, repo_model.ErrRepoAlreadyExist{Name: "repo1", Uname: "user2"}) + }) +}