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