Merge pull request #1477 from cmu-delphi/fix_datafetcher_mock

Fix requests.get mock
cmu-delphi · Jan 20, 2022 · 6ef4968 · 6ef4968
2 parents 88476b1 + e8dac87
commit 6ef4968
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/_delphi_utils_python/tests/validator/test_datafetcher.py b/_delphi_utils_python/tests/validator/test_datafetcher.py
@@ -31,7 +31,10 @@ def __init__(self, json_data, status_code):
 
             def json(self):
                 return self.json_data
-        if kwargs["params"] == {'signal':'chng:inactive'}:
+        if len(kwargs) == 0:
+            return MockResponse([{'source': 'chng', 'db_source': 'chng'},
+                {'source': 'covid-act-now', 'db_source': 'covid-act-now'}], 200)
+        elif kwargs["params"] == {'signal': 'chng:inactive'}:
             return MockResponse([{"signals": [{"active": False}]}], 200)
         else:
             return MockResponse([{"signals": [{"active": True}]}], 200)