def test_post_blocks_raises_api_error(mock_requests_post, mock_get_api_key):
with fake_creds():
# setUp
creds = {
"WEBAPP_URL": "https://staging.jovian.ml/",
"GUEST_KEY": "b6538d4dfde04fcf993463a828a9cec6",
"API_URL": "https://api-staging.jovian.ai",
"API_KEY": "fake_invalid_api_key",
"ORG_ID": "staging",
}
write_creds(creds)
blocks = [{"data": {"key": "value"}, "record_type": "metrics"}]
with pytest.raises(ApiError) as context:
post_blocks(blocks)
mock_requests_post.assert_called_with(
"https://api-staging.jovian.ai/data/record",
data=None,
headers={
"Authorization": "Bearer fake_invalid_api_key",
"x-jovian-source": "library",
"x-jovian-library-version": __version__,
"x-jovian-guest": "b6538d4dfde04fcf993463a828a9cec6",
"x-jovian-org": "staging",
},
json=[{"data": {"key": "value"}, "record_type": "metrics"}],
)
assert (
str(context.value)
== "Data logging failed: (HTTP 500) Internal Server Error"
)
def test_post_blocks(mock_requests_post):
with fake_creds():
blocks = [{"data": {"key": "value"}, "record_type": "metrics"}]
post_blocks(blocks)
mock_requests_post.assert_called_with(
"https://api-staging.jovian.ai/data/record",
data=None,
headers={
"Authorization": "Bearer fake_api_key",
"x-jovian-source": "library",
"x-jovian-library-version": __version__,
"x-jovian-guest": "b6538d4dfde04fcf993463a828a9cec6",
"x-jovian-org": "staging",
},
json=[{"data": {"key": "value"}, "record_type": "metrics"}],
)
def test_post_blocks(self, mock_requests_post):
with fake_creds('.jovian', 'credentials.json'):
blocks = [{'data': {'key': 'value'}, 'record_type': 'metrics'}]
post_blocks(blocks)
mock_requests_post.assert_called_with(
'https://api-staging.jovian.ai/data/record',
data=None,
headers={
"Authorization": "Bearer fake_api_key",
"x-jovian-source": "library",
"x-jovian-library-version": __version__,
"x-jovian-guest": "b6538d4dfde04fcf993463a828a9cec6",
"x-jovian-org": "staging"
},
json=[{
'data': {
'key': 'value'
},
'record_type': 'metrics'
}])
def test_post_blocks_raises_api_error(self, mock_requests_post,
mock_get_api_key):
with fake_creds('.jovian-invalid-key', 'credentials.json'):
# setUp
creds = {
"WEBAPP_URL": "https://staging.jovian.ml/",
"GUEST_KEY": "b6538d4dfde04fcf993463a828a9cec6",
"API_URL": "https://api-staging.jovian.ai",
"API_KEY": "fake_invalid_api_key",
"ORG_ID": "staging"
}
write_creds(creds)
blocks = [{'data': {'key': 'value'}, 'record_type': 'metrics'}]
with self.assertRaises(ApiError) as context:
post_blocks(blocks)
mock_requests_post.assert_called_with(
'https://api-staging.jovian.ai/data/record',
data=None,
headers={
"Authorization": "Bearer fake_invalid_api_key",
"x-jovian-source": "library",
"x-jovian-library-version": __version__,
"x-jovian-guest": "b6538d4dfde04fcf993463a828a9cec6",
"x-jovian-org": "staging"
},
json=[{
'data': {
'key': 'value'
},
'record_type': 'metrics'
}])
assert context.exception.args[
0] == 'Data logging failed: (HTTP 500) Internal Server Error'