diff --git a/tests/api/test_search.py b/tests/api/test_search.py index cd2433bc4..01425fb76 100644 --- a/tests/api/test_search.py +++ b/tests/api/test_search.py @@ -21,6 +21,7 @@ def patch_get_map_ateam_curies_to_names(): mock_func.return_value = {'ATP:0000196': 'antibody extraction complete'} yield + @pytest.fixture(scope="module", autouse=True) def patch_get_parent_or_children(): with patch("agr_literature_service.api.crud.search_crud.get_parent_or_children") as mock_func: diff --git a/tests/api/test_search_dates.py b/tests/api/test_search_dates.py index 47b2a1c5b..de55081d7 100644 --- a/tests/api/test_search_dates.py +++ b/tests/api/test_search_dates.py @@ -39,6 +39,13 @@ def patch_get_map_ateam_curies_to_names(): yield +@pytest.fixture(scope="module", autouse=True) +def patch_get_parent_or_children(): + with patch("agr_literature_service.api.crud.search_crud.get_parent_or_children") as mock_func: + mock_func.return_value = ["ATP:0000141", "ATP:0000134", "ATP:0000139", "ATP:0000135"] + yield + + @pytest.fixture(scope='module') def initialize_elasticsearch(): print("***** Initializing Elasticsearch Data *****")