Skip to content

Commit 9008296

Browse files
author
Nat Williams
committed
use _bool for pull request merge status
1 parent efee843 commit 9008296

File tree

2 files changed

+6
-19
lines changed

2 files changed

+6
-19
lines changed

pygithub3/services/pull_requests/__init__.py

Lines changed: 4 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -95,23 +95,10 @@ def merge_status(self, number, user=None, repo=None):
9595
:param str repo: Repository
9696
9797
"""
98-
# for this to work with a proper Resource, we would need to pass the
99-
# response's status code to the Resource constructor, and that's kind
100-
# of scary
101-
try:
102-
resp = self._client.get(
103-
self.make_request('pull_requests.merge_status', number=number,
104-
user=user, repo=repo)
105-
)
106-
except NotFound:
107-
return False
108-
code = resp.status_code
109-
if code == 204:
110-
return True
111-
# TODO: more flexible way to return arbitrary objects based on
112-
# response. Probably something on Request
113-
raise BadRequest('got code %s: %s' % (code, resp.content))
114-
# again, I'm sorry.
98+
return self._bool(
99+
self.make_request('pull_requests.merge_status', number=number,
100+
user=user, repo=repo)
101+
)
115102

116103
def merge(self, number, message='', user=None, repo=None):
117104
"""Merge a pull request.

pygithub3/tests/services/test_pull_requests.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def test_MERGE_STATUS_true(self, reqm):
113113
self.assertEqual(True, resp)
114114
self.assertEqual(
115115
reqm.call_args[0],
116-
('get', _('repos/user/repo/pulls/123/merge'))
116+
('head', _('repos/user/repo/pulls/123/merge'))
117117
)
118118

119119
def test_MERGE_STATUS_false(self, reqm):
@@ -122,7 +122,7 @@ def test_MERGE_STATUS_false(self, reqm):
122122
self.assertEqual(False, resp)
123123
self.assertEqual(
124124
reqm.call_args[0],
125-
('get', _('repos/user/repo/pulls/123/merge'))
125+
('head', _('repos/user/repo/pulls/123/merge'))
126126
)
127127

128128
def test_MERGE(self, reqm):

0 commit comments

Comments
 (0)