def test__get_headers_Should_ReturnHeaders_When_Called(self, *patches):
client = RESTclient('hostname1.company.com')
result = client.get_headers()
expected_result = {
'Content-Type': 'application/json',
}
self.assertEqual(result, expected_result)
def test__get_headers_Should_ReturnHeaders_When_ApiKey(self, *patches):
client = RESTclient('hostname1.company.com', api_key='some-api-key')
result = client.get_headers()
expected_result = {
'Content-Type': 'application/json',
'x-api-key': 'some-api-key'
}
self.assertEqual(result, expected_result)
def test__get_headers_Should_ReturnHeaders_When_BearerToken(
self, *patches):
client = RESTclient('hostname1.company.com', bearer_token='token')
result = client.get_headers()
expected_result = {
'Content-Type': 'application/json',
'Authorization': 'Bearer token'
}
self.assertEqual(result, expected_result)
def test__get_headers_Should_SetAuthorizationHeader_When_UsernamePasswordAttributesExist(
self, *patches):
client = RESTclient('hostname', username='******', password='******')
results = client.get_headers()
self.assertTrue('Authorization' in results)
self.assertTrue('Basic' in results['Authorization'])