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()