def test_get_all_datasets_credentials_without_do_enabled(self, mocked_repo):
# Given
def raise_exception(a, b):
raise ServerErrorException(['The user does not have Data Observatory enabled'])
mocked_repo.side_effect = raise_exception
credentials = Credentials('fake_user', '1234')
# When
with pytest.raises(Exception) as e:
Dataset.get_all(credentials=credentials)
# Then
assert str(e.value) == (
'We are sorry, the Data Observatory is not enabled for your account yet. '
'Please contact your customer success manager or send an email to '
'[email protected] to request access to it.')
def test_get_all_datasets(self, mocked_repo):
# Given
mocked_repo.return_value = test_datasets
# When
datasets = Dataset.get_all()
# Then
assert isinstance(datasets, list)
assert isinstance(datasets, CatalogList)
assert datasets == test_datasets
def test_get_all_datasets_credentials(self, mocked_repo):
# Given
mocked_repo.return_value = test_datasets
credentials = Credentials('fake_user', '1234')
# When
datasets = Dataset.get_all(credentials=credentials)
# Then
mocked_repo.assert_called_once_with(None, credentials)
assert isinstance(datasets, list)
assert isinstance(datasets, CatalogList)
assert datasets == test_datasets