def dataset():
dataset = DatasetFactory(name='Test Dataset')
dataset.project_id = UUID('6b608f78-e341-422c-8076-35adc8828545')
dataset.uid = UUID("503d7bf6-8e2d-4d29-88af-257af0d4fe4a")
dataset.session = FakeSession()
return dataset
def test_register_dataset_with_existing_id(collection, session):
# Given
name = 'Test Dataset'
summary = 'testing summary'
description = 'testing description'
session.set_response(
DatasetDataFactory(name=name, summary=summary,
description=description))
# When
dataset = DatasetFactory(name=name,
summary=summary,
description=description)
ds_uid = UUID('cafebeef-e341-422c-8076-35adc8828545')
dataset.uid = ds_uid
dataset = collection.register(dataset)
expected_call = FakeCall(method='PUT',
path='projects/{}/datasets/{}'.format(
collection.project_id, ds_uid),
json={
'name': name,
'summary': summary,
'description': description,
'id': str(ds_uid)
})
assert session.num_calls == 1
assert expected_call == session.last_call
assert name == dataset.name
def test_register_dataset_with_idempotent_put(collection, session):
# Given
name = 'Test Dataset'
summary = 'testing summary'
description = 'testing description'
unique_name = 'foo'
session.set_response(
DatasetDataFactory(name=name,
summary=summary,
description=description,
unique_name=unique_name))
# When
session.use_idempotent_dataset_put = True
dataset = collection.register(
DatasetFactory(name=name,
summary=summary,
description=description,
unique_name=unique_name))
expected_call = FakeCall(method='PUT',
path='projects/{}/datasets'.format(
collection.project_id),
json={
'name': name,
'summary': summary,
'description': description,
'unique_name': unique_name
})
assert session.num_calls == 1
assert expected_call == session.last_call
assert name == dataset.name
def test_register_dataset(collection, session):
# Given
name = 'Test Dataset'
summary = 'testing summary'
description = 'testing description'
session.set_response(
DatasetDataFactory(name=name, summary=summary,
description=description))
# When
dataset = collection.register(
DatasetFactory(name=name, summary=summary, description=description))
expected_call = FakeCall(method='POST',
path='projects/{}/datasets'.format(
collection.project_id),
json={
'name': name,
'summary': summary,
'description': description
})
assert session.num_calls == 1
assert expected_call == session.last_call
assert name == dataset.name
def dataset():
dataset = DatasetFactory(name='Test Dataset')
dataset.project_id = UUID('6b608f78-e341-422c-8076-35adc8828545')
dataset.session = None
return dataset