Merge pull request drone#63 from bigkevmcd/create-pull-request

feat: Add support for creating PullRequests.

scm/driver/bitbucket/pr.go

@@ -67,6 +67,10 @@ func (s *pullService) Close(ctx context.Context, repo string, number int) (*scm.
 	return nil, scm.ErrNotSupported
 }
 
+func (s *pullService) Create(ctx context.Context, repo string, input *scm.PullRequestInput) (*scm.PullRequest, *scm.Response, error) {
+	return nil, nil, scm.ErrNotSupported
+}
+
 type prCommit struct {
 	LatestCommit string `json:"hash"`
 }

scm/driver/bitbucket/pr_test.go

@@ -92,3 +92,18 @@ func TestPullClose(t *testing.T) {
 		t.Errorf("Expect Not Supported error")
 	}
 }
+
+func TestPullCreate(t *testing.T) {
+	client, _ := New("https://api.bitbucket.org")
+	input := &scm.PullRequestInput{
+		Title: "Bitbucket feature",
+		Body:  "New Bitbucket feature",
+		Head:  "new-feature",
+		Base:  "master",
+	}
+
+	_, _, err := client.PullRequests.Create(context.Background(), "atlassian/atlaskit", input)
+	if err != scm.ErrNotSupported {
+		t.Errorf("Expect Not Supported error")
+	}
+}

scm/driver/fake/data.go

@@ -22,6 +22,8 @@ type Data struct {
 	IssueEvents                map[int][]*scm.ListedIssueEvent
 	Commits                    map[string]*scm.Commit
 	TestRef                    string
+	PullRequestsCreated        map[int]*scm.PullRequestInput
+	PullRequestID              int
 
 	//All Labels That Exist In The Repo
 	RepoLabelsExisting []string
@@ -77,6 +79,7 @@ func NewData() *Data {
 		PullRequestLabelsAdded:    []string{},
 		PullRequestLabelsRemoved:  []string{},
 		PullRequestLabelsExisting: []string{},
+		PullRequestsCreated:       map[int]*scm.PullRequestInput{},
 		Reviews:                   map[int][]*scm.Review{},
 		Statuses:                  map[string][]*scm.Status{},
 		IssueEvents:               map[int][]*scm.ListedIssueEvent{},

scm/driver/fake/pr.go

@@ -124,3 +124,21 @@ func (s *pullService) DeleteComment(ctx context.Context, repo string, number int
 	}
 	return nil, fmt.Errorf("could not find issue comment %d", id)
 }
+
+func (s *pullService) Create(ctx context.Context, repo string, input *scm.PullRequestInput) (*scm.PullRequest, *scm.Response, error) {
+	f := s.data
+	answer := &scm.PullRequest{
+		Number: f.PullRequestID,
+		Title:  input.Title,
+		Body:   input.Body,
+		Base: scm.PullRequestBranch{
+			Ref: input.Base,
+		},
+		Head: scm.PullRequestBranch{
+			Ref: input.Head,
+		},
+	}
+	f.PullRequestsCreated[f.PullRequestID] = input
+	f.PullRequestID++
+	return answer, nil, nil
+}

scm/driver/gitea/pr.go

@@ -73,6 +73,10 @@ func (s *pullService) Close(context.Context, string, int) (*scm.Response, error)
 	return nil, scm.ErrNotSupported
 }
 
+func (s *pullService) Create(ctx context.Context, repo string, input *scm.PullRequestInput) (*scm.PullRequest, *scm.Response, error) {
+	return nil, nil, scm.ErrNotSupported
+}
+
 //
 // native data structures
 //

scm/driver/gitea/pr_test.go

@@ -139,3 +139,18 @@ func TestPullRequestCommentDelete(t *testing.T) {
 		t.Errorf("Expect Not Supported error")
 	}
 }
+
+func TestPullCreate(t *testing.T) {
+	client, _ := New("https://try.gitea.io")
+	input := &scm.PullRequestInput{
+		Title: "Gitea feature",
+		Body:  "New Gitea feature",
+		Head:  "new-feature",
+		Base:  "master",
+	}
+
+	_, _, err := client.PullRequests.Create(context.Background(), "go-gitea/gitea", input)
+	if err != scm.ErrNotSupported {
+		t.Errorf("Expect Not Supported error")
+	}
+}

scm/driver/github/pr.go

@@ -51,6 +51,20 @@ func (s *pullService) Close(ctx context.Context, repo string, number int) (*scm.
 	return res, err
 }
 
+func (s *pullService) Create(ctx context.Context, repo string, input *scm.PullRequestInput) (*scm.PullRequest, *scm.Response, error) {
+	path := fmt.Sprintf("repos/%s/pulls", repo)
+	in := &prInput{
+		Title: input.Title,
+		Head:  input.Head,
+		Base:  input.Base,
+		Body:  input.Body,
+	}
+
+	out := new(pr)
+	res, err := s.client.do(ctx, "POST", path, in, out)
+	return convertPullRequest(out), res, err
+}
+
 type prBranch struct {
 	Ref  string     `json:"ref"`
 	Sha  string     `json:"sha"`
@@ -102,6 +116,13 @@ type file struct {
 	PreviousFilename string `json:"previous_filename"`
 }
 
+type prInput struct {
+	Title string `json:"title"`
+	Body  string `json:"body"`
+	Head  string `json:"head"`
+	Base  string `json:"base"`
+}
+
 func convertPullRequestList(from []*pr) []*scm.PullRequest {
 	to := []*scm.PullRequest{}
 	for _, v := range from {

scm/driver/github/pr_test.go

@@ -152,3 +152,40 @@ func TestPullClose(t *testing.T) {
 	t.Run("Request", testRequest(res))
 	t.Run("Rate", testRate(res))
 }
+
+func TestPullCreate(t *testing.T) {
+	defer gock.Off()
+
+	gock.New("https://api.github.com").
+		Post("/repos/octocat/hello-world/pulls").
+		Reply(201).
+		Type("application/json").
+		SetHeaders(mockHeaders).
+		File("testdata/pr_create.json")
+
+	input := &scm.PullRequestInput{
+		Title: "Amazing new feature",
+		Body:  "Please pull these awesome changes in!",
+		Head:  "octocat:new-feature",
+		Base:  "master",
+	}
+
+	client := NewDefault()
+
+	got, res, err := client.PullRequests.Create(context.Background(), "octocat/hello-world", input)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	want := new(scm.PullRequest)
+	raw, err := ioutil.ReadFile("testdata/pr_create.json.golden")
+	json.Unmarshal(raw, want)
+
+	if diff := cmp.Diff(got, want); diff != "" {
+		t.Errorf("Unexpected Results")
+		t.Log(diff)
+	}
+
+	t.Run("Request", testRequest(res))
+	t.Run("Rate", testRate(res))
+}