Mock watchdog observer

mlcommons · Jun 13, 2024 · 3ae2c00 · 3ae2c00
1 parent 20f9f60
commit 3ae2c00
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cli/medperf/tests/commands/dataset/test_prepare.py b/cli/medperf/tests/commands/dataset/test_prepare.py
@@ -8,7 +8,7 @@
 
 from medperf.tests.mocks.dataset import TestDataset
 from medperf.tests.mocks.cube import TestCube
-from medperf.commands.dataset.prepare import DataPreparation, ReportHandler
+from medperf.commands.dataset.prepare import DataPreparation, Observer
 
 PATCH_REGISTER = "medperf.commands.dataset.prepare.{}"
 
@@ -62,8 +62,8 @@ def test_prepare_runs_then_stops_report_handler(
     data_preparation.allow_sending_reports = allow_sending_reports
     mocker.patch.object(cube, "run")
     mocker.patch.object(data_preparation.dataset, "write")
-    mock_handler = mocker.create_autospec(spec=ReportHandler)
-    mocker.patch(PATCH_REGISTER.format("ReportHandler", side_effect=mock_handler))
+    mocked_obs = mocker.create_autospec(spec=Observer)
+    mocker.patch(PATCH_REGISTER.format("Observer", side_effect=mocked_obs))
     gen_report_spy = mocker.patch(PATCH_REGISTER.format("DataPreparation._DataPreparation__generate_report_dict"))
 
     # Act