Merge 'services/pull_requests'

1 files changed, 190 insertions, 0 deletions

diff --git a/pygithub3/tests/services/test_pull_requests.py b/pygithub3/tests/services/test_pull_requests.py
new file mode 100644
index 0000000..93646df
--- /dev/null
+++ b/pygithub3/tests/services/test_pull_requests.py
@@ -0,0 +1,190 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+import requests
+from mock import patch, Mock
+from nose.tools import raises
+
+from pygithub3.tests.utils.core import TestCase
+from pygithub3.services.pull_requests import PullRequests, Comments
+from pygithub3.requests.base import ValidationError, json
+from pygithub3.tests.utils.base import (mock_response, mock_response_result,
+                                        mock_json)
+from pygithub3.tests.utils.services import _
+
+
+json.dumps = Mock(side_effect=mock_json)
+json.loads = Mock(side_effect=mock_json)
+
+
+@patch.object(requests.sessions.Session, 'request')
+class TestPullRequestsService(TestCase):
+    def setUp(self):
+        self.service = PullRequests(user='user', repo='repo')
+
+    def test_LIST(self, reqm):
+        reqm.return_value = mock_response_result()
+        self.service.list().all()
+        self.assertEqual(
+            reqm.call_args[0],
+            ('get', _('repos/user/repo/pulls'))
+        )
+
+    def test_GET(self, reqm):
+        reqm.return_value = mock_response()
+        self.service.get(123)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('get', _('repos/user/repo/pulls/123'))
+        )
+
+    def test_CREATE_with_title_and_body(self, reqm):
+        reqm.return_value = mock_response('post')
+        data = {
+            'title': 'this is a pull request',
+            'body': 'merge me!',
+            'head': 'octocat:some-feature',
+            'base': 'master',
+        }
+        self.service.create(data)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('post', _('repos/user/repo/pulls'))
+        )
+
+    def test_CREATE_with_issue(self, reqm):
+        reqm.return_value = mock_response('post')
+        data = {
+            'issue': 1,
+            'head': 'octocat:some-feature',
+            'base': 'master',
+        }
+        self.service.create(data)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('post', _('repos/user/repo/pulls'))
+        )
+
+    @raises(ValidationError)
+    def test_CREATE_with_no_title(self, reqm):
+        reqm.return_value = mock_response('post')
+        data = {
+            'body': 'merge me!',
+            'head': 'octocat:some-feature',
+            'base': 'master',
+        }
+        self.service.create(data)
+
+    def test_UPDATE(self, reqm):
+        reqm.return_value = mock_response('patch')
+        data = {}
+        self.service.update(123, data)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('patch', _('repos/user/repo/pulls/123'))
+        )
+
+    @raises(ValidationError)
+    def test_UPDATE_with_invalid_state(self, reqm):
+        reqm.return_value = mock_response('patch')
+        data = {'state': 'Illinois'}
+        self.service.update(123, data)
+
+    def test_LIST_COMMITS(self, reqm):
+        reqm.return_value = mock_response_result('get')
+        self.service.list_commits(123).all()
+        self.assertEqual(
+            reqm.call_args[0],
+            ('get', _('repos/user/repo/pulls/123/commits'))
+        )
+
+    def test_LIST_FILES(self, reqm):
+        reqm.return_value = mock_response_result('get')
+        self.service.list_files(123).all()
+        self.assertEqual(
+            reqm.call_args[0],
+            ('get', _('repos/user/repo/pulls/123/files'))
+        )
+
+    def test_IS_MERGED(self, reqm):
+        resp = self.service.is_merged(123)
+        self.assertTrue(resp)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('head', _('repos/user/repo/pulls/123/merge'))
+        )
+
+    def test_MERGE(self, reqm):
+        reqm.return_value = mock_response(200)
+        self.service.merge(123, 'merging this')
+        self.assertEqual(
+            reqm.call_args[0],
+            ('put', _('repos/user/repo/pulls/123/merge'))
+        )
+
+
+@patch.object(requests.sessions.Session, 'request')
+class TestPullRequestCommentsService(TestCase):
+    def setUp(self):
+        self.service = Comments(user='user', repo='repo')
+
+    def test_LIST(self, reqm):
+        reqm.return_value = mock_response_result(200)
+        self.service.list(123).all()
+        self.assertEqual(
+            reqm.call_args[0],
+            ('get', _('repos/user/repo/pulls/123/comments'))
+        )
+
+    def test_GET(self, reqm):
+        reqm.return_value = mock_response(200)
+        self.service.get(1)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('get', _('repos/user/repo/pulls/comments/1'))
+        )
+
+    def test_CREATE(self, reqm):
+        reqm.return_value = mock_response(201)
+        data = {
+            'body': ':sparkles:',
+            'commit_id': 'abc123',
+            'path': 'foo.txt',
+            'position': '2',
+        }
+        self.service.create(1, data)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('post', _('repos/user/repo/pulls/1/comments'))
+        )
+
+    def test_CREATE_in_reply_to(self, reqm):
+        reqm.return_value = mock_response(201)
+        data = {
+            'body': ':sparkles:',
+            'in_reply_to': '5',
+        }
+        self.service.create(1, data)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('post', _('repos/user/repo/pulls/1/comments'))
+        )
+
+    def test_EDIT(self, reqm):
+        reqm.return_value = mock_response(200)
+        data = {
+            'body': 'something completely different',
+        }
+        self.service.edit(1, data)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('patch', _('repos/user/repo/pulls/comments/1'))
+        )
+
+    def test_DELETE(self, reqm):
+        reqm.return_value = mock_response(204)
+        self.service.delete(1)
+        self.assertEqual(
+            reqm.call_args[0],
+            ('delete', _('repos/user/repo/pulls/comments/1'))
+        )