add logic for adding labels and assignees to the resulting PR that gets created by this CLI

Author: GrantBirki

go.mod

@@ -10,11 +10,13 @@ require (
 	github.com/cli/go-gh/v2 v2.12.0
 	github.com/cli/shurcooL-graphql v0.0.4
 	github.com/spf13/cobra v1.9.1
+	github.com/stretchr/testify v1.10.0
 )
 
 require (
 	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
 	github.com/cli/safeexec v1.0.0 // indirect
+	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/fatih/color v1.7.0 // indirect
 	github.com/henvic/httpretty v0.0.6 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
@@ -23,6 +25,7 @@ require (
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
 	github.com/muesli/termenv v0.16.0 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/spf13/pflag v1.0.6 // indirect
 	github.com/thlib/go-timezone-local v0.0.0-20210907160436-ef149e42d28e // indirect

go.sum

@@ -42,8 +42,8 @@ github.com/spf13/cobra v1.9.1 h1:CXSaggrXdbHK9CF+8ywj8Amf7PBRmPCOJugH954Nnlo=
 github.com/spf13/cobra v1.9.1/go.mod h1:nDyEzZ8ogv936Cinf6g1RU9MRY64Ir93oCnqb9wxYW0=
 github.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o=
 github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
-github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
-github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
+github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/thlib/go-timezone-local v0.0.0-20210907160436-ef149e42d28e h1:BuzhfgfWQbX0dWzYzT1zsORLnHRv3bcRcsaUk0VmXA8=
 github.com/thlib/go-timezone-local v0.0.0-20210907160436-ef149e42d28e/go.mod h1:/Tnicc6m/lsJE0irFMA0LfIwTBo4QP7A8IfyIv4zZKI=
 golang.org/x/sys v0.0.0-20210831042530-f4d43177bf5e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=

internal/cmd/combine_prs.go

@@ -12,7 +12,15 @@ import (
 	"github.com/github/gh-combine/internal/github"
 )
 
-func CombinePRs(ctx context.Context, graphQlClient *api.GraphQLClient, restClient *api.RESTClient, repo github.Repo, pulls github.Pulls) error {
+// Updated RESTClientInterface to match the method signatures of api.RESTClient
+type RESTClientInterface interface {
+	Post(endpoint string, body io.Reader, response interface{}) error
+	Get(endpoint string, response interface{}) error
+	Delete(endpoint string, response interface{}) error
+	Patch(endpoint string, body io.Reader, response interface{}) error
+}
+
+func CombinePRs(ctx context.Context, graphQlClient *api.GraphQLClient, restClient RESTClientInterface, repo github.Repo, pulls github.Pulls) error {
 	// Define the combined branch name
 	workingBranchName := combineBranchName + workingBranchSuffix
 
@@ -87,7 +95,7 @@ func CombinePRs(ctx context.Context, graphQlClient *api.GraphQLClient, restClien
 	// Create the combined PR
 	prBody := generatePRBody(combinedPRs, mergeFailedPRs)
 	prTitle := "Combined PRs"
-	err = createPullRequest(ctx, restClient, repo, prTitle, combineBranchName, repoDefaultBranch, prBody)
+	err = createPullRequest(ctx, restClient, repo, prTitle, combineBranchName, repoDefaultBranch, prBody, addLabels, addAssignees)
 	if err != nil {
 		return fmt.Errorf("failed to create combined PR: %w", err)
 	}
@@ -102,7 +110,7 @@ func isMergeConflictError(err error) bool {
 }
 
 // Find the default branch of a repository
-func getDefaultBranch(ctx context.Context, client *api.RESTClient, repo github.Repo) (string, error) {
+func getDefaultBranch(ctx context.Context, client RESTClientInterface, repo github.Repo) (string, error) {
 	var repoInfo struct {
 		DefaultBranch string `json:"default_branch"`
 	}
@@ -115,7 +123,7 @@ func getDefaultBranch(ctx context.Context, client *api.RESTClient, repo github.R
 }
 
 // Get the SHA of a given branch
-func getBranchSHA(ctx context.Context, client *api.RESTClient, repo github.Repo, branch string) (string, error) {
+func getBranchSHA(ctx context.Context, client RESTClientInterface, repo github.Repo, branch string) (string, error) {
 	var ref struct {
 		Object struct {
 			SHA string `json:"sha"`
@@ -148,13 +156,13 @@ func generatePRBody(combinedPRs, mergeFailedPRs []string) string {
 }
 
 // deleteBranch deletes a branch in the repository
-func deleteBranch(ctx context.Context, client *api.RESTClient, repo github.Repo, branch string) error {
+func deleteBranch(ctx context.Context, client RESTClientInterface, repo github.Repo, branch string) error {
 	endpoint := fmt.Sprintf("repos/%s/%s/git/refs/heads/%s", repo.Owner, repo.Repo, branch)
 	return client.Delete(endpoint, nil)
 }
 
 // createBranch creates a new branch in the repository
-func createBranch(ctx context.Context, client *api.RESTClient, repo github.Repo, branch, sha string) error {
+func createBranch(ctx context.Context, client RESTClientInterface, repo github.Repo, branch, sha string) error {
 	endpoint := fmt.Sprintf("repos/%s/%s/git/refs", repo.Owner, repo.Repo)
 	payload := map[string]string{
 		"ref": "refs/heads/" + branch,
@@ -168,7 +176,7 @@ func createBranch(ctx context.Context, client *api.RESTClient, repo github.Repo,
 }
 
 // mergeBranch merges a branch into the base branch
-func mergeBranch(ctx context.Context, client *api.RESTClient, repo github.Repo, base, head string) error {
+func mergeBranch(ctx context.Context, client RESTClientInterface, repo github.Repo, base, head string) error {
 	endpoint := fmt.Sprintf("repos/%s/%s/merges", repo.Owner, repo.Repo)
 	payload := map[string]string{
 		"base": base,
@@ -182,7 +190,7 @@ func mergeBranch(ctx context.Context, client *api.RESTClient, repo github.Repo,
 }
 
 // updateRef updates a branch to point to the latest commit of another branch
-func updateRef(ctx context.Context, client *api.RESTClient, repo github.Repo, branch, sourceBranch string) error {
+func updateRef(ctx context.Context, client RESTClientInterface, repo github.Repo, branch, sourceBranch string) error {
 	// Get the SHA of the source branch
 	var ref struct {
 		Object struct {
@@ -208,15 +216,25 @@ func updateRef(ctx context.Context, client *api.RESTClient, repo github.Repo, br
 	return client.Patch(endpoint, body, nil)
 }
 
-// createPullRequest creates a new pull request
-func createPullRequest(ctx context.Context, client *api.RESTClient, repo github.Repo, title, head, base, body string) error {
+func createPullRequest(ctx context.Context, client RESTClientInterface, repo github.Repo, title, head, base, body string, labels, assignees []string) error {
 	endpoint := fmt.Sprintf("repos/%s/%s/pulls", repo.Owner, repo.Repo)
-	payload := map[string]string{
-		"title": title,
-		"head":  head,
-		"base":  base,
-		"body":  body,
+	payload := map[string]interface{}{
+		"title":     title,
+		"head":      head,
+		"base":      base,
+		"body":      body,
 	}
+
+	// Add labels if provided
+	if len(labels) > 0 {
+		payload["labels"] = labels
+	}
+
+	// Add assignees if provided
+	if len(assignees) > 0 {
+		payload["assignees"] = assignees
+	}
+
 	requestBody, err := encodePayload(payload)
 	if err != nil {
 		return fmt.Errorf("failed to encode payload: %w", err)

internal/cmd/combine_prs_test.go

@@ -0,0 +1,30 @@
+package cmd
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/github/gh-combine/internal/github"
+)
+
+func TestCreatePullRequest(t *testing.T) {
+	client := &MockRESTClient{
+		PostFunc: func(endpoint string, body interface{}, response interface{}) error {
+			return nil
+		},
+	}
+	repo := github.Repo{
+		Owner: "test-owner",
+		Repo:  "test-repo",
+	}
+	title := "Test PR"
+	head := "test-branch"
+	base := "main"
+	body := "This is a test PR."
+	labels := []string{"bug", "enhancement"}
+	assignees := []string{"octocat", "hubot"}
+
+	err := createPullRequest(context.Background(), client, repo, title, head, base, body, labels, assignees)
+	assert.NoError(t, err)
+}

internal/cmd/mock_restclient.go

@@ -0,0 +1,64 @@
+package cmd
+
+import (
+	"context"
+	"io"
+	"net/http"
+)
+
+type MockRESTClient struct {
+	PostFunc    func(endpoint string, body interface{}, response interface{}) error
+	GetFunc     func(endpoint string, response interface{}) error
+	DeleteFunc  func(endpoint string, response interface{}) error
+	PatchFunc   func(endpoint string, body io.Reader, response interface{}) error
+}
+
+// Updated the Post method to match the RESTClientInterface signature
+func (m *MockRESTClient) Post(endpoint string, body io.Reader, response interface{}) error {
+	if m.PostFunc != nil {
+		return m.PostFunc(endpoint, body, response)
+	}
+	return nil
+}
+
+func (m *MockRESTClient) Get(endpoint string, response interface{}) error {
+	if m.GetFunc != nil {
+		return m.GetFunc(endpoint, response)
+	}
+	return nil
+}
+
+func (m *MockRESTClient) Delete(endpoint string, response interface{}) error {
+	if m.DeleteFunc != nil {
+		return m.DeleteFunc(endpoint, response)
+	}
+	return nil
+}
+
+// Updated the Patch method to match the RESTClientInterface signature
+func (m *MockRESTClient) Patch(endpoint string, body io.Reader, response interface{}) error {
+	if m.PatchFunc != nil {
+		return m.PatchFunc(endpoint, body, response)
+	}
+	return nil
+}
+
+func (m *MockRESTClient) RequestWithContext(ctx context.Context, method string, path string, body io.Reader) (*http.Response, error) {
+	return nil, nil
+}
+
+func (m *MockRESTClient) Request(method string, path string, body io.Reader) (*http.Response, error) {
+	return nil, nil
+}
+
+func (m *MockRESTClient) DoWithContext(ctx context.Context, method string, path string, body io.Reader, response interface{}) error {
+	return nil
+}
+
+func (m *MockRESTClient) Do(method string, path string, body io.Reader, response interface{}) error {
+	return nil
+}
+
+func (m *MockRESTClient) Put(path string, body io.Reader, resp interface{}) error {
+	return nil
+}

internal/cmd/root.go

@@ -289,8 +289,13 @@ func processRepository(ctx context.Context, client *api.RESTClient, graphQlClien
 
 	Logger.Debug("Matched PRs", "repo", repo, "count", len(matchedPRs))
 
+	// Wrap the *api.RESTClient to implement RESTClientInterface
+	restClientWrapper := struct {
+		RESTClientInterface
+	}{client}
+
 	// Combine the PRs
-	err = CombinePRs(ctx, graphQlClient, client, repo, matchedPRs)
+	err = CombinePRs(ctx, graphQlClient, restClientWrapper, repo, matchedPRs)
 	if err != nil {
 		return fmt.Errorf("failed to combine PRs: %w", err)
 	}