add IsMerged to CIService

Author: veziak

pkg/azure/azure.go

@@ -295,7 +295,18 @@ func (a *AzureReposService) IsClosed(prNumber int) (bool, error) {
 	if err != nil {
 		return false, err
 	}
-	return *pullRequest.Status == git.PullRequestStatusValues.Completed || *pullRequest.Status == git.PullRequestStatusValues.Abandoned, nil
+	return *pullRequest.Status == git.PullRequestStatusValues.Abandoned, nil
+}
+
+func (a *AzureReposService) IsMerged(prNumber int) (bool, error) {
+	pullRequest, err := a.Client.GetPullRequestById(context.Background(), git.GetPullRequestByIdArgs{
+		Project:       &a.ProjectName,
+		PullRequestId: &prNumber,
+	})
+	if err != nil {
+		return false, err
+	}
+	return *pullRequest.Status == git.PullRequestStatusValues.Completed, nil
 }
 
 func ProcessAzureReposEvent(azureEvent interface{}, diggerConfig *configuration.DiggerConfig, ciService ci.CIService) ([]configuration.Project, *configuration.Project, int, error) {

pkg/ci/ci.go

@@ -7,6 +7,10 @@ type CIService interface {
 	SetStatus(prNumber int, status string, statusContext string) error
 	GetCombinedPullRequestStatus(prNumber int) (string, error)
 	MergePullRequest(prNumber int) error
+	// IsMergeable is still open and ready to be merged
 	IsMergeable(prNumber int) (bool, error)
+	// IsMerged merged and closed
+	IsMerged(prNumber int) (bool, error)
+	// IsClosed closed without merging
 	IsClosed(prNumber int) (bool, error)
 }

pkg/digger/digger.go

@@ -173,18 +173,18 @@ func RunCommandsPerProject(
 					return false, false, fmt.Errorf("failed to set PR status. %v", err)
 				}
 
-				isClosed, err := ciService.IsClosed(prNumber)
+				isMerged, err := ciService.IsMerged(prNumber)
 				if err != nil {
-					return false, false, fmt.Errorf("error validating is PR is closed: %v", err)
+					return false, false, fmt.Errorf("error checking if PR is merged: %v", err)
 				}
 
 				// this might go into some sort of "appliability" plugin later
 				isMergeable, err := ciService.IsMergeable(prNumber)
 				if err != nil {
 					return false, false, fmt.Errorf("error validating is PR is mergeable: %v", err)
 				}
-				fmt.Printf("PR status, mergeable: %v, closed: %v\n", isMergeable, isClosed)
-				if !isMergeable && !isClosed {
+				fmt.Printf("PR status, mergeable: %v, merged: %v\n", isMergeable, isMerged)
+				if !isMergeable && !isMerged {
 					comment := "Cannot perform Apply since the PR is not currently mergeable."
 					fmt.Println(comment)
 					err = ciService.PublishComment(prNumber, comment)

pkg/digger/digger_test.go

@@ -90,6 +90,11 @@ func (m *MockPRManager) DownloadLatestPlans(prNumber int) (string, error) {
 	return "plan", nil
 }
 
+func (m *MockPRManager) IsMerged(prNumber int) (bool, error) {
+	m.Commands = append(m.Commands, RunInfo{"IsClosed", strconv.Itoa(prNumber), time.Now()})
+	return false, nil
+}
+
 func (m *MockPRManager) IsClosed(prNumber int) (bool, error) {
 	m.Commands = append(m.Commands, RunInfo{"IsClosed", strconv.Itoa(prNumber), time.Now()})
 	return false, nil

pkg/github/github.go

@@ -109,15 +109,26 @@ func (svc *GithubService) IsMergeable(prNumber int) (bool, error) {
 	pr, _, err := svc.Client.PullRequests.Get(context.Background(), svc.Owner, svc.RepoName, prNumber)
 	if err != nil {
 		log.Fatalf("error getting pull request: %v", err)
+		return false, err
 	}
 
 	return pr.GetMergeable() && isMergeableState(pr.GetMergeableState()), nil
 }
 
+func (svc *GithubService) IsMerged(prNumber int) (bool, error) {
+	pr, _, err := svc.Client.PullRequests.Get(context.Background(), svc.Owner, svc.RepoName, prNumber)
+	if err != nil {
+		log.Fatalf("error getting pull request: %v", err)
+		return false, err
+	}
+	return *pr.Merged, nil
+}
+
 func (svc *GithubService) IsClosed(prNumber int) (bool, error) {
 	pr, _, err := svc.Client.PullRequests.Get(context.Background(), svc.Owner, svc.RepoName, prNumber)
 	if err != nil {
 		log.Fatalf("error getting pull request: %v", err)
+		return false, err
 	}
 
 	return pr.GetState() == "closed", nil

pkg/gitlab/gitlab.go

@@ -208,23 +208,32 @@ func (gitlabService GitLabService) IsMergeable(mergeRequestID int) (bool, error)
 }
 
 func (gitlabService GitLabService) IsClosed(mergeRequestID int) (bool, error) {
+	mergeRequest := getMergeRequest(gitlabService)
+	if mergeRequest.State == "closed" {
+		return true, nil
+	}
+	return false, nil
+}
+
+func (gitlabService GitLabService) IsMerged(mergeRequestID int) (bool, error) {
+	mergeRequest := getMergeRequest(gitlabService)
+	if mergeRequest.State == "merged" {
+		return true, nil
+	}
+	return false, nil
+}
+
+func getMergeRequest(gitlabService GitLabService) *go_gitlab.MergeRequest {
 	projectId := *gitlabService.Context.ProjectId
 	mergeRequestIID := *gitlabService.Context.MergeRequestIId
-
-	fmt.Printf("IsClosed mergeRequestIID : %d, projectId: %d \n", mergeRequestIID, projectId)
+	fmt.Printf("getMergeRequest mergeRequestIID : %d, projectId: %d \n", mergeRequestIID, projectId)
 	opt := &go_gitlab.GetMergeRequestsOptions{}
-
 	mergeRequest, _, err := gitlabService.Client.MergeRequests.GetMergeRequest(projectId, mergeRequestIID, opt)
-
 	if err != nil {
 		fmt.Printf("Failed to get a MergeRequest: %d, %v \n", mergeRequestIID, err)
 		print(err.Error())
 	}
-
-	if mergeRequest.State == "closed" || mergeRequest.State == "merged" {
-		return true, nil
-	}
-	return false, nil
+	return mergeRequest
 }
 
 type GitLabEvent struct {

pkg/utils/mocks.go

@@ -73,6 +73,10 @@ func (t MockPullRequestManager) IsMergeable(prNumber int) (bool, error) {
 	return true, nil
 }
 
+func (t MockPullRequestManager) IsMerged(prNumber int) (bool, error) {
+	return false, nil
+}
+
 func (t MockPullRequestManager) DownloadLatestPlans(prNumber int) (string, error) {
 	return "", nil
 }

pkg/utils/pullrequestmanager_mock.go

@@ -43,3 +43,8 @@ func (mockGithubPullrequestManager *MockGithubPullrequestManager) IsClosed(prNum
 	mockGithubPullrequestManager.commands = append(mockGithubPullrequestManager.commands, "IsClosed")
 	return false, nil
 }
+
+func (mockGithubPullrequestManager *MockGithubPullrequestManager) IsMerged(prNumber int) (bool, error) {
+	mockGithubPullrequestManager.commands = append(mockGithubPullrequestManager.commands, "IsClosed")
+	return false, nil
+}