def test_get_by_id_unknown_fails(self, mocked_repo):
# Given
mocked_repo.return_value = []
requested_id = 'unknown_id'
repo = ProviderRepository()
# Then
with pytest.raises(CatalogError):
repo.get_by_id(requested_id)
def test_get_all_when_empty(self, mocked_repo):
# Given
mocked_repo.return_value = []
repo = ProviderRepository()
# When
providers = repo.get_all()
# Then
mocked_repo.assert_called_once_with(None)
assert providers is None
def test_get_all(self, mocked_repo):
# Given
mocked_repo.return_value = [db_provider1, db_provider2]
repo = ProviderRepository()
# When
providers = repo.get_all()
# Then
mocked_repo.assert_called_once_with(None)
assert isinstance(providers, CatalogList)
assert providers == test_providers
def test_get_by_id(self, mocked_repo):
# Given
mocked_repo.return_value = [db_provider1, db_provider2]
requested_id = db_provider1['id']
repo = ProviderRepository()
# When
provider = repo.get_by_id(requested_id)
# Then
mocked_repo.assert_called_once_with({'id': requested_id})
assert isinstance(provider, Provider)
assert provider == test_provider1
def test_missing_fields_are_mapped_as_None(self, mocked_repo):
# Given
mocked_repo.return_value = [{'id': 'provider1'}]
repo = ProviderRepository()
expected_providers = CatalogList(
[Provider({
'id': 'provider1',
'name': None
})])
# When
providers = repo.get_all()
# Then
assert providers == expected_providers
def test_get_all_only_uses_allowed_filters(self, mocked_repo):
# Given
mocked_repo.return_value = [db_provider1, db_provider2]
repo = ProviderRepository()
filters = {
COUNTRY_FILTER: 'usa',
DATASET_FILTER: 'carto-do.project.census2011',
CATEGORY_FILTER: 'demographics',
VARIABLE_FILTER: 'population',
GEOGRAPHY_FILTER: 'census-geo',
VARIABLE_GROUP_FILTER: 'var-group',
'fake_field_id': 'fake_value'
}
# When
providers = repo.get_all(filters)
# Then
mocked_repo.assert_called_once_with({
COUNTRY_FILTER: 'usa',
CATEGORY_FILTER: 'demographics'
})
assert providers == test_providers