Merge pull request #159 from ndmckinley/add-label

Add a label to the PR.

Author: crstamps2

README.md

@@ -182,6 +182,8 @@ Set the status message for `concourse-ci` context on specified pull request.
 
 * `merge.commit_msg`: *Optional.* Used with `merge` to set the commit message for the merge. Specify a file path to the merge commit message.
 
+* `label`: *Optional.* A label to add to the pull request.
+
 ## Example pipeline
 
 Please see this repo's [pipeline](https://github.com/jtarchie/pullrequest-resource/blob/master/.concourse.yml) for a perfect example.

assets/lib/commands/out.rb

@@ -66,6 +66,11 @@ def output
         metadata << { 'name' => 'comment', 'value' => comment }
       end
 
+      if params.label
+        Octokit.add_labels_to_an_issue(input.source.repo, id, [params.label])
+        metadata << { 'name' => 'label', 'value' => params.label }
+      end
+
       if params.merge.method
         commit_msg = if params.merge.commit_msg
                        commit_path = File.join(destination, params.merge.commit_msg)

spec/commands/out_spec.rb

@@ -195,6 +195,26 @@ def stub_json(method, uri, body)
       end
     end
 
+    context 'when setting a status with a label' do
+      before do
+        stub_request(:post, "https://api.github.com/repos/jtarchie/test/issues/1/labels").with(
+          body: "[\"test_label\"]").to_return(
+            status: 200, body: "", headers: {})
+      end
+      it 'posts a comment to the PR\'s SHA' do
+        stub_status_post
+        stub_json(:post, 'https://api.github.com:443/repos/jtarchie/test/issues/1/comments', id: 1)
+
+        output, = put('params' => { 'status' => 'success', 'path' => 'resource', 'label' => 'test_label' }, 'source' => { 'repo' => 'jtarchie/test' })
+        expect(output).to eq('version'  => { 'ref' => @sha, 'pr' => '1' },
+                             'metadata' => [
+                               { 'name' => 'status', 'value' => 'success' },
+                               { 'name' => 'url', 'value' => 'http://example.com' },
+                               { 'name' => 'label', 'value' => 'test_label' },
+                             ])
+      end
+    end
+
     context 'when the pull request is being release' do
       context 'and the build passed' do
         it 'sets into success mode' do