fix(test): test_vaccine imunizacao url is forbidden to be fetched on CI

Author: luabida

.github/workflows/python-package.yml

@@ -33,14 +33,12 @@ jobs:
 
     - name: Run jupyterlab with PySUS
       run: |
-        # make run-jupyter-pysus
+        make run-jupyter-pysus
         # make test-jupyter-pysus  ## takes too long
 
     - name: Linting & Tests
       run: |
+        export CI=1
         poetry install
         pre-commit run --all-files
-        # make test-pysus
-
-    - name: Setup tmate session
-      uses: mxschmitt/action-tmate@v3
+        make test-pysus

docker/docker-compose.yaml

@@ -9,6 +9,7 @@ services:
     privileged: true
     environment:
       - DISPLAY=:0
+      - CI=${CI:-0}
     volumes:
       - /tmp/.X11-unix:/tmp/.X11-unix
     entrypoint: ["/entrypoint.sh"]

pysus/online_data/vaccine.py

@@ -107,8 +107,6 @@ def elasticsearch_fetch(uri, auth, json_body={}):
             print(resp)
         total += len(resp["hits"]["hits"])
         print(f"Downloaded {total} records\r", end="")
-        # print(resp)
-        # print(uri)
         yield [h["_source"] for h in resp["hits"]["hits"]]
         if "_scroll_id" in resp:
             scroll_id = resp["_scroll_id"]

pysus/tests/test_data/test_vaccine.py

@@ -1,3 +1,4 @@
+import os
 import unittest
 
 import pandas as pd
@@ -7,8 +8,8 @@
 
 class VaccineTestCase(unittest.TestCase):
     @pytest.mark.timeout(15)
+    @unittest.skipIf(os.getenv("CI"), "Forbidden on CI")
     def test_Download(self):
-        """Careful! this download can take a long time"""
         df = download_covid("BA", only_header=True)
         self.assertIsInstance(df, pd.DataFrame)
         self.assertEqual(df.shape, (10000, 42))