Rename mocking queue functions with scheduler

equinor · Nov 28, 2023 · 7d7e295 · 7d7e295
1 parent ff3a55b
commit 7d7e295
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/unit_tests/ensemble_evaluator/conftest.py b/tests/unit_tests/ensemble_evaluator/conftest.py
@@ -65,7 +65,7 @@ def queue_config_fixture():
 def make_ensemble_builder(queue_config):
     def _make_ensemble_builder(monkeypatch, tmpdir, num_reals, num_jobs, job_sleep=0):
         monkeypatch.setattr(
-            "ert.scheduler.queue.forward_model_ok",
+            "ert.scheduler.scheduler.forward_model_ok",
             AsyncMock(return_value=(LoadStatus.LOAD_SUCCESSFUL, "")),
         )
         builder = ert.ensemble_evaluator.EnsembleBuilder()

diff --git a/tests/unit_tests/job_queue/conftest.py b/tests/unit_tests/job_queue/conftest.py
@@ -11,7 +11,7 @@
 @pytest.fixture
 def mock_fm_ok(monkeypatch):
     fm_ok = AsyncMock(return_value=(LoadStatus.LOAD_SUCCESSFUL, ""))
-    monkeypatch.setattr(ert.scheduler.queue, "forward_model_ok", fm_ok)
+    monkeypatch.setattr(ert.scheduler.scheduler, "forward_model_ok", fm_ok)
     yield fm_ok
 
 

diff --git a/tests/unit_tests/job_queue/test_job_queue.py b/tests/unit_tests/job_queue/test_job_queue.py
@@ -188,7 +188,7 @@ async def test_run_done_callback(
 ):
     monkeypatch.chdir(tmpdir)
     monkeypatch.setattr(
-        "ert.scheduler.queue.forward_model_ok",
+        "ert.scheduler.scheduler.forward_model_ok",
         AsyncMock(return_value=(loadstatus, "foo")),
     )
     scheduler = create_local_queue(simple_script)