예제 #1
0
파일: client_test.py 프로젝트: yenNSTH/Tron
 def test_request_http_error(self):
     self.mock_urlopen.side_effect = HTTPError(
         self.url, 500, 'broke', {}, build_file_mock('oops'),
     )
     response = client.request(self.url)
     expected = client.Response(500, 'broke', 'oops')
     assert_equal(response, expected)
예제 #2
0
 def test_request_http_error(self, _):
     self.mock_urlopen.side_effect = HTTPError(
         self.url,
         500,
         'broke',
         mock.Mock(get_content_charset=mock.Mock(return_value='utf-8'), ),
         build_file_mock(b'oops'),
     )
     response = client.request(self.url)
     expected = client.Response(500, 'broke', 'oops')
     assert_equal(response, expected)
예제 #3
0
 def test_request_success(self):
     ok_response = {'ok': 'ok'}
     client.request.return_value = client.Response(None, None, ok_response)
     response = self.client.request('/jobs')
     assert_equal(response, ok_response)
예제 #4
0
 def test_request_success(self):
     self.mock_urlopen.return_value = build_file_mock('{"ok": "ok"}')
     response = client.request(self.url)
     expected = client.Response(None, None, {'ok': 'ok'})
     assert_equal(response, expected)
예제 #5
0
 def test_request_url_error(self):
     self.mock_urlopen.side_effect = urllib2.URLError('broke')
     response = client.request(self.url)
     expected = client.Response(client.URL_ERROR, 'broke', None)
     assert_equal(response, expected)