Unit tests for models/pull.go

2025-11-04 16:40:24 +08:00 · 2017-01-01 13:15:09 -05:00
parent 1a7fc53c98
commit 72bfabfada
8 changed files with 421 additions and 21 deletions
--- a/models/pull_test.go
+++ b/models/pull_test.go
@@ -0,0 +1,251 @@
+// Copyright 2017 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 models
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+// getPullRequest load a fixture pull request from the test database
+func loadFixturePullRequest(t *testing.T, id int64) *PullRequest {
+	sess := x.NewSession()
+	defer sess.Close()
+
+	pr := &PullRequest{ID: id}
+	has, err := sess.Get(pr)
+	assert.NoError(t, err)
+	assert.True(t, has)
+	return pr
+}
+
+func TestPullRequest_LoadAttributes(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr := loadFixturePullRequest(t, 1)
+	assert.NoError(t, pr.LoadAttributes())
+	assert.NotNil(t, pr.Merger)
+	assert.Equal(t, pr.MergerID, pr.Merger.ID)
+}
+
+func TestPullRequest_LoadIssue(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr := loadFixturePullRequest(t, 1)
+	assert.NoError(t, pr.LoadIssue())
+	assert.NotNil(t, pr.Issue)
+	assert.Equal(t, int64(2), pr.Issue.ID)
+}
+
+// TODO TestPullRequest_APIFormat
+
+func TestPullRequest_GetBaseRepo(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr := loadFixturePullRequest(t, 1)
+	assert.NoError(t, pr.GetBaseRepo())
+	assert.NotNil(t, pr.BaseRepo)
+	assert.Equal(t, pr.BaseRepoID, pr.BaseRepo.ID)
+}
+
+func TestPullRequest_GetHeadRepo(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr := loadFixturePullRequest(t, 1)
+	assert.NoError(t, pr.GetHeadRepo())
+	assert.NotNil(t, pr.HeadRepo)
+	assert.Equal(t, pr.HeadRepoID, pr.HeadRepo.ID)
+}
+
+// TODO TestMerge
+
+// TODO TestNewPullRequest
+
+func TestPullRequestsNewest(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	prs, count, err := PullRequests(1, &PullRequestsOptions{
+		Page:     1,
+		State:    "open",
+		SortType: "newest",
+		Labels:   []string{},
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, int64(2), count)
+	assert.Len(t, prs, 2)
+	assert.Equal(t, int64(2), prs[0].ID)
+	assert.Equal(t, int64(1), prs[1].ID)
+}
+
+func TestPullRequestsOldest(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	prs, count, err := PullRequests(1, &PullRequestsOptions{
+		Page:     1,
+		State:    "open",
+		SortType: "oldest",
+		Labels:   []string{},
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, int64(2), count)
+	assert.Len(t, prs, 2)
+	assert.Equal(t, int64(1), prs[0].ID)
+	assert.Equal(t, int64(2), prs[1].ID)
+}
+
+func TestGetUnmergedPullRequest(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr, err := GetUnmergedPullRequest(1, 1, "branch2", "master")
+	assert.NoError(t, err)
+	assert.Equal(t, int64(2), pr.ID)
+
+	pr, err = GetUnmergedPullRequest(1, 9223372036854775807, "branch1", "master")
+	assert.Error(t, err)
+	assert.True(t, IsErrPullRequestNotExist(err))
+}
+
+func TestGetUnmergedPullRequestsByHeadInfo(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	prs, err := GetUnmergedPullRequestsByHeadInfo(1, "branch2")
+	assert.NoError(t, err)
+	assert.Len(t, prs, 1)
+	for _, pr := range prs {
+		assert.Equal(t, int64(1), pr.HeadRepoID)
+		assert.Equal(t, "branch2", pr.HeadBranch)
+	}
+}
+
+func TestGetUnmergedPullRequestsByBaseInfo(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	prs, err := GetUnmergedPullRequestsByBaseInfo(1, "master")
+	assert.NoError(t, err)
+	assert.Len(t, prs, 1)
+	pr := prs[0]
+	assert.Equal(t, int64(2), pr.ID)
+	assert.Equal(t, int64(1), pr.BaseRepoID)
+	assert.Equal(t, "master", pr.BaseBranch)
+}
+
+func TestGetPullRequestByIndex(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr, err := GetPullRequestByIndex(1, 2)
+	assert.NoError(t, err)
+	assert.Equal(t, int64(1), pr.BaseRepoID)
+	assert.Equal(t, int64(2), pr.Index)
+
+	pr, err = GetPullRequestByIndex(9223372036854775807, 9223372036854775807)
+	assert.Error(t, err)
+	assert.True(t, IsErrPullRequestNotExist(err))
+}
+
+func TestGetPullRequestByID(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr, err := GetPullRequestByID(1)
+	assert.NoError(t, err)
+	assert.Equal(t, int64(1), pr.ID)
+	assert.Equal(t, int64(2), pr.IssueID)
+
+	_, err = GetPullRequestByID(9223372036854775807)
+	assert.Error(t, err)
+	assert.True(t, IsErrPullRequestNotExist(err))
+}
+
+func TestGetPullRequestByIssueID(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr, err := GetPullRequestByIssueID(2)
+	assert.NoError(t, err)
+	assert.Equal(t, int64(2), pr.IssueID)
+
+	pr, err = GetPullRequestByIssueID(9223372036854775807)
+	assert.Error(t, err)
+	assert.True(t, IsErrPullRequestNotExist(err))
+}
+
+func TestPullRequest_Update(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr := &PullRequest{
+		ID:         1,
+		IssueID:    100,
+		BaseBranch: "baseBranch",
+		HeadBranch: "headBranch",
+	}
+	pr.Update()
+
+	sess := x.NewSession()
+	defer sess.Close()
+	pr = &PullRequest{ID: 1}
+	has, err := sess.Get(pr)
+	assert.NoError(t, err)
+	assert.True(t, has)
+	assert.Equal(t, int64(100), pr.IssueID)
+	assert.Equal(t, "baseBranch", pr.BaseBranch)
+	assert.Equal(t, "headBranch", pr.HeadBranch)
+}
+
+func TestPullRequest_UpdateCols(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	pr := &PullRequest{
+		ID:         1,
+		IssueID:    int64(100),
+		BaseBranch: "baseBranch",
+		HeadBranch: "headBranch",
+	}
+	pr.UpdateCols("issue_id", "head_branch")
+
+	sess := x.NewSession()
+	defer sess.Close()
+	pr = &PullRequest{ID: 1}
+	has, err := sess.Get(pr)
+	assert.NoError(t, err)
+	assert.True(t, has)
+	assert.Equal(t, int64(100), pr.IssueID)
+	assert.Equal(t, "master", pr.BaseBranch)
+	assert.Equal(t, "headBranch", pr.HeadBranch)
+}
+
+// TODO TestPullRequest_UpdatePatch
+
+// TODO TestPullRequest_PushToBaseRepo
+
+func TestPullRequest_AddToTaskQueue(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+
+	pr := loadFixturePullRequest(t, 1)
+	pr.AddToTaskQueue()
+
+	// briefly sleep so that background threads have time to run
+	time.Sleep(time.Millisecond)
+
+	assert.True(t, pullRequestQueue.Exist(pr.ID))
+	pr = loadFixturePullRequest(t, 1)
+	assert.Equal(t, PullRequestStatusChecking, pr.Status)
+}
+
+func TestPullRequestList_LoadAttributes(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+
+	prs := []*PullRequest{
+		loadFixturePullRequest(t, 1),
+		loadFixturePullRequest(t, 2),
+	}
+	assert.NoError(t, PullRequestList(prs).LoadAttributes())
+	for _, pr := range prs {
+		assert.NotNil(t, pr.Issue)
+		assert.Equal(t, pr.IssueID, pr.Issue.ID)
+	}
+}
+
+// TODO TestAddTestPullRequestTask
+
+func TestChangeUsernameInPullRequests(t *testing.T) {
+	assert.NoError(t, PrepareTestDatabase())
+	const newUsername = "newusername"
+	assert.NoError(t, ChangeUsernameInPullRequests("user1", newUsername))
+	sess := x.NewSession()
+	defer sess.Close()
+
+	prs := make([]*PullRequest, 0, 10)
+	assert.NoError(t, sess.Where("head_user_name = ?", newUsername).Find(&prs))
+	assert.Len(t, prs, 2)
+	for _, pr := range prs {
+		assert.Equal(t, newUsername, pr.HeadUserName)
+	}
+}