def test_dataset_download_without_do_enabled(self, download_stream_mock,
get_by_id_mock, get_all_mock):
# Given
get_by_id_mock.return_value = test_dataset1
dataset = Dataset.get(test_dataset1.id)
get_all_mock.return_value = []
def raise_exception(limit=None,
order_by=None,
sql_query=None,
add_geom=None,
is_geography=None):
raise ServerErrorException(
['The user does not have Data Observatory enabled'])
download_stream_mock.side_effect = raise_exception
credentials = Credentials('fake_user', '1234')
# When
with pytest.raises(Exception) as e:
dataset.to_csv('fake_path', 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_dataset_download_not_subscribed_but_public(self, mocked_bq_client, get_by_id_mock, get_all_mock):
# Given
get_by_id_mock.return_value = test_dataset1 # is public
dataset = Dataset.get(test_dataset1.id)
get_all_mock.return_value = []
mocked_bq_client.return_value = BigQueryClientMock()
credentials = Credentials('fake_user', '1234')
dataset.to_csv('fake_path', credentials)
def test_dataset_download_not_subscribed_but_public(
self, download_stream_mock, get_by_id_mock, get_all_mock):
# Given
get_by_id_mock.return_value = test_dataset1 # is public
dataset = Dataset.get(test_dataset1.id)
get_all_mock.return_value = []
download_stream_mock.return_value = []
credentials = Credentials('fake_user', '1234')
dataset.to_csv('fake_path', credentials)
def test_dataset_download(self, mocked_bq_client, get_by_id_mock, get_all_mock):
# Given
get_by_id_mock.return_value = test_dataset1
dataset = Dataset.get(test_dataset1.id)
get_all_mock.return_value = [dataset]
mocked_bq_client.return_value = BigQueryClientMock()
credentials = Credentials('fake_user', '1234')
# Then
dataset.to_csv('fake_path', credentials)
def test_dataset_download_not_subscribed_but_public(
self, mock_download_stream, mock_get_by_id):
# Given
mock_get_by_id.return_value = test_dataset1 # is public
dataset = Dataset.get(test_dataset1.id)
mock_download_stream.return_value = []
credentials = Credentials('fake_user', '1234')
dataset.to_csv('fake_path', credentials)
os.remove('fake_path')
def test_dataset_download(self, download_stream_mock, get_by_id_mock,
get_all_mock):
# Given
get_by_id_mock.return_value = test_dataset1
dataset = Dataset.get(test_dataset1.id)
get_all_mock.return_value = [dataset]
download_stream_mock.return_value = []
credentials = Credentials('fake_user', '1234')
# Then
dataset.to_csv('fake_path', credentials)
def test_get_dataset_by_id(self, mocked_repo):
# Given
mocked_repo.return_value = test_dataset1
# When
dataset = Dataset.get(test_dataset1.id)
# Then
assert isinstance(dataset, object)
assert isinstance(dataset, Dataset)
assert dataset == test_dataset1
def test_dataset_download(self, mock_download_stream, mock_get_by_id,
mock_subscription_ids):
# Given
mock_get_by_id.return_value = test_dataset1
dataset = Dataset.get(test_dataset1.id)
mock_download_stream.return_value = []
mock_subscription_ids.return_value = [test_dataset1.id]
credentials = Credentials('fake_user', '1234')
# Then
dataset.to_csv('fake_path', credentials)
os.remove('fake_path')
def test_dataset_not_subscribed_download_fails(self, get_by_id_mock,
get_all_mock):
# mock dataset
get_by_id_mock.return_value = test_dataset2 # is private
dataset = Dataset.get(test_dataset2.id)
get_all_mock.return_value = []
credentials = Credentials('fake_user', '1234')
# When
with pytest.raises(Exception) as e:
dataset.to_csv('fake_path', credentials)
# Then
assert str(e.value) == ('You are not subscribed to this Dataset yet. '
'Please, use the subscribe method first.')
def test_dataset_download_without_do_enabled(self, mocked_bq_client, get_by_id_mock, get_all_mock):
# Given
get_by_id_mock.return_value = test_dataset1
dataset = Dataset.get(test_dataset1.id)
get_all_mock.return_value = []
mocked_bq_client.return_value = BigQueryClientMock(
ServerErrorException(['The user does not have Data Observatory enabled'])
)
credentials = Credentials('fake_user', '1234')
# When
with pytest.raises(Exception) as e:
dataset.to_csv('fake_path', 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.')