Fix automerge will not work because of some events haven't been triggered (#30780)
Replace #25741
Close #24445
Close #30658
Close #20646
~Depends on #30805~
Since #25741 has been rewritten totally, to make the contribution
easier, I will continue the work in this PR. Thanks @6543
---------
Co-authored-by: 6543 <6543@obermui.de>
Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
(cherry picked from commit c6cf96d31d80ab79d370a6192fd761b4443daec2)
Conflicts:
	tests/integration/editor_test.go
	trivial context conflict because of 75ce1e2ac1 [GITEA] Allow user to select email for file operations in Web UI
	tests/integration/pull_merge_test.go
	trivial context conflicts in imports because more tests were added in Forgejo
			
			
This commit is contained in:
		
							parent
							
								
									0e0ab349fb
								
							
						
					
					
						commit
						7d7ea45465
					
				
					 7 changed files with 347 additions and 67 deletions
				
			
		| 
						 | 
				
			
			@ -155,14 +155,14 @@ func (r *Review) LoadCodeComments(ctx context.Context) (err error) {
 | 
			
		|||
	if r.CodeComments != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	if err = r.loadIssue(ctx); err != nil {
 | 
			
		||||
	if err = r.LoadIssue(ctx); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	r.CodeComments, err = fetchCodeCommentsByReview(ctx, r.Issue, nil, r, false)
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *Review) loadIssue(ctx context.Context) (err error) {
 | 
			
		||||
func (r *Review) LoadIssue(ctx context.Context) (err error) {
 | 
			
		||||
	if r.Issue != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
| 
						 | 
				
			
			@ -199,7 +199,7 @@ func (r *Review) LoadReviewerTeam(ctx context.Context) (err error) {
 | 
			
		|||
 | 
			
		||||
// LoadAttributes loads all attributes except CodeComments
 | 
			
		||||
func (r *Review) LoadAttributes(ctx context.Context) (err error) {
 | 
			
		||||
	if err = r.loadIssue(ctx); err != nil {
 | 
			
		||||
	if err = r.LoadIssue(ctx); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	if err = r.LoadCodeComments(ctx); err != nil {
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -22,6 +22,7 @@ import (
 | 
			
		|||
	"code.gitea.io/gitea/modules/log"
 | 
			
		||||
	"code.gitea.io/gitea/modules/process"
 | 
			
		||||
	"code.gitea.io/gitea/modules/queue"
 | 
			
		||||
	notify_service "code.gitea.io/gitea/services/notify"
 | 
			
		||||
	pull_service "code.gitea.io/gitea/services/pull"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -30,6 +31,8 @@ var prAutoMergeQueue *queue.WorkerPoolQueue[string]
 | 
			
		|||
 | 
			
		||||
// Init runs the task queue to that handles auto merges
 | 
			
		||||
func Init() error {
 | 
			
		||||
	notify_service.RegisterNotifier(NewNotifier())
 | 
			
		||||
 | 
			
		||||
	prAutoMergeQueue = queue.CreateUniqueQueue(graceful.GetManager().ShutdownContext(), "pr_auto_merge", handler)
 | 
			
		||||
	if prAutoMergeQueue == nil {
 | 
			
		||||
		return fmt.Errorf("unable to create pr_auto_merge queue")
 | 
			
		||||
| 
						 | 
				
			
			@ -47,7 +50,7 @@ func handler(items ...string) []string {
 | 
			
		|||
			log.Error("could not parse data from pr_auto_merge queue (%v): %v", s, err)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		handlePull(id, sha)
 | 
			
		||||
		handlePullRequestAutoMerge(id, sha)
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -62,16 +65,6 @@ func addToQueue(pr *issues_model.PullRequest, sha string) {
 | 
			
		|||
// ScheduleAutoMerge if schedule is false and no error, pull can be merged directly
 | 
			
		||||
func ScheduleAutoMerge(ctx context.Context, doer *user_model.User, pull *issues_model.PullRequest, style repo_model.MergeStyle, message string) (scheduled bool, err error) {
 | 
			
		||||
	err = db.WithTx(ctx, func(ctx context.Context) error {
 | 
			
		||||
		lastCommitStatus, err := pull_service.GetPullRequestCommitStatusState(ctx, pull)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// we don't need to schedule
 | 
			
		||||
		if lastCommitStatus.IsSuccess() {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if err := pull_model.ScheduleAutoMerge(ctx, doer, pull.ID, style, message); err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
| 
						 | 
				
			
			@ -95,8 +88,8 @@ func RemoveScheduledAutoMerge(ctx context.Context, doer *user_model.User, pull *
 | 
			
		|||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// MergeScheduledPullRequest merges a previously scheduled pull request when all checks succeeded
 | 
			
		||||
func MergeScheduledPullRequest(ctx context.Context, sha string, repo *repo_model.Repository) error {
 | 
			
		||||
// StartPRCheckAndAutoMergeBySHA start an automerge check and auto merge task for all pull requests of repository and SHA
 | 
			
		||||
func StartPRCheckAndAutoMergeBySHA(ctx context.Context, sha string, repo *repo_model.Repository) error {
 | 
			
		||||
	pulls, err := getPullRequestsByHeadSHA(ctx, sha, repo, func(pr *issues_model.PullRequest) bool {
 | 
			
		||||
		return !pr.HasMerged && pr.CanAutoMerge()
 | 
			
		||||
	})
 | 
			
		||||
| 
						 | 
				
			
			@ -111,6 +104,32 @@ func MergeScheduledPullRequest(ctx context.Context, sha string, repo *repo_model
 | 
			
		|||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// StartPRCheckAndAutoMerge start an automerge check and auto merge task for a pull request
 | 
			
		||||
func StartPRCheckAndAutoMerge(ctx context.Context, pull *issues_model.PullRequest) {
 | 
			
		||||
	if pull == nil || pull.HasMerged || !pull.CanAutoMerge() {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if err := pull.LoadBaseRepo(ctx); err != nil {
 | 
			
		||||
		log.Error("LoadBaseRepo: %v", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	gitRepo, err := gitrepo.OpenRepository(ctx, pull.BaseRepo)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error("OpenRepository: %v", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	defer gitRepo.Close()
 | 
			
		||||
	commitID, err := gitRepo.GetRefCommitID(pull.GetGitRefName())
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error("GetRefCommitID: %v", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	addToQueue(pull, commitID)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getPullRequestsByHeadSHA(ctx context.Context, sha string, repo *repo_model.Repository, filter func(*issues_model.PullRequest) bool) (map[int64]*issues_model.PullRequest, error) {
 | 
			
		||||
	gitRepo, err := gitrepo.OpenRepository(ctx, repo)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
| 
						 | 
				
			
			@ -161,7 +180,8 @@ func getPullRequestsByHeadSHA(ctx context.Context, sha string, repo *repo_model.
 | 
			
		|||
	return pulls, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func handlePull(pullID int64, sha string) {
 | 
			
		||||
// handlePullRequestAutoMerge merge the pull request if all checks are successful
 | 
			
		||||
func handlePullRequestAutoMerge(pullID int64, sha string) {
 | 
			
		||||
	ctx, _, finished := process.GetManager().AddContext(graceful.GetManager().HammerContext(),
 | 
			
		||||
		fmt.Sprintf("Handle AutoMerge of PR[%d] with sha[%s]", pullID, sha))
 | 
			
		||||
	defer finished()
 | 
			
		||||
| 
						 | 
				
			
			@ -182,24 +202,50 @@ func handlePull(pullID int64, sha string) {
 | 
			
		|||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if err = pr.LoadBaseRepo(ctx); err != nil {
 | 
			
		||||
		log.Error("%-v LoadBaseRepo: %v", pr, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// check the sha is the same as pull request head commit id
 | 
			
		||||
	baseGitRepo, err := gitrepo.OpenRepository(ctx, pr.BaseRepo)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error("OpenRepository: %v", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	defer baseGitRepo.Close()
 | 
			
		||||
 | 
			
		||||
	headCommitID, err := baseGitRepo.GetRefCommitID(pr.GetGitRefName())
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error("GetRefCommitID: %v", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if headCommitID != sha {
 | 
			
		||||
		log.Warn("Head commit id of auto merge %-v does not match sha [%s], it may means the head branch has been updated. Just ignore this request because a new request expected in the queue", pr, sha)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Get all checks for this pr
 | 
			
		||||
	// We get the latest sha commit hash again to handle the case where the check of a previous push
 | 
			
		||||
	// did not succeed or was not finished yet.
 | 
			
		||||
 | 
			
		||||
	if err = pr.LoadHeadRepo(ctx); err != nil {
 | 
			
		||||
		log.Error("%-v LoadHeadRepo: %v", pr, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	headGitRepo, err := gitrepo.OpenRepository(ctx, pr.HeadRepo)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error("OpenRepository %-v: %v", pr.HeadRepo, err)
 | 
			
		||||
		return
 | 
			
		||||
	var headGitRepo *git.Repository
 | 
			
		||||
	if pr.BaseRepoID == pr.HeadRepoID {
 | 
			
		||||
		headGitRepo = baseGitRepo
 | 
			
		||||
	} else {
 | 
			
		||||
		headGitRepo, err = gitrepo.OpenRepository(ctx, pr.HeadRepo)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Error("OpenRepository %-v: %v", pr.HeadRepo, err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		defer headGitRepo.Close()
 | 
			
		||||
	}
 | 
			
		||||
	defer headGitRepo.Close()
 | 
			
		||||
 | 
			
		||||
	headBranchExist := headGitRepo.IsBranchExist(pr.HeadBranch)
 | 
			
		||||
 | 
			
		||||
	if pr.HeadRepo == nil || !headBranchExist {
 | 
			
		||||
		log.Warn("Head branch of auto merge %-v does not exist [HeadRepoID: %d, Branch: %s]", pr, pr.HeadRepoID, pr.HeadBranch)
 | 
			
		||||
		return
 | 
			
		||||
| 
						 | 
				
			
			@ -238,25 +284,11 @@ func handlePull(pullID int64, sha string) {
 | 
			
		|||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var baseGitRepo *git.Repository
 | 
			
		||||
	if pr.BaseRepoID == pr.HeadRepoID {
 | 
			
		||||
		baseGitRepo = headGitRepo
 | 
			
		||||
	} else {
 | 
			
		||||
		if err = pr.LoadBaseRepo(ctx); err != nil {
 | 
			
		||||
			log.Error("%-v LoadBaseRepo: %v", pr, err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		baseGitRepo, err = gitrepo.OpenRepository(ctx, pr.BaseRepo)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Error("OpenRepository %-v: %v", pr.BaseRepo, err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		defer baseGitRepo.Close()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if err := pull_service.Merge(ctx, pr, doer, baseGitRepo, scheduledPRM.MergeStyle, "", scheduledPRM.Message, true); err != nil {
 | 
			
		||||
		log.Error("pull_service.Merge: %v", err)
 | 
			
		||||
		// FIXME: if merge failed, we should display some error message to the pull request page.
 | 
			
		||||
		// The resolution is add a new column on automerge table named `error_message` to store the error message and displayed
 | 
			
		||||
		// on the pull request page. But this should not be finished in a bug fix PR which will be backport to release branch.
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
							
								
								
									
										46
									
								
								services/automerge/notify.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										46
									
								
								services/automerge/notify.go
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,46 @@
 | 
			
		|||
// Copyright 2024 The Gitea Authors. All rights reserved.
 | 
			
		||||
// SPDX-License-Identifier: MIT
 | 
			
		||||
 | 
			
		||||
package automerge
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	issues_model "code.gitea.io/gitea/models/issues"
 | 
			
		||||
	user_model "code.gitea.io/gitea/models/user"
 | 
			
		||||
	"code.gitea.io/gitea/modules/log"
 | 
			
		||||
	notify_service "code.gitea.io/gitea/services/notify"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type automergeNotifier struct {
 | 
			
		||||
	notify_service.NullNotifier
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var _ notify_service.Notifier = &automergeNotifier{}
 | 
			
		||||
 | 
			
		||||
// NewNotifier create a new automergeNotifier notifier
 | 
			
		||||
func NewNotifier() notify_service.Notifier {
 | 
			
		||||
	return &automergeNotifier{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (n *automergeNotifier) PullRequestReview(ctx context.Context, pr *issues_model.PullRequest, review *issues_model.Review, comment *issues_model.Comment, mentions []*user_model.User) {
 | 
			
		||||
	// as a missing / blocking reviews could have blocked a pending automerge let's recheck
 | 
			
		||||
	if review.Type == issues_model.ReviewTypeApprove {
 | 
			
		||||
		if err := StartPRCheckAndAutoMergeBySHA(ctx, review.CommitID, pr.BaseRepo); err != nil {
 | 
			
		||||
			log.Error("StartPullRequestAutoMergeCheckBySHA: %v", err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (n *automergeNotifier) PullReviewDismiss(ctx context.Context, doer *user_model.User, review *issues_model.Review, comment *issues_model.Comment) {
 | 
			
		||||
	if err := review.LoadIssue(ctx); err != nil {
 | 
			
		||||
		log.Error("LoadIssue: %v", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if err := review.Issue.LoadPullRequest(ctx); err != nil {
 | 
			
		||||
		log.Error("LoadPullRequest: %v", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	// as reviews could have blocked a pending automerge let's recheck
 | 
			
		||||
	StartPRCheckAndAutoMerge(ctx, review.Issue.PullRequest)
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -117,7 +117,7 @@ func CreateCommitStatus(ctx context.Context, repo *repo_model.Repository, creato
 | 
			
		|||
	}
 | 
			
		||||
 | 
			
		||||
	if status.State.IsSuccess() {
 | 
			
		||||
		if err := automerge.MergeScheduledPullRequest(ctx, sha, repo); err != nil {
 | 
			
		||||
		if err := automerge.StartPRCheckAndAutoMergeBySHA(ctx, sha, repo); err != nil {
 | 
			
		||||
			return fmt.Errorf("MergeScheduledPullRequest[repo_id: %d, user_id: %d, sha: %s]: %w", repo.ID, creator.ID, sha, err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -30,28 +30,32 @@ import (
 | 
			
		|||
func TestCreateFile(t *testing.T) {
 | 
			
		||||
	onGiteaRun(t, func(t *testing.T, u *url.URL) {
 | 
			
		||||
		session := loginUser(t, "user2")
 | 
			
		||||
 | 
			
		||||
		// Request editor page
 | 
			
		||||
		req := NewRequest(t, "GET", "/user2/repo1/_new/master/")
 | 
			
		||||
		resp := session.MakeRequest(t, req, http.StatusOK)
 | 
			
		||||
 | 
			
		||||
		doc := NewHTMLParser(t, resp.Body)
 | 
			
		||||
		lastCommit := doc.GetInputValueByName("last_commit")
 | 
			
		||||
		assert.NotEmpty(t, lastCommit)
 | 
			
		||||
 | 
			
		||||
		// Save new file to master branch
 | 
			
		||||
		req = NewRequestWithValues(t, "POST", "/user2/repo1/_new/master/", map[string]string{
 | 
			
		||||
			"_csrf":          doc.GetCSRF(),
 | 
			
		||||
			"last_commit":    lastCommit,
 | 
			
		||||
			"tree_path":      "test.txt",
 | 
			
		||||
			"content":        "Content",
 | 
			
		||||
			"commit_choice":  "direct",
 | 
			
		||||
			"commit_mail_id": "3",
 | 
			
		||||
		})
 | 
			
		||||
		session.MakeRequest(t, req, http.StatusSeeOther)
 | 
			
		||||
		testCreateFile(t, session, "user2", "repo1", "master", "test.txt", "Content")
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func testCreateFile(t *testing.T, session *TestSession, user, repo, branch, filePath, content string) *httptest.ResponseRecorder {
 | 
			
		||||
	// Request editor page
 | 
			
		||||
	newURL := fmt.Sprintf("/%s/%s/_new/%s/", user, repo, branch)
 | 
			
		||||
	req := NewRequest(t, "GET", newURL)
 | 
			
		||||
	resp := session.MakeRequest(t, req, http.StatusOK)
 | 
			
		||||
 | 
			
		||||
	doc := NewHTMLParser(t, resp.Body)
 | 
			
		||||
	lastCommit := doc.GetInputValueByName("last_commit")
 | 
			
		||||
	assert.NotEmpty(t, lastCommit)
 | 
			
		||||
 | 
			
		||||
	// Save new file to master branch
 | 
			
		||||
	req = NewRequestWithValues(t, "POST", newURL, map[string]string{
 | 
			
		||||
		"_csrf":          doc.GetCSRF(),
 | 
			
		||||
		"last_commit":    lastCommit,
 | 
			
		||||
		"tree_path":      filePath,
 | 
			
		||||
		"content":        content,
 | 
			
		||||
		"commit_choice":  "direct",
 | 
			
		||||
		"commit_mail_id": "3",
 | 
			
		||||
	})
 | 
			
		||||
	return session.MakeRequest(t, req, http.StatusSeeOther)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestCreateFileOnProtectedBranch(t *testing.T) {
 | 
			
		||||
	onGiteaRun(t, func(t *testing.T, u *url.URL) {
 | 
			
		||||
		session := loginUser(t, "user2")
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -12,6 +12,8 @@ import (
 | 
			
		|||
	"net/url"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path"
 | 
			
		||||
	"path/filepath"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"testing"
 | 
			
		||||
	"time"
 | 
			
		||||
| 
						 | 
				
			
			@ -19,7 +21,9 @@ import (
 | 
			
		|||
	"code.gitea.io/gitea/models"
 | 
			
		||||
	auth_model "code.gitea.io/gitea/models/auth"
 | 
			
		||||
	"code.gitea.io/gitea/models/db"
 | 
			
		||||
	git_model "code.gitea.io/gitea/models/git"
 | 
			
		||||
	issues_model "code.gitea.io/gitea/models/issues"
 | 
			
		||||
	pull_model "code.gitea.io/gitea/models/pull"
 | 
			
		||||
	repo_model "code.gitea.io/gitea/models/repo"
 | 
			
		||||
	"code.gitea.io/gitea/models/unittest"
 | 
			
		||||
	user_model "code.gitea.io/gitea/models/user"
 | 
			
		||||
| 
						 | 
				
			
			@ -32,7 +36,9 @@ import (
 | 
			
		|||
	api "code.gitea.io/gitea/modules/structs"
 | 
			
		||||
	"code.gitea.io/gitea/modules/test"
 | 
			
		||||
	"code.gitea.io/gitea/modules/translation"
 | 
			
		||||
	"code.gitea.io/gitea/services/automerge"
 | 
			
		||||
	"code.gitea.io/gitea/services/pull"
 | 
			
		||||
	commitstatus_service "code.gitea.io/gitea/services/repository/commitstatus"
 | 
			
		||||
	files_service "code.gitea.io/gitea/services/repository/files"
 | 
			
		||||
	webhook_service "code.gitea.io/gitea/services/webhook"
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -661,3 +667,195 @@ func TestPullMergeIndexerNotifier(t *testing.T) {
 | 
			
		|||
		}
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func testResetRepo(t *testing.T, repoPath, branch, commitID string) {
 | 
			
		||||
	f, err := os.OpenFile(filepath.Join(repoPath, "refs", "heads", branch), os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0o644)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
	_, err = f.WriteString(commitID + "\n")
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
	f.Close()
 | 
			
		||||
 | 
			
		||||
	repo, err := git.OpenRepository(context.Background(), repoPath)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
	defer repo.Close()
 | 
			
		||||
	id, err := repo.GetBranchCommitID(branch)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
	assert.EqualValues(t, commitID, id)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestPullAutoMergeAfterCommitStatusSucceed(t *testing.T) {
 | 
			
		||||
	onGiteaRun(t, func(t *testing.T, giteaURL *url.URL) {
 | 
			
		||||
		// create a pull request
 | 
			
		||||
		session := loginUser(t, "user1")
 | 
			
		||||
		user1 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
 | 
			
		||||
		forkedName := "repo1-1"
 | 
			
		||||
		testRepoFork(t, session, "user2", "repo1", "user1", forkedName)
 | 
			
		||||
		defer func() {
 | 
			
		||||
			testDeleteRepository(t, session, "user1", forkedName)
 | 
			
		||||
		}()
 | 
			
		||||
		testEditFile(t, session, "user1", forkedName, "master", "README.md", "Hello, World (Edited)\n")
 | 
			
		||||
		testPullCreate(t, session, "user1", forkedName, false, "master", "master", "Indexer notifier test pull")
 | 
			
		||||
 | 
			
		||||
		baseRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{OwnerName: "user2", Name: "repo1"})
 | 
			
		||||
		forkedRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{OwnerName: "user1", Name: forkedName})
 | 
			
		||||
		pr := unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{
 | 
			
		||||
			BaseRepoID: baseRepo.ID,
 | 
			
		||||
			BaseBranch: "master",
 | 
			
		||||
			HeadRepoID: forkedRepo.ID,
 | 
			
		||||
			HeadBranch: "master",
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// add protected branch for commit status
 | 
			
		||||
		csrf := GetCSRF(t, session, "/user2/repo1/settings/branches")
 | 
			
		||||
		// Change master branch to protected
 | 
			
		||||
		req := NewRequestWithValues(t, "POST", "/user2/repo1/settings/branches/edit", map[string]string{
 | 
			
		||||
			"_csrf":                 csrf,
 | 
			
		||||
			"rule_name":             "master",
 | 
			
		||||
			"enable_push":           "true",
 | 
			
		||||
			"enable_status_check":   "true",
 | 
			
		||||
			"status_check_contexts": "gitea/actions",
 | 
			
		||||
		})
 | 
			
		||||
		session.MakeRequest(t, req, http.StatusSeeOther)
 | 
			
		||||
 | 
			
		||||
		// first time insert automerge record, return true
 | 
			
		||||
		scheduled, err := automerge.ScheduleAutoMerge(db.DefaultContext, user1, pr, repo_model.MergeStyleMerge, "auto merge test")
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
		assert.True(t, scheduled)
 | 
			
		||||
 | 
			
		||||
		// second time insert automerge record, return false because it does exist
 | 
			
		||||
		scheduled, err = automerge.ScheduleAutoMerge(db.DefaultContext, user1, pr, repo_model.MergeStyleMerge, "auto merge test")
 | 
			
		||||
		assert.Error(t, err)
 | 
			
		||||
		assert.False(t, scheduled)
 | 
			
		||||
 | 
			
		||||
		// reload pr again
 | 
			
		||||
		pr = unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{ID: pr.ID})
 | 
			
		||||
		assert.False(t, pr.HasMerged)
 | 
			
		||||
		assert.Empty(t, pr.MergedCommitID)
 | 
			
		||||
 | 
			
		||||
		// update commit status to success, then it should be merged automatically
 | 
			
		||||
		baseGitRepo, err := gitrepo.OpenRepository(db.DefaultContext, baseRepo)
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
		sha, err := baseGitRepo.GetRefCommitID(pr.GetGitRefName())
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
		masterCommitID, err := baseGitRepo.GetBranchCommitID("master")
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
 | 
			
		||||
		branches, _, err := baseGitRepo.GetBranchNames(0, 100)
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
		assert.ElementsMatch(t, []string{"sub-home-md-img-check", "home-md-img-check", "pr-to-update", "branch2", "DefaultBranch", "develop", "feature/1", "master"}, branches)
 | 
			
		||||
		baseGitRepo.Close()
 | 
			
		||||
		defer func() {
 | 
			
		||||
			testResetRepo(t, baseRepo.RepoPath(), "master", masterCommitID)
 | 
			
		||||
		}()
 | 
			
		||||
 | 
			
		||||
		err = commitstatus_service.CreateCommitStatus(db.DefaultContext, baseRepo, user1, sha, &git_model.CommitStatus{
 | 
			
		||||
			State:     api.CommitStatusSuccess,
 | 
			
		||||
			TargetURL: "https://gitea.com",
 | 
			
		||||
			Context:   "gitea/actions",
 | 
			
		||||
		})
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
 | 
			
		||||
		time.Sleep(2 * time.Second)
 | 
			
		||||
 | 
			
		||||
		// realod pr again
 | 
			
		||||
		pr = unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{ID: pr.ID})
 | 
			
		||||
		assert.True(t, pr.HasMerged)
 | 
			
		||||
		assert.NotEmpty(t, pr.MergedCommitID)
 | 
			
		||||
 | 
			
		||||
		unittest.AssertNotExistsBean(t, &pull_model.AutoMerge{PullID: pr.ID})
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestPullAutoMergeAfterCommitStatusSucceedAndApproval(t *testing.T) {
 | 
			
		||||
	onGiteaRun(t, func(t *testing.T, giteaURL *url.URL) {
 | 
			
		||||
		// create a pull request
 | 
			
		||||
		session := loginUser(t, "user1")
 | 
			
		||||
		user1 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
 | 
			
		||||
		forkedName := "repo1-2"
 | 
			
		||||
		testRepoFork(t, session, "user2", "repo1", "user1", forkedName)
 | 
			
		||||
		defer func() {
 | 
			
		||||
			testDeleteRepository(t, session, "user1", forkedName)
 | 
			
		||||
		}()
 | 
			
		||||
		testEditFile(t, session, "user1", forkedName, "master", "README.md", "Hello, World (Edited)\n")
 | 
			
		||||
		testPullCreate(t, session, "user1", forkedName, false, "master", "master", "Indexer notifier test pull")
 | 
			
		||||
 | 
			
		||||
		baseRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{OwnerName: "user2", Name: "repo1"})
 | 
			
		||||
		forkedRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{OwnerName: "user1", Name: forkedName})
 | 
			
		||||
		pr := unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{
 | 
			
		||||
			BaseRepoID: baseRepo.ID,
 | 
			
		||||
			BaseBranch: "master",
 | 
			
		||||
			HeadRepoID: forkedRepo.ID,
 | 
			
		||||
			HeadBranch: "master",
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// add protected branch for commit status
 | 
			
		||||
		csrf := GetCSRF(t, session, "/user2/repo1/settings/branches")
 | 
			
		||||
		// Change master branch to protected
 | 
			
		||||
		req := NewRequestWithValues(t, "POST", "/user2/repo1/settings/branches/edit", map[string]string{
 | 
			
		||||
			"_csrf":                 csrf,
 | 
			
		||||
			"rule_name":             "master",
 | 
			
		||||
			"enable_push":           "true",
 | 
			
		||||
			"enable_status_check":   "true",
 | 
			
		||||
			"status_check_contexts": "gitea/actions",
 | 
			
		||||
			"required_approvals":    "1",
 | 
			
		||||
		})
 | 
			
		||||
		session.MakeRequest(t, req, http.StatusSeeOther)
 | 
			
		||||
 | 
			
		||||
		// first time insert automerge record, return true
 | 
			
		||||
		scheduled, err := automerge.ScheduleAutoMerge(db.DefaultContext, user1, pr, repo_model.MergeStyleMerge, "auto merge test")
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
		assert.True(t, scheduled)
 | 
			
		||||
 | 
			
		||||
		// second time insert automerge record, return false because it does exist
 | 
			
		||||
		scheduled, err = automerge.ScheduleAutoMerge(db.DefaultContext, user1, pr, repo_model.MergeStyleMerge, "auto merge test")
 | 
			
		||||
		assert.Error(t, err)
 | 
			
		||||
		assert.False(t, scheduled)
 | 
			
		||||
 | 
			
		||||
		// reload pr again
 | 
			
		||||
		pr = unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{ID: pr.ID})
 | 
			
		||||
		assert.False(t, pr.HasMerged)
 | 
			
		||||
		assert.Empty(t, pr.MergedCommitID)
 | 
			
		||||
 | 
			
		||||
		// update commit status to success, then it should be merged automatically
 | 
			
		||||
		baseGitRepo, err := gitrepo.OpenRepository(db.DefaultContext, baseRepo)
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
		sha, err := baseGitRepo.GetRefCommitID(pr.GetGitRefName())
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
		masterCommitID, err := baseGitRepo.GetBranchCommitID("master")
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
		baseGitRepo.Close()
 | 
			
		||||
		defer func() {
 | 
			
		||||
			testResetRepo(t, baseRepo.RepoPath(), "master", masterCommitID)
 | 
			
		||||
		}()
 | 
			
		||||
 | 
			
		||||
		err = commitstatus_service.CreateCommitStatus(db.DefaultContext, baseRepo, user1, sha, &git_model.CommitStatus{
 | 
			
		||||
			State:     api.CommitStatusSuccess,
 | 
			
		||||
			TargetURL: "https://gitea.com",
 | 
			
		||||
			Context:   "gitea/actions",
 | 
			
		||||
		})
 | 
			
		||||
		assert.NoError(t, err)
 | 
			
		||||
 | 
			
		||||
		time.Sleep(2 * time.Second)
 | 
			
		||||
 | 
			
		||||
		// reload pr again
 | 
			
		||||
		pr = unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{ID: pr.ID})
 | 
			
		||||
		assert.False(t, pr.HasMerged)
 | 
			
		||||
		assert.Empty(t, pr.MergedCommitID)
 | 
			
		||||
 | 
			
		||||
		// approve the PR from non-author
 | 
			
		||||
		approveSession := loginUser(t, "user2")
 | 
			
		||||
		req = NewRequest(t, "GET", fmt.Sprintf("/user2/repo1/pulls/%d", pr.Index))
 | 
			
		||||
		resp := approveSession.MakeRequest(t, req, http.StatusOK)
 | 
			
		||||
		htmlDoc := NewHTMLParser(t, resp.Body)
 | 
			
		||||
		testSubmitReview(t, approveSession, htmlDoc.GetCSRF(), "user2", "repo1", strconv.Itoa(int(pr.Index)), sha, "approve", http.StatusOK)
 | 
			
		||||
 | 
			
		||||
		time.Sleep(2 * time.Second)
 | 
			
		||||
 | 
			
		||||
		// realod pr again
 | 
			
		||||
		pr = unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{ID: pr.ID})
 | 
			
		||||
		assert.True(t, pr.HasMerged)
 | 
			
		||||
		assert.NotEmpty(t, pr.MergedCommitID)
 | 
			
		||||
 | 
			
		||||
		unittest.AssertNotExistsBean(t, &pull_model.AutoMerge{PullID: pr.ID})
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -425,10 +425,10 @@ func TestPullView_GivenApproveOrRejectReviewOnClosedPR(t *testing.T) {
 | 
			
		|||
			htmlDoc := NewHTMLParser(t, resp.Body)
 | 
			
		||||
 | 
			
		||||
			// Submit an approve review on the PR.
 | 
			
		||||
			testSubmitReview(t, user2Session, htmlDoc.GetCSRF(), "user2", "repo1", elem[4], "approve", http.StatusUnprocessableEntity)
 | 
			
		||||
			testSubmitReview(t, user2Session, htmlDoc.GetCSRF(), "user2", "repo1", elem[4], "", "approve", http.StatusUnprocessableEntity)
 | 
			
		||||
 | 
			
		||||
			// Submit a reject review on the PR.
 | 
			
		||||
			testSubmitReview(t, user2Session, htmlDoc.GetCSRF(), "user2", "repo1", elem[4], "reject", http.StatusUnprocessableEntity)
 | 
			
		||||
			testSubmitReview(t, user2Session, htmlDoc.GetCSRF(), "user2", "repo1", elem[4], "", "reject", http.StatusUnprocessableEntity)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		t.Run("Submit approve/reject review on closed PR", func(t *testing.T) {
 | 
			
		||||
| 
						 | 
				
			
			@ -445,18 +445,18 @@ func TestPullView_GivenApproveOrRejectReviewOnClosedPR(t *testing.T) {
 | 
			
		|||
			htmlDoc := NewHTMLParser(t, resp.Body)
 | 
			
		||||
 | 
			
		||||
			// Submit an approve review on the PR.
 | 
			
		||||
			testSubmitReview(t, user2Session, htmlDoc.GetCSRF(), "user2", "repo1", elem[4], "approve", http.StatusUnprocessableEntity)
 | 
			
		||||
			testSubmitReview(t, user2Session, htmlDoc.GetCSRF(), "user2", "repo1", elem[4], "", "approve", http.StatusUnprocessableEntity)
 | 
			
		||||
 | 
			
		||||
			// Submit a reject review on the PR.
 | 
			
		||||
			testSubmitReview(t, user2Session, htmlDoc.GetCSRF(), "user2", "repo1", elem[4], "reject", http.StatusUnprocessableEntity)
 | 
			
		||||
			testSubmitReview(t, user2Session, htmlDoc.GetCSRF(), "user2", "repo1", elem[4], "", "reject", http.StatusUnprocessableEntity)
 | 
			
		||||
		})
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func testSubmitReview(t *testing.T, session *TestSession, csrf, owner, repo, pullNumber, reviewType string, expectedSubmitStatus int) *httptest.ResponseRecorder {
 | 
			
		||||
func testSubmitReview(t *testing.T, session *TestSession, csrf, owner, repo, pullNumber, commitID, reviewType string, expectedSubmitStatus int) *httptest.ResponseRecorder {
 | 
			
		||||
	options := map[string]string{
 | 
			
		||||
		"_csrf":     csrf,
 | 
			
		||||
		"commit_id": "",
 | 
			
		||||
		"commit_id": commitID,
 | 
			
		||||
		"content":   "test",
 | 
			
		||||
		"type":      reviewType,
 | 
			
		||||
	}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue