def test_https_verify(self, mock_requests):
url = "https://127.0.0.1:8888"
client = HTTPClient(url=url, verify=True)
mock_result = MockResult()
mock_result.text = "foo bar ponies"
mock_result.headers = {"Content-Type": "text/html"}
mock_result.status_code = 200
mock_requests.request.return_value = mock_result
client.run()
self.assertTrue(client.verify)
mock_requests.request.assert_called_with(
"GET",
url,
allow_redirects=False,
auth=None,
cookies=None,
data="",
files=None,
headers={},
params=None,
proxies=None,
timeout=60,
verify=True,
)
def test_https_auth_basic(self, mock_requests):
url = "https://127.0.0.1:8888"
username = "******"
password = "******"
client = HTTPClient(url=url, username=username, password=password)
mock_result = MockResult()
mock_result.text = "muppet show"
mock_result.headers = {"Authorization": "bWlzc3BpZ2d5Omtlcm1pdA=="}
mock_result.status_code = 200
mock_requests.request.return_value = mock_result
result = client.run()
self.assertEqual(result["headers"], mock_result.headers)
mock_requests.request.assert_called_once_with(
"GET",
url,
allow_redirects=False,
auth=client.auth,
cookies=None,
data="",
files=None,
headers={},
params=None,
proxies=None,
timeout=60,
verify=False,
)
def test_https_verify(self, mock_requests):
url = 'https://127.0.0.1:8888'
client = HTTPClient(url=url, verify=True)
mock_result = MockResult()
mock_result.text = 'foo bar ponies'
mock_result.headers = {'Content-Type': 'text/html'}
mock_result.status_code = 200
mock_requests.request.return_value = mock_result
client.run()
self.assertTrue(client.verify)
mock_requests.request.assert_called_with('GET',
url,
allow_redirects=False,
auth=None,
cookies=None,
data='',
files=None,
headers={},
params=None,
proxies=None,
timeout=60,
verify=True)
def test_https_auth_basic(self, mock_requests):
url = 'https://127.0.0.1:8888'
username = '******'
password = '******'
client = HTTPClient(url=url, username=username, password=password)
mock_result = MockResult()
mock_result.text = 'muppet show'
mock_result.headers = {'Authorization': 'bWlzc3BpZ2d5Omtlcm1pdA=='}
mock_result.status_code = 200
mock_requests.request.return_value = mock_result
result = client.run()
self.assertEqual(result['headers'], mock_result.headers)
mock_requests.request.assert_called_once_with('GET',
url,
allow_redirects=False,
auth=client.auth,
cookies=None,
data='',
files=None,
headers={},
params=None,
proxies=None,
timeout=60,
verify=False)
def test_https_verify_false(self, mock_requests):
url = 'https://127.0.0.1:8888'
client = HTTPClient(url=url)
mock_result = MockResult()
mock_result.text = 'foo bar ponies'
mock_result.headers = {'Content-Type': 'text/html'}
mock_result.status_code = 200
mock_requests.request.return_value = mock_result
client.run()
self.assertFalse(client.verify)
mock_requests.request.assert_called_with(
'GET', url, allow_redirects=False, auth=None, cookies=None,
data='', files=None, headers={}, params=None, proxies=None,
timeout=60, verify=False)
def test_parse_response_body(self, mock_requests):
client = HTTPClient(url='http://127.0.0.1')
mock_result = MockResult()
# Unknown content type, body should be returned raw
mock_result.text = 'foo bar ponies'
mock_result.headers = {'Content-Type': 'text/html'}
mock_result.status_code = 200
mock_requests.request.return_value = mock_result
result = client.run()
self.assertEqual(result['body'], mock_result.text)
self.assertEqual(result['status_code'], mock_result.status_code)
self.assertEqual(result['headers'], mock_result.headers)
# Unknown content type, JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {'Content-Type': 'text/html'}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertEqual(result['body'], mock_result.text)
# JSON content-type and JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {'Content-Type': 'application/json'}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertTrue(isinstance(result['body'], dict))
self.assertEqual(result['body'], {'test1': 'val1'})
# JSON content-type with charset and JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {
'Content-Type': 'application/json; charset=UTF-8'
}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertTrue(isinstance(result['body'], dict))
self.assertEqual(result['body'], {'test1': 'val1'})
# JSON content-type and invalid json body
mock_result.text = 'not json'
mock_result.headers = {'Content-Type': 'application/json'}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertFalse(isinstance(result['body'], dict))
self.assertEqual(result['body'], mock_result.text)
def test_parse_response_body(self, mock_requests):
client = HTTPClient(url="http://127.0.0.1")
mock_result = MockResult()
# Unknown content type, body should be returned raw
mock_result.text = "foo bar ponies"
mock_result.headers = {"Content-Type": "text/html"}
mock_result.status_code = 200
mock_requests.request.return_value = mock_result
result = client.run()
self.assertEqual(result["body"], mock_result.text)
self.assertEqual(result["status_code"], mock_result.status_code)
self.assertEqual(result["headers"], mock_result.headers)
# Unknown content type, JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {"Content-Type": "text/html"}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertEqual(result["body"], mock_result.text)
# JSON content-type and JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {"Content-Type": "application/json"}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertTrue(isinstance(result["body"], dict))
self.assertEqual(result["body"], {"test1": "val1"})
# JSON content-type with charset and JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {"Content-Type": "application/json; charset=UTF-8"}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertTrue(isinstance(result["body"], dict))
self.assertEqual(result["body"], {"test1": "val1"})
# JSON content-type and invalid json body
mock_result.text = "not json"
mock_result.headers = {"Content-Type": "application/json"}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertFalse(isinstance(result["body"], dict))
self.assertEqual(result["body"], mock_result.text)
def test_parse_response_body(self, mock_requests):
client = HTTPClient(url='http://127.0.0.1')
mock_result = MockResult()
# Unknown content type, body should be returned raw
mock_result.text = 'foo bar ponies'
mock_result.headers = {'Content-Type': 'text/html'}
mock_result.status_code = 200
mock_requests.request.return_value = mock_result
result = client.run()
self.assertEqual(result['body'], mock_result.text)
self.assertEqual(result['status_code'], mock_result.status_code)
self.assertEqual(result['headers'], mock_result.headers)
# Unknown content type, JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {'Content-Type': 'text/html'}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertEqual(result['body'], mock_result.text)
# JSON content-type and JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {'Content-Type': 'application/json'}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertTrue(isinstance(result['body'], dict))
self.assertEqual(result['body'], {'test1': 'val1'})
# JSON content-type with charset and JSON body
mock_result.text = '{"test1": "val1"}'
mock_result.headers = {'Content-Type': 'application/json; charset=UTF-8'}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertTrue(isinstance(result['body'], dict))
self.assertEqual(result['body'], {'test1': 'val1'})
# JSON content-type and invalid json body
mock_result.text = 'not json'
mock_result.headers = {'Content-Type': 'application/json'}
mock_requests.request.return_value = mock_result
result = client.run()
self.assertFalse(isinstance(result['body'], dict))
self.assertEqual(result['body'], mock_result.text)
