mirror of
				https://codeberg.org/forgejo/forgejo.git
				synced 2025-10-30 22:11:07 +00:00 
			
		
		
		
	* Refactor AssertExistsAndLoadBean to use generics * Fix tests Co-authored-by: zeripath <art27@cantab.net>
		
			
				
	
	
		
			48 lines
		
	
	
	
		
			1.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			48 lines
		
	
	
	
		
			1.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2020 The Gitea Authors. All rights reserved.
 | |
| // Use of this source code is governed by a MIT-style
 | |
| // license that can be found in the LICENSE file.
 | |
| 
 | |
| package convert
 | |
| 
 | |
| import (
 | |
| 	"testing"
 | |
| 
 | |
| 	issues_model "code.gitea.io/gitea/models/issues"
 | |
| 	"code.gitea.io/gitea/models/perm"
 | |
| 	repo_model "code.gitea.io/gitea/models/repo"
 | |
| 	"code.gitea.io/gitea/models/unittest"
 | |
| 	"code.gitea.io/gitea/modules/git"
 | |
| 	"code.gitea.io/gitea/modules/structs"
 | |
| 
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| )
 | |
| 
 | |
| func TestPullRequest_APIFormat(t *testing.T) {
 | |
| 	// with HeadRepo
 | |
| 	assert.NoError(t, unittest.PrepareTestDatabase())
 | |
| 	headRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
 | |
| 	pr := unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{ID: 1})
 | |
| 	assert.NoError(t, pr.LoadAttributes())
 | |
| 	assert.NoError(t, pr.LoadIssue())
 | |
| 	apiPullRequest := ToAPIPullRequest(git.DefaultContext, pr, nil)
 | |
| 	assert.NotNil(t, apiPullRequest)
 | |
| 	assert.EqualValues(t, &structs.PRBranchInfo{
 | |
| 		Name:       "branch1",
 | |
| 		Ref:        "refs/pull/2/head",
 | |
| 		Sha:        "4a357436d925b5c974181ff12a994538ddc5a269",
 | |
| 		RepoID:     1,
 | |
| 		Repository: ToRepo(headRepo, perm.AccessModeRead),
 | |
| 	}, apiPullRequest.Head)
 | |
| 
 | |
| 	// withOut HeadRepo
 | |
| 	pr = unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{ID: 1})
 | |
| 	assert.NoError(t, pr.LoadIssue())
 | |
| 	assert.NoError(t, pr.LoadAttributes())
 | |
| 	// simulate fork deletion
 | |
| 	pr.HeadRepo = nil
 | |
| 	pr.HeadRepoID = 100000
 | |
| 	apiPullRequest = ToAPIPullRequest(git.DefaultContext, pr, nil)
 | |
| 	assert.NotNil(t, apiPullRequest)
 | |
| 	assert.Nil(t, apiPullRequest.Head.Repository)
 | |
| 	assert.EqualValues(t, -1, apiPullRequest.Head.RepoID)
 | |
| }
 |