From d3fed76f5aeef1e4f0907b4072badcde809c8c8f Mon Sep 17 00:00:00 2001 From: Beatriz Rizental Date: Tue, 8 Aug 2017 09:43:11 -0300 Subject: [PATCH] Fixes #1618 - Fixes some style inconsistencies. --- tests/test_api_urls.py | 24 +++++++++++++----------- 1 file changed, 13 insertions(+), 11 deletions(-) diff --git a/tests/test_api_urls.py b/tests/test_api_urls.py index 8a45020ef..646c5d139 100644 --- a/tests/test_api_urls.py +++ b/tests/test_api_urls.py @@ -10,6 +10,7 @@ import unittest from mock import patch, MagicMock +from mock import MagicMock from requests import Response import webcompat @@ -21,16 +22,16 @@ 'HTTP_ACCEPT': 'application/json'} -def mock_api_response(dict): +def mock_api_response(response_config={}): ''' Helper method to create a mock response from the Github API.''' api_response = MagicMock(spec=Response) api_response.content_type = 'application/json' api_response.headers = { - 'etag': 'thisisatest', - 'cache-control': 'thisisatest', - 'content-type': 'application/json' + 'ETag': 'W/"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"', + 'Cache-Control': 'public, max-age=60, s-maxage=60', + 'Content-Type': 'application/json; charset=utf-8' } - for k, v in dict.iteritems(): + for k, v in response_config.iteritems(): setattr(api_response, k, v) return api_response @@ -56,7 +57,7 @@ def test_api_issues_out_of_range(self): with patch('webcompat.helpers.proxy_request') as github_data: github_data.return_value = mock_api_response({ 'status_code': 404, - 'content': '{"message":"Not Found","documentation_url":"https://developer.github.com/v3"}' # nopep8 + 'content': '[{"message":"Not Found","documentation_url":"https://developer.github.com/v3"}]' # nopep8 }) rv = self.app.get('/api/issues/1', environ_base=headers) json_body = json.loads(rv.data) @@ -84,7 +85,7 @@ def test_api_labels_without_auth(self): '''API access to labels without auth returns JSON 200.''' with patch('webcompat.helpers.proxy_request') as github_data: github_data.return_value = mock_api_response( - {'status_code': 200, 'content': '{}'}) + {'status_code': 200, 'content': '[]'}) rv = self.app.get('/api/issues/labels', environ_base=headers) self.assertEqual(rv.status_code, 200) self.assertEqual(rv.content_type, 'application/json') @@ -97,15 +98,17 @@ def test_api_comments_link_header_auth(self): github_data.side_effect = [ mock_api_response({ 'status_code': 200, - 'content': '{"Link":[["https://api.github.com/repositories/17839063/issues/398/comments?callback=foo&page=2",{"rel":"next"}],["https://api.github.com/repositories/17839063/issues/398/comments?callback=foo&page=4",{"rel":"last"}]]}' # nopep8 + 'content': '[{"Link":[["https://api.github.com/repositories/17839063/issues/398/comments?callback=foo&page=2",{"rel":"next"}],["https://api.github.com/repositories/17839063/issues/398/comments?callback=foo&page=4",{"rel":"last"}]]}]' # nopep8 }), - mock_api_response({'status_code': 200, 'content': '{}'}) + mock_api_response({'status_code': 200, 'content': '[]'}) ] # Force a JSONP callback response with `query_string` # because it gives us the header properties we want to test. query_string = {'callback': 'foo'} rv = self.app.get('/api/issues/398/comments', query_string=query_string, environ_base=headers) + # Make sure that rv.data is not empty. + self.assertTrue(len(rv.data) > 0) self.assertTrue = all(x in rv.data for x in ['Link', 'rel', 'next', 'last', 'page']) self.assertEqual(rv.status_code, 200) @@ -116,8 +119,7 @@ def test_api_comments_link_header_auth(self): rv = self.app.get('/api/issues/4/comments', query_string=query_string, environ_base=headers) self.assertTrue = not all( - x in rv.data for x in [ - 'Link', 'rel', 'next', 'last', 'page']) + x in rv.data for x in ['Link', 'rel', 'next', 'last', 'page']) self.assertEqual(rv.status_code, 200) self.assertEqual(rv.content_type, 'application/json')