Return tests, mock out other instantiations of MainWindowView

Author: Dimitar Tasev

mantidimaging/gui/windows/main/test/view_test.py

@@ -23,7 +23,8 @@
 class MainWindowViewTest(unittest.TestCase):
     def setUp(self) -> None:
         with mock.patch("mantidimaging.gui.windows.main.view.WelcomeScreenPresenter"):
-            self.view = MainWindowView()
+            with mock.patch("mantidimaging.gui.windows.main.view.has_other_shared_arrays", return_value=False):
+                self.view = MainWindowView()
         self.presenter = mock.MagicMock()
         self.view.presenter = self.presenter
 
@@ -306,3 +307,34 @@ def test_get_images_from_stack_uuid(self):
 
         self.presenter.get_stack_visualiser.assert_called_once_with(uuid)
         self.assertEqual(images, return_value)
+
+    @mock.patch("mantidimaging.gui.windows.main.view.has_other_shared_arrays")
+    @mock.patch("mantidimaging.gui.windows.main.view.free_all")
+    @mock.patch("mantidimaging.gui.windows.main.view.QMessageBox")
+    def test_ask_user_to_free_data(self, QMessageBox: Mock, free_all: Mock, has_other_shared_arrays: Mock):
+        has_other_shared_arrays.return_value = True
+        # makes the clickedButton the same return value mock as the addButton return
+        QMessageBox.return_value.clickedButton.return_value = QMessageBox.return_value.addButton.return_value
+
+        self.view.ask_user_to_free_data()
+
+        QMessageBox.return_value.setWindowTitle.assert_called_once()
+        QMessageBox.return_value.setText.assert_called_once()
+        self.assertEquals(QMessageBox.return_value.addButton.call_count, 2)
+        QMessageBox.return_value.exec.assert_called_once()
+        free_all.assert_called_once()
+
+    @mock.patch("mantidimaging.gui.windows.main.view.has_other_shared_arrays")
+    @mock.patch("mantidimaging.gui.windows.main.view.free_all")
+    @mock.patch("mantidimaging.gui.windows.main.view.QMessageBox")
+    def test_ask_user_to_free_data_ignore_pressed(self, QMessageBox: Mock, free_all: Mock,
+                                                  has_other_shared_arrays: Mock):
+        has_other_shared_arrays.return_value = True
+
+        self.view.ask_user_to_free_data()
+
+        QMessageBox.return_value.setWindowTitle.assert_called_once()
+        QMessageBox.return_value.setText.assert_called_once()
+        self.assertEquals(QMessageBox.return_value.addButton.call_count, 2)
+        QMessageBox.return_value.exec.assert_called_once()
+        free_all.assert_not_called()

mantidimaging/gui/windows/operations/test/test_view.py

@@ -17,7 +17,8 @@ class OperationsWindowsViewTest(unittest.TestCase):
     def setUp(self):
         # mock the view so it has the same methods
         with mock.patch("mantidimaging.gui.windows.main.view.WelcomeScreenPresenter"):
-            self.main_window = MainWindowView()
+            with mock.patch("mantidimaging.gui.windows.main.view.has_other_shared_arrays", return_value=False):
+                self.main_window = MainWindowView()
         self.window = FiltersWindowView(self.main_window)
 
     def test_collapse(self):

mantidimaging/gui/windows/stack_visualiser/test/view_test.py

@@ -38,7 +38,8 @@ def tearDown(self) -> None:
 
     def setUp(self):
         with mock.patch("mantidimaging.gui.windows.main.view.WelcomeScreenPresenter"):
-            self.window = MainWindowView()
+            with mock.patch("mantidimaging.gui.windows.main.view.has_other_shared_arrays", return_value=False):
+                self.window = MainWindowView()
         self.window.remove_stack = mock.Mock()
         self.dock, self.view, self.test_data = self._add_stack_visualiser()