Merge pull request #1415 from CartoDB/fix/catalog-provider-geo-filter

Geographies filtered by provider
CartoDB · Dec 30, 2019 · 9a4935d · 9a4935d
2 parents d0ebaf5 + 6afbba7
commit 9a4935d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cartoframes/data/observatory/catalog/repository/geography_repo.py b/cartoframes/data/observatory/catalog/repository/geography_repo.py
@@ -1,14 +1,14 @@
 from cartoframes.auth import Credentials
 
-from .constants import COUNTRY_FILTER, CATEGORY_FILTER
+from .constants import COUNTRY_FILTER, CATEGORY_FILTER, PROVIDER_FILTER
 from .entity_repo import EntityRepository
 
 from .....io.carto import read_carto
 
 
 _GEOGRAPHY_ID_FIELD = 'id'
 _GEOGRAPHY_SLUG_FIELD = 'slug'
-_ALLOWED_FILTERS = [COUNTRY_FILTER, CATEGORY_FILTER]
+_ALLOWED_FILTERS = [COUNTRY_FILTER, CATEGORY_FILTER, PROVIDER_FILTER]
 
 _DO_CREDENTIALS = Credentials('do-metadata', 'default_public')
 

diff --git a/tests/unit/data/observatory/catalog/repository/test_geography_repo.py b/tests/unit/data/observatory/catalog/repository/test_geography_repo.py
@@ -79,7 +79,8 @@ def test_get_all_only_uses_allowed_filters(self, mocked_repo):
         # Then
         mocked_repo.assert_called_once_with({
             'country_id': 'usa',
-            'category_id': 'demographics'
+            'category_id': 'demographics',
+            'provider_id': 'open_data'
         })
         assert geographies == test_geographies