diff --git a/_delphi_utils_python/tests/validator/test_datafetcher.py b/_delphi_utils_python/tests/validator/test_datafetcher.py index 8f75bd6ea..6016df3a2 100644 --- 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)