mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 00:20:25 +08:00 
			
		
		
		
	Refactor AssertExistsAndLoadBean to use generics (#20797)
* Refactor AssertExistsAndLoadBean to use generics * Fix tests Co-authored-by: zeripath <art27@cantab.net>
This commit is contained in:
		@@ -37,7 +37,7 @@ func TestGiteaUploadRepo(t *testing.T) {
 | 
			
		||||
 | 
			
		||||
	unittest.PrepareTestEnv(t)
 | 
			
		||||
 | 
			
		||||
	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}).(*user_model.User)
 | 
			
		||||
	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
 | 
			
		||||
 | 
			
		||||
	var (
 | 
			
		||||
		ctx        = context.Background()
 | 
			
		||||
@@ -63,7 +63,7 @@ func TestGiteaUploadRepo(t *testing.T) {
 | 
			
		||||
	}, nil)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
 | 
			
		||||
	repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{OwnerID: user.ID, Name: repoName}).(*repo_model.Repository)
 | 
			
		||||
	repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{OwnerID: user.ID, Name: repoName})
 | 
			
		||||
	assert.True(t, repo.HasWiki())
 | 
			
		||||
	assert.EqualValues(t, repo_model.RepositoryReady, repo.Status)
 | 
			
		||||
 | 
			
		||||
@@ -127,8 +127,8 @@ func TestGiteaUploadRepo(t *testing.T) {
 | 
			
		||||
 | 
			
		||||
func TestGiteaUploadRemapLocalUser(t *testing.T) {
 | 
			
		||||
	unittest.PrepareTestEnv(t)
 | 
			
		||||
	doer := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}).(*user_model.User)
 | 
			
		||||
	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}).(*user_model.User)
 | 
			
		||||
	doer := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
 | 
			
		||||
	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
 | 
			
		||||
 | 
			
		||||
	repoName := "migrated"
 | 
			
		||||
	uploader := NewGiteaLocalUploader(context.Background(), doer, doer.Name, repoName)
 | 
			
		||||
@@ -177,7 +177,7 @@ func TestGiteaUploadRemapLocalUser(t *testing.T) {
 | 
			
		||||
 | 
			
		||||
func TestGiteaUploadRemapExternalUser(t *testing.T) {
 | 
			
		||||
	unittest.PrepareTestEnv(t)
 | 
			
		||||
	doer := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}).(*user_model.User)
 | 
			
		||||
	doer := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
 | 
			
		||||
 | 
			
		||||
	repoName := "migrated"
 | 
			
		||||
	uploader := NewGiteaLocalUploader(context.Background(), doer, doer.Name, repoName)
 | 
			
		||||
@@ -205,7 +205,7 @@ func TestGiteaUploadRemapExternalUser(t *testing.T) {
 | 
			
		||||
	//
 | 
			
		||||
	// Link the external ID to an existing user
 | 
			
		||||
	//
 | 
			
		||||
	linkedUser := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}).(*user_model.User)
 | 
			
		||||
	linkedUser := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
 | 
			
		||||
	externalLoginUser := &user_model.ExternalLoginUser{
 | 
			
		||||
		ExternalID:    strconv.FormatInt(externalID, 10),
 | 
			
		||||
		UserID:        linkedUser.ID,
 | 
			
		||||
@@ -232,7 +232,7 @@ func TestGiteaUploadUpdateGitForPullRequest(t *testing.T) {
 | 
			
		||||
	//
 | 
			
		||||
	// fromRepo master
 | 
			
		||||
	//
 | 
			
		||||
	fromRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1}).(*repo_model.Repository)
 | 
			
		||||
	fromRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
 | 
			
		||||
	baseRef := "master"
 | 
			
		||||
	assert.NoError(t, git.InitRepository(git.DefaultContext, fromRepo.RepoPath(), false))
 | 
			
		||||
	err := git.NewCommand(git.DefaultContext, "symbolic-ref", "HEAD", git.BranchPrefix+baseRef).Run(&git.RunOpts{Dir: fromRepo.RepoPath()})
 | 
			
		||||
@@ -273,13 +273,13 @@ func TestGiteaUploadUpdateGitForPullRequest(t *testing.T) {
 | 
			
		||||
	headSHA, err := fromGitRepo.GetBranchCommitID(headRef)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
 | 
			
		||||
	fromRepoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: fromRepo.OwnerID}).(*user_model.User)
 | 
			
		||||
	fromRepoOwner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: fromRepo.OwnerID})
 | 
			
		||||
 | 
			
		||||
	//
 | 
			
		||||
	// forkRepo branch2
 | 
			
		||||
	//
 | 
			
		||||
	forkHeadRef := "branch2"
 | 
			
		||||
	forkRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 8}).(*repo_model.Repository)
 | 
			
		||||
	forkRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 8})
 | 
			
		||||
	assert.NoError(t, git.CloneWithArgs(git.DefaultContext, fromRepo.RepoPath(), forkRepo.RepoPath(), []string{}, git.CloneRepoOptions{
 | 
			
		||||
		Branch: headRef,
 | 
			
		||||
	}))
 | 
			
		||||
 
 | 
			
		||||
@@ -19,8 +19,8 @@ import (
 | 
			
		||||
func TestMigrateWhiteBlocklist(t *testing.T) {
 | 
			
		||||
	assert.NoError(t, unittest.PrepareTestDatabase())
 | 
			
		||||
 | 
			
		||||
	adminUser := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user1"}).(*user_model.User)
 | 
			
		||||
	nonAdminUser := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user2"}).(*user_model.User)
 | 
			
		||||
	adminUser := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user1"})
 | 
			
		||||
	nonAdminUser := unittest.AssertExistsAndLoadBean(t, &user_model.User{Name: "user2"})
 | 
			
		||||
 | 
			
		||||
	setting.Migrations.AllowedDomains = "github.com"
 | 
			
		||||
	setting.Migrations.AllowLocalNetworks = false
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user