From 8c3e123e0a232b5e8dfb9f405711f2bcf78ff33d Mon Sep 17 00:00:00 2001 From: David Medina Date: Sun, 19 Feb 2012 19:41:45 +0100 Subject: Support to Mimetypes in some services --- pygithub3/tests/services/test_core.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'pygithub3/tests/services') diff --git a/pygithub3/tests/services/test_core.py b/pygithub3/tests/services/test_core.py index 007f94f..612b5ea 100644 --- a/pygithub3/tests/services/test_core.py +++ b/pygithub3/tests/services/test_core.py @@ -53,3 +53,24 @@ class TestServiceCalls(TestCase): result = self.s._get_result(self.r, **self.args) self.assertFalse(request_method.called) self.assertIsInstance(result, Result) + + +@patch.object(requests.sessions.Session, 'request') +class TestMimeType(TestCase): + + def setUp(self): + self.ms = DummyService() + + def test_WITHOUT_mimetype(self, request_method): + request_method.return_value = mock_response() + self.ms.dummy_request() + request_method.assert_called_with('get', _('dummyrequest'), params={}) + + def test_WITH_mimetype(self, request_method): + request_method.return_value = mock_response() + self.ms.set_html_mimetype() + self.ms.dummy_request() + request_method.assert_called_with('get', _('dummyrequest'), + headers={'Accept': 'application/vnd.github.%s.html+json' % + MimeTypeMixin.VERSION}, + params={}) -- cgit v1.2.3-59-g8ed1b