def test_get_request_headers(self):
rc = RequestsClient(self.retries, self.timeout, self.backoff_factor)
platform_info = {'troll': 'goat'}
magic_admin.api_secret_key = 'magic_secret_key'
with mock.patch.object(
rc,
'_get_platform_info',
return_value=platform_info,
) as mock_get_platform_info:
assert rc._get_request_headers() == {
'X-Magic-Secret-Key': magic_admin.api_secret_key,
'User-Agent': json.dumps({
'language': 'python',
'sdk_version': version.VERSION,
'publisher': 'magic',
'http_lib': rc.__class__.__name__,
**platform_info,
}),
}
mock_get_platform_info.assert_called_once_with()
def test_get_request_headers_raises_error(self):
rc = RequestsClient(self.retries, self.timeout, self.backoff_factor)
magic_admin.api_secret_key = None
with pytest.raises(AuthenticationError):
rc._get_request_headers()