Update dataset by default (and mock it for tests)

okfn-brasil · May 23, 2017 · efe32b9 · efe32b9
1 parent 5d7f157
commit efe32b9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/rosie/federal_senate/adapter.py b/rosie/federal_senate/adapter.py
@@ -16,11 +16,11 @@ class Adapter:
 
     def __init__(self, path):
         self.path = path
-        self.dataset_path = os.path.join(self.path, 'federal_senate_reimbursements.xz')
 
     @property
     def dataset(self):
-        self._dataset = pd.read_csv(self.dataset_path, dtype={'cnpj_cpf': np.str}, encoding='utf-8')
+        path = self.update_datasets()
+        self._dataset = pd.read_csv(path, dtype={'cnpj_cpf': np.str}, encoding='utf-8')
         self.prepare_dataset()
         return self._dataset
 

diff --git a/rosie/federal_senate/tests/test_adapter.py b/rosie/federal_senate/tests/test_adapter.py
@@ -2,6 +2,7 @@
 import shutil
 from tempfile import mkdtemp
 from unittest import TestCase
+from unittest.mock import patch
 
 import pandas as pd
 
@@ -20,8 +21,9 @@ class TestAdapter(TestCase):
     def setUp(self):
         self.temp_path = mkdtemp()
         subject = subject_class(self.temp_path)
-        subject.dataset_path = FIXTURE_PATH
-        self.dataset = subject.dataset
+        with patch.object(subject_class, 'update_datasets') as mocked_update:
+            mocked_update.return_value = FIXTURE_PATH
+            self.dataset = subject.dataset
 
     def tearDown(self):
         shutil.rmtree(self.temp_path)