Add test for automerge deletion

routers/private/hook_post_receive_test.go

@@ -8,7 +8,10 @@ import (
 
 	"code.gitea.io/gitea/models/db"
 	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"
 	"code.gitea.io/gitea/modules/private"
 	repo_module "code.gitea.io/gitea/modules/repository"
 	"code.gitea.io/gitea/services/contexttest"
@@ -21,6 +24,14 @@ func TestHandlePullRequestMerging(t *testing.T) {
 	pr, err := issues_model.GetUnmergedPullRequest(db.DefaultContext, 1, 1, "branch2", "master", issues_model.PullRequestFlowGithub)
 	assert.NoError(t, err)
 	assert.NoError(t, pr.LoadBaseRepo(db.DefaultContext))
+
+	user1 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
+
+	err = pull_model.ScheduleAutoMerge(db.DefaultContext, user1, pr.ID, repo_model.MergeStyleSquash, "squash merge a pr")
+	assert.NoError(t, err)
+
+	autoMerge := unittest.AssertExistsAndLoadBean(t, &pull_model.AutoMerge{PullID: pr.ID})
+
 	ctx, resp := contexttest.MockPrivateContext(t, "/")
 	handlePullRequestMerging(ctx, &private.HookOptions{
 		PullRequestID: pr.ID,
@@ -33,5 +44,6 @@ func TestHandlePullRequestMerging(t *testing.T) {
 	assert.NoError(t, err)
 	assert.True(t, pr.HasMerged)
 	assert.EqualValues(t, "01234567", pr.MergedCommitID)
-	// TODO: test the removal of auto merge
+
+	unittest.AssertNotExistsBean(t, &pull_model.AutoMerge{ID: autoMerge.ID})
 }