diff options
Diffstat (limited to 'pygithub3/tests')
| -rw-r--r-- | pygithub3/tests/core/test_client.py | 13 | ||||
| -rw-r--r-- | pygithub3/tests/services/test_core.py | 5 | ||||
| -rw-r--r-- | pygithub3/tests/services/test_users.py | 5 | ||||
| -rw-r--r-- | pygithub3/tests/utils/base.py | 6 |
4 files changed, 24 insertions, 5 deletions
diff --git a/pygithub3/tests/core/test_client.py b/pygithub3/tests/core/test_client.py index b22b1fd..e5b1ea5 100644 --- a/pygithub3/tests/core/test_client.py +++ b/pygithub3/tests/core/test_client.py @@ -4,9 +4,11 @@ from unittest import TestCase import requests +from mock import patch + from pygithub3.core.client import Client from pygithub3.exceptions import NotFound, BadRequest, UnprocessableEntity -from mock import patch +from pygithub3.tests.utils.base import mock_response class TestClient(TestCase): @@ -55,16 +57,25 @@ class TestClient(TestCase): @patch.object(Client, 'request') def test_DELEGATES_methods(self, request_method): + request_method.return_value = mock_response() self.c.get('') request_method.assert_called_with('get', '') + + request_method.return_value = mock_response(201) self.c.post('') request_method.assert_called_with('post', '') + + request_method.return_value = mock_response(200) self.c.patch('') request_method.assert_called_with('patch', '') + self.c.put('') request_method.assert_called_with('put', '') + + request_method.return_value = mock_response(204) self.c.delete('') request_method.assert_called_with('delete', '') + self.c.head('') request_method.assert_called_with('head', '') diff --git a/pygithub3/tests/services/test_core.py b/pygithub3/tests/services/test_core.py index c52b044..cdfe176 100644 --- a/pygithub3/tests/services/test_core.py +++ b/pygithub3/tests/services/test_core.py @@ -8,7 +8,7 @@ import requests from pygithub3.services.base import Base from pygithub3.core.result import Result -from pygithub3.tests.utils.base import DummyRequest +from pygithub3.tests.utils.base import DummyRequest, mock_response from .utils import _ @@ -32,16 +32,19 @@ class TestServiceCalls(TestCase): data=data, params=self.args) def test_DELETE(self, request_method): + request_method.return_value = mock_response(204) self.s._delete(self.r, **self.args) request_method.assert_called_with('delete', _('dummyrequest'), data=None, params=self.args) def test_POST(self, request_method): + request_method.return_value = mock_response(201) self.s._post(self.r, **self.args) request_method.assert_called_with('post', _('dummyrequest'), data=None, params=self.args) def test_GET(self, request_method): + request_method.return_value = mock_response() self.s._get(self.r, **self.args) request_method.assert_called_with('get', _('dummyrequest'), params=self.args) diff --git a/pygithub3/tests/services/test_users.py b/pygithub3/tests/services/test_users.py index f2daa36..9c333ec 100644 --- a/pygithub3/tests/services/test_users.py +++ b/pygithub3/tests/services/test_users.py @@ -8,6 +8,7 @@ from mock import patch, Mock from pygithub3.services.users import User from pygithub3.resources.base import json +from pygithub3.tests.utils.base import mock_response from pygithub3.tests.utils.services import _, mock_json json.dumps = Mock(side_effect=mock_json) @@ -20,8 +21,6 @@ class TestUserService(TestCase): self.us = User() def test_GET_without_user(self, request_method): - response = Mock(name='response') - response.content = {'dummy': 'dummy'} - request_method.return_value = response + request_method.return_value = mock_response() self.us.get() request_method.assert_called_with('get', _('user'), params={}) diff --git a/pygithub3/tests/utils/base.py b/pygithub3/tests/utils/base.py index cda2b2c..571d8ac 100644 --- a/pygithub3/tests/utils/base.py +++ b/pygithub3/tests/utils/base.py @@ -11,6 +11,12 @@ def mock_json(content): return content +def mock_response(status_code=200, content={}): + response = Mock(name='response') + response.status_code = status_code + response.content = content + return response + class DummyResource(Resource): pass |