Skip to content
Snippets Groups Projects
Commit 537fc8fa authored by Conor Finn's avatar Conor Finn
Browse files

RE #27779 Fix data handling tests

Should have fixed these as I went along, but I'm not sure where they
failed to fixup the commits. Most were introduced as a result of
handling the checking of the plot box when repopulating the table.
parent a9b4ab32
No related branches found
No related tags found
No related merge requests found
......@@ -73,6 +73,7 @@ class FittingDataPresenterTest(unittest.TestCase):
model_dict = {"ENGINX_1_bank_1": "ws1", "ENGINX_2_bank_South": "ws2"}
self.model.get_loaded_workspaces.return_value = model_dict
self.model.get_sample_log_from_ws.return_value = "bankOrRunNumber"
self.presenter.plot_added_notifier = mock.MagicMock()
self.presenter._on_worker_success("info")
......@@ -85,6 +86,7 @@ class FittingDataPresenterTest(unittest.TestCase):
model_dict = {"invalid": "ws1", "invalid2": "ws2"}
self.model.get_loaded_workspaces.return_value = model_dict
self.model.get_sample_log_from_ws.side_effect = RuntimeError("No sample logs present")
self.presenter.plot_added_notifier = mock.MagicMock()
self.presenter._on_worker_success("info")
......@@ -98,6 +100,7 @@ class FittingDataPresenterTest(unittest.TestCase):
model_dict = {"INSTRUMENT_10_bank_2": "ws1", "INSTRUMENT_20_bank_1": "ws2"}
self.model.get_loaded_workspaces.return_value = model_dict
self.model.get_sample_log_from_ws.side_effect = RuntimeError("No sample logs present")
self.presenter.plot_added_notifier = mock.MagicMock()
self.presenter._on_worker_success("info")
......@@ -133,12 +136,14 @@ class FittingDataPresenterTest(unittest.TestCase):
model_dict = {"name1": "ws1", "name2": "ws2"}
self.model.get_loaded_workspaces.return_value = model_dict
self.presenter.row_numbers = {"name1": 0, "name2": 1}
self.presenter.all_plots_removed_notifier = mock.MagicMock()
self.presenter.rename_workspace("name1", "new")
self.assertEqual({"new": "ws1", "name2": "ws2"}, model_dict)
self.assertTrue("new" in self.presenter.row_numbers)
self.assertFalse("name1" is self.presenter.row_numbers)
self.assertEqual(1, self.presenter.all_plots_removed_notifier.notify_subscribers.call_count)
def test_rename_workspace_not_tracked(self):
model_dict = {"name1": "ws1", "name2": "ws2"}
......@@ -192,6 +197,7 @@ class FittingDataPresenterTest(unittest.TestCase):
self.model.get_loaded_workspaces.return_value = model_dict
self.view.remove_selected.return_value = [0, 2]
self.presenter.plot_removed_notifier = mock.MagicMock()
self.presenter.plot_added_notifier = mock.MagicMock()
self.presenter._remove_selected_tracked_workspaces()
......@@ -201,6 +207,7 @@ class FittingDataPresenterTest(unittest.TestCase):
self.assertEqual(self.presenter.row_numbers, test_dict)
self.assertEqual(model_dict, {"name2": "ws2"})
self.assertEqual(2, self.presenter.plot_removed_notifier.notify_subscribers.call_count)
self.assertEqual(1, self.presenter.plot_added_notifier.notify_subscribers.call_count)
def test_handle_table_cell_changed_checkbox_ticked(self):
mocked_table_item = mock.MagicMock()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment