def __init__(self, loaded_data, context, parent):
# set up the views
self.load_file_view = BrowseFileWidgetView(parent)
self.load_file_view.hide_browse()
self.load_run_view = LoadRunWidgetView(parent)
self.load_run_view.hide_current_run_button()
self.load_run_view.hide_instrument_label()
self.load_widget_view = LoadWidgetView(
parent=parent,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view)
self.load_widget = LoadWidgetPresenterEA(
self.load_widget_view, LoadWidgetModel(loaded_data, context))
self.file_widget = BrowseFileWidgetPresenter(
self.load_file_view, BrowseFileWidgetModel(loaded_data, context))
self.run_widget = LoadRunWidgetPresenterEA(
self.load_run_view, LoadRunWidgetModel(loaded_data, context))
self.load_widget.set_load_file_widget(self.file_widget)
self.load_widget.set_load_run_widget(self.run_widget)
self.load_widget.set_current_instrument(
context.data_context.instrument)
self.run_widget.updated_directory.add_subscriber(
self.file_widget.updated_file_path)
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
self.load_patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
self.data = MuonLoadData()
self.context = MuonDataContext(self.data)
self.context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.data, self.context)
self.load_run_model = LoadRunWidgetModel(self.data, self.context)
self.model = LoadWidgetModel(self.data, self.context)
self.view = LoadWidgetView(parent=self.obj,
load_run_view=self.load_run_view,
load_file_view=self.load_file_view)
self.presenter = LoadWidgetPresenter(view=self.view, model=self.model)
self.presenter.set_load_file_widget(
BrowseFileWidgetPresenter(self.load_file_view,
self.load_file_model))
self.presenter.set_load_run_widget(
LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:\\dir1\\EMU0001234.nxs"])
self.workspace_mock = self.create_fake_workspace(1)
self.load_mock.return_value = (self.workspace_mock, 1234,
"C:\\dir1\\EMU0001234.nxs")
self.load_run_mock.return_value = (self.workspace_mock, 1234,
"C:\\dir1\\EMU0001234.nxs")
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.mock_loading_to_throw()
file_utils.get_current_run_filename = mock.Mock(
return_value="EMU0001234.nxs")
def setUp(self):
self.view = BrowseFileWidgetView()
self.view.on_browse_clicked = mock.Mock()
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.model.exception_message_for_failed_files = mock.Mock()
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename'
)
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
self.load_utils_patcher.return_value = (self.create_fake_workspace(1),
'22222', 'filename')
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.view = BrowseFileWidgetView()
self.view.on_browse_clicked = mock.Mock()
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.data = MuonLoadData()
self.context = MuonDataContext
self.context.instrument = 'EMU'
self.model = BrowseFileWidgetModel(self.data, self.context)
self.model.exception_message_for_failed_files = mock.Mock()
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def setUp(self):
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QWidget()
ConfigService['default.instrument'] = 'MUSR'
setup_context_for_tests(self)
self.context.instrument = 'MUSR'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data,
self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data,
self.context)
self.view = LoadWidgetView(parent=self.obj,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view)
self.presenter = LoadWidgetPresenter(
self.view, LoadWidgetModel(self.loaded_data, self.context))
self.presenter.set_load_file_widget(
BrowseFileWidgetPresenter(self.load_file_view,
self.load_file_model))
self.presenter.set_load_run_widget(
LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.view.multiple_loading_check.setCheckState(1)
self.presenter.handle_multiple_files_option_changed()
self.runs = [15196, 15197]
self.workspaces = [self.create_fake_workspace(1) for _ in self.runs]
self.filenames = FileFinder.findRuns(
'MUSR00015196.nxs, MUSR00015197.nxs')
def setUp(self):
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QWidget()
setup_context_for_tests(self)
self.context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data,
self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data,
self.context)
self.presenter = LoadWidgetPresenter(
LoadWidgetView(parent=self.obj,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view),
LoadWidgetModel(self.loaded_data, self.context))
self.presenter.set_load_file_widget(
BrowseFileWidgetPresenter(self.load_file_view,
self.load_file_model))
self.presenter.set_load_run_widget(
LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.filepath = FileFinder.findRuns('MUSR00022725.nxs')[0]
self.load_patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
self.mock_workspace = self.create_fake_workspace(1)
self.mock_loading_from_browse(self.mock_workspace,
"C:\dir1\dir2\dir3\EMU0001234.nxs", 1234)
file_utils.get_current_run_filename = mock.Mock(
return_value="C:\dir1\dir2\dir3\EMU0001234.nxs")
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
def setGroupAndPairsToEmptyList(grouping_context):
grouping_context._groups = []
grouping_context._pairs = []
self.group_context.reset_group_and_pairs_to_default = mock.MagicMock(
side_effect=setGroupAndPairsToEmptyList(self.group_context))
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
self.data = MuonLoadData()
self.context = MuonDataContext(self.data)
self.context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.data, self.context)
self.load_run_model = LoadRunWidgetModel(self.data, self.context)
self.presenter = LoadWidgetPresenter(
LoadWidgetView(parent=self.obj,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view),
LoadWidgetModel(self.data, self.context))
self.presenter.set_load_file_widget(
BrowseFileWidgetPresenter(self.load_file_view,
self.load_file_model))
self.presenter.set_load_run_widget(
LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.filepath = FileFinder.findRuns('MUSR00022725.nxs')[0]
self.load_patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
self.mock_workspace = self.create_fake_workspace(1)
self.mock_loading_from_browse(self.mock_workspace,
"C:\dir1\dir2\dir3\EMU0001234.nxs", 1234)
file_utils.get_current_run_filename = mock.Mock(
return_value="C:\dir1\dir2\dir3\EMU0001234.nxs")
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.view = BrowseFileWidgetView()
self.view.on_browse_clicked = mock.Mock()
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.model.exception_message_for_failed_files = mock.Mock()
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def setUp(self):
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.view = BrowseFileWidgetView()
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
self.load_utils_patcher.return_value = (self.create_fake_workspace(1), '22222', 'filename')
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
ConfigService['default.instrument'] = 'MUSR'
setup_context_for_tests(self)
self.context.instrument = 'MUSR'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data, self.context)
self.view = LoadWidgetView(parent=self.obj, load_file_view=self.load_file_view,
load_run_view=self.load_run_view)
self.presenter = LoadWidgetPresenter(self.view, LoadWidgetModel(self.loaded_data, self.context))
self.presenter.set_load_file_widget(BrowseFileWidgetPresenter(self.load_file_view, self.load_file_model))
self.presenter.set_load_run_widget(LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.view.multiple_loading_check.setCheckState(1)
self.presenter.handle_multiple_files_option_changed()
self.runs = [15196, 15197]
self.workspaces = [self.create_fake_workspace(1) for _ in self.runs]
self.filenames = FileFinder.findRuns('MUSR00015196.nxs, MUSR00015197.nxs')
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.data = MuonLoadData()
self.view = BrowseFileWidgetView()
self.model = BrowseFileWidgetModel(self.data)
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
self.presenter.enable_multiple_files(True)
patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.data = MuonLoadData()
self.context = MuonDataContext
self.context.instrument = 'EMU'
self.view = BrowseFileWidgetView()
self.model = BrowseFileWidgetModel(self.data, self.context)
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
self.load_patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data, self.context)
self.model = LoadWidgetModel(self.loaded_data, self.context)
self.view = LoadWidgetView(parent=self.obj, load_run_view=self.load_run_view,
load_file_view=self.load_file_view)
self.presenter = LoadWidgetPresenter(view=self.view, model=self.model)
self.presenter.set_load_file_widget(BrowseFileWidgetPresenter(self.load_file_view, self.load_file_model))
self.presenter.set_load_run_widget(LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:\\dir1\\EMU0001234.nxs"])
self.workspace_mock = self.create_fake_workspace(1)
self.load_mock.return_value = (self.workspace_mock, 1234, "C:\\dir1\\EMU0001234.nxs")
self.load_run_mock.return_value = (self.workspace_mock, 1234, "C:\\dir1\\EMU0001234.nxs")
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.mock_loading_to_throw()
file_utils.get_current_run_filename = mock.Mock(return_value="EMU0001234.nxs")
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.data = MuonLoadData()
self.view = BrowseFileWidgetView()
self.model = BrowseFileWidgetModel(self.data)
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
self.presenter.enable_multiple_files(True)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def setUp(self):
self._qapp = mock_widget.mockQapp()
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.view = BrowseFileWidgetView()
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def __init__(self, loaded_data, context, parent):
# set up the views
self.load_file_view = BrowseFileWidgetView(parent)
self.load_run_view = LoadRunWidgetView(parent)
self.load_widget_view = LoadWidgetView(parent=parent,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view)
self.load_widget = LoadWidgetPresenter(self.load_widget_view,
LoadWidgetModel(loaded_data, context))
self.file_widget = BrowseFileWidgetPresenter(self.load_file_view, BrowseFileWidgetModel(loaded_data, context))
self.run_widget = LoadRunWidgetPresenter(self.load_run_view, LoadRunWidgetModel(loaded_data, context))
self.load_widget.set_load_file_widget(self.file_widget)
self.load_widget.set_load_run_widget(self.run_widget)
self.load_widget.set_current_instrument(context.instrument)
def setUp(self):
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QWidget()
setup_context_for_ea_tests(self)
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_run_view.set_current_instrument = mock.Mock()
self.view = LoadWidgetView(parent=self.obj,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view)
self.model = LoadWidgetModel(self.loaded_data, self.context)
self.model.instrument = mock.Mock()
self.presenter = LoadWidgetPresenterEA(self.view, self.model)
loaded_data = mock.Mock()
context = mock.Mock()
self.run_widget = LoadRunWidgetPresenterEA(
self.load_run_view, LoadRunWidgetModel(loaded_data, context))
self.presenter.set_load_run_widget(self.run_widget)
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
setup_context_for_tests(self)
self.context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data, self.context)
self.presenter = LoadWidgetPresenter(
LoadWidgetView(parent=self.obj, load_file_view=self.load_file_view, load_run_view=self.load_run_view),
LoadWidgetModel(self.loaded_data, self.context))
self.presenter.set_load_file_widget(BrowseFileWidgetPresenter(self.load_file_view, self.load_file_model))
self.presenter.set_load_run_widget(LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.filepath = FileFinder.findRuns('MUSR00022725.nxs')[0]
self.load_patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
self.mock_workspace = self.create_fake_workspace(1)
self.mock_loading_from_browse(self.mock_workspace, "C:\dir1\dir2\dir3\EMU0001234.nxs", 1234)
file_utils.get_current_run_filename = mock.Mock(return_value="C:\dir1\dir2\dir3\EMU0001234.nxs")
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
def __init__(self, loaded_data, context, parent):
# set up the views
self.load_file_view = BrowseFileWidgetView(parent)
self.load_run_view = LoadRunWidgetView(parent)
self.load_widget_view = LoadWidgetView(
parent=parent,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view)
self.load_widget = LoadWidgetPresenter(
self.load_widget_view, LoadWidgetModel(loaded_data, context))
self.file_widget = BrowseFileWidgetPresenter(
self.load_file_view, BrowseFileWidgetModel(loaded_data, context))
self.run_widget = LoadRunWidgetPresenter(
self.load_run_view, LoadRunWidgetModel(loaded_data, context))
self.load_widget.set_load_file_widget(self.file_widget)
self.load_widget.set_load_run_widget(self.run_widget)
self.load_widget.set_current_instrument(
context.data_context.instrument)
context.update_view_from_model_notifier.add_subscriber(
self.load_widget.update_view_from_model_observer)
class LoadRunWidgetPresenterLoadFailTest(GuiTest):
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
QApplication.instance().processEvents()
def create_fake_workspace(self, name):
workspace_mock = CreateSampleWorkspace(StoreInADS=False)
LoadInstrument(Workspace=workspace_mock,
InstrumentName='EMU',
RewriteSpectraMap=False,
StoreInADS=False)
return {
'OutputWorkspace': [MuonWorkspaceWrapper(workspace_mock)],
'MainFieldDirection': 'transverse'
}
def setUp(self):
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QWidget()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
self.load_patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data,
self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data,
self.context)
self.model = LoadWidgetModel(self.loaded_data, self.context)
self.view = LoadWidgetView(parent=self.obj,
load_run_view=self.load_run_view,
load_file_view=self.load_file_view)
self.presenter = LoadWidgetPresenter(view=self.view, model=self.model)
self.presenter.set_load_file_widget(
BrowseFileWidgetPresenter(self.load_file_view,
self.load_file_model))
self.presenter.set_load_run_widget(
LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:\\dir1\\EMU0001234.nxs"])
self.workspace_mock = self.create_fake_workspace(1)
self.load_mock.return_value = (self.workspace_mock, 1234,
"C:\\dir1\\EMU0001234.nxs", False)
self.load_run_mock.return_value = (self.workspace_mock, 1234,
"C:\\dir1\\EMU0001234.nxs", False)
self.presenter.load_file_widget.on_browse_button_clicked()
self.context.update_current_data = mock.MagicMock(return_value=([],
[]))
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
file_utils.get_current_run_filename = mock.Mock(
return_value="EMU0001234.nxs")
def tearDown(self):
self.obj = None
def load_failure(self):
raise ValueError("Error text")
def mock_loading_to_throw(self):
self.load_mock.side_effect = self.load_failure
self.load_run_mock.side_effect = self.load_failure
def mock_disabling_buttons_in_run_and_file_widget(self):
self.load_run_view.disable_load_buttons = mock.Mock()
self.load_run_view.enable_load_buttons = mock.Mock()
self.load_file_view.disable_load_buttons = mock.Mock()
self.load_file_view.enable_load_buttons = mock.Mock()
def assert_model_unchanged(self):
self.assertEqual(self.presenter._model.filenames,
["C:\\dir1\\EMU0001234.nxs"])
self.assertEqual(self.presenter._model.workspaces,
[self.workspace_mock])
self.assertEqual(self.presenter._model.runs, [[1234]])
def assert_interface_unchanged(self):
self.assertEqual(self.load_file_view.get_file_edit_text(),
"C:\\dir1\\EMU0001234.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1234")
# ------------------------------------------------------------------------------------------------------------------
# TESTS : The interface should always revert to its previous state if a load fails from anywhere in the widget
# and a warning should be shown to the user.
# ------------------------------------------------------------------------------------------------------------------
def test_that_if_load_fails_from_browse_that_model_and_interface_are_unchanged_from_previous_state(
self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_user_file_entry_that_model_and_interface_are_unchanged_from_previous_state(
self):
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_current_run_that_model_and_interface_are_unchanged_from_previous_state(
self):
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_user_run_entry_that_model_and_interface_are_unchanged_from_previous_state(
self):
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_browse_that_warning_is_displayed(self):
self.mock_loading_to_throw()
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(
self.presenter.load_file_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_user_file_entry_that_warning_is_displayed(
self):
self.mock_loading_to_throw()
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(
self.presenter.load_file_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_current_run_that_warning_is_displayed(
self):
self.mock_loading_to_throw()
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(
self.presenter.load_run_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_user_run_entry_that_warning_is_displayed(
self):
self.mock_loading_to_throw()
self.presenter.load_run_widget._view.run_edit.setText('1239')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(
self.presenter.load_run_widget._view.warning_popup.call_count, 1)
class LoadFileWidgetPresenterMultipleFileModeTest(unittest.TestCase):
def run_test_with_and_without_threading(test_function):
def run_twice(self):
test_function(self)
self.setUp()
self.presenter._use_threading = False
test_function(self)
return run_twice
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def setUp(self):
self._qapp = mock_widget.mockQapp()
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.view = BrowseFileWidgetView()
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def mock_loading_multiple_files_from_browse(self, runs, workspaces, filenames):
self.view.show_file_browser_and_return_selection = mock.Mock(return_value=filenames)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip(workspaces, runs, filenames))
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'EMU'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock]}
# ------------------------------------------------------------------------------------------------------------------
# TESTS : Multiple runs can be selected via browse and entered explicitly using the ";" separator
# ------------------------------------------------------------------------------------------------------------------
@run_test_with_and_without_threading
def test_that_cannot_load_same_file_twice_from_same_browse_even_if_filepaths_are_different(self):
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir2/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.load_utils_patcher.load_workspace_from_filename.call_count, 2)
self.load_utils_patcher.load_workspace_from_filename.assert_any_call("C:/dir1/file1.nxs")
self.load_utils_patcher.load_workspace_from_filename.assert_any_call("C:/dir2/file2.nxs")
@run_test_with_and_without_threading
def test_that_cannot_load_same_file_twice_from_user_input_even_if_filepaths_are_different(self):
self.view.set_file_edit("C:/dir1/file1.nxs;C:/dir2/file1.nxs;C:/dir2/file2.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.load_utils_patcher.load_workspace_from_filename.call_count, 2)
self.load_utils_patcher.load_workspace_from_filename.assert_any_call("C:/dir1/file1.nxs")
self.load_utils_patcher.load_workspace_from_filename.assert_any_call("C:/dir2/file2.nxs")
@run_test_with_and_without_threading
def test_that_cannot_browse_and_load_same_run_twice_even_if_filenames_are_different(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip([workspace_1, workspace_2],
[1234, 1234],
["C:/dir1/file1.nxs", "C:/dir1/file2.nxs"]))
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir1/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
# Load will take the last occurrence of the run from the list
six.assertCountEqual(self, self.model.loaded_filenames, ["C:/dir1/file2.nxs"])
six.assertCountEqual(self, self.model.loaded_workspaces, [workspace_2])
six.assertCountEqual(self, self.model.loaded_runs, [[1234]])
#
@run_test_with_and_without_threading
def test_that_cannot_input_and_load_same_run_twice_even_if_filenames_are_different(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip([workspace_1, workspace_2], [1234, 1234], ["C:/dir1/file1.nxs", "C:/dir1/file2.nxs"]))
self.view.set_file_edit("C:/dir1/file1.nxs;C:/dir1/file2.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
# Load will take the last occurrence of the run from the user input
six.assertCountEqual(self, self.model.loaded_filenames, ["C:/dir1/file2.nxs"])
six.assertCountEqual(self, self.model.loaded_workspaces, [workspace_2])
six.assertCountEqual(self, self.model.loaded_runs, [[1234]])
@run_test_with_and_without_threading
def test_that_loading_two_files_from_browse_sets_model_and_interface_correctly(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.mock_loading_multiple_files_from_browse([1234, 1235], [workspace_1, workspace_2],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.assertEqual(self.model.loaded_workspaces, [workspace_1, workspace_2])
self.assertEqual(self.model.loaded_runs, [[1234], [1235]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir2/file2.nxs")
@run_test_with_and_without_threading
def test_that_loading_two_files_from_user_input_sets_model_and_interface_correctly(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip([workspace_1, workspace_2],
[1234, 1235],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"]))
self.view.set_file_edit("C:/dir1/file1.nxs;C:/dir2/file2.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.assertEqual(self.model.loaded_workspaces, [workspace_1, workspace_2])
self.assertEqual(self.model.loaded_runs, [[1234], [1235]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir2/file2.nxs")
@run_test_with_and_without_threading
def test_that_loading_two_files_from_browse_sets_interface_alphabetically(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.mock_loading_multiple_files_from_browse([1234, 1235], [workspace_1, workspace_2],
["C:/dir1/file2.nxs", "C:/dir1/file1.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir1/file2.nxs")
@run_test_with_and_without_threading
def test_that_loading_two_files_from_user_input_sets_interface_alphabetically(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=zip([workspace_2, workspace_1], [1235, 1234], ["C:/dir1/file2.nxs", "C:/dir1/file1.nxs"]))
self.view.set_file_edit("C:/dir1/file2.nxs;C:/dir1/file1.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir1/file2.nxs")
@run_test_with_and_without_threading
def test_that_loading_multiple_files_from_browse_ignores_loads_which_throw(self):
workspace= self.create_fake_workspace(1)
files = ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs", "C:/dir2/file3.nxs"]
self.view.show_file_browser_and_return_selection = mock.Mock(return_value=files)
load_return_values = [(workspace, 1234 + i, filename) for i, filename in enumerate(files)]
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=iter(IteratorWithException(load_return_values, [1])))
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs", "C:/dir2/file3.nxs"])
self.assertEqual(self.model.loaded_runs, [[1234], [1236]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir2/file3.nxs")
@run_test_with_and_without_threading
def test_that_browse_allows_loading_of_additional_files(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
workspace_3 = self.create_fake_workspace(3)
self.mock_loading_multiple_files_from_browse([1234, 1235], [workspace_1, workspace_2],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.mock_loading_multiple_files_from_browse([1236], [workspace_3], ["C:/dir1/file3.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
six.assertCountEqual(self, self.model.loaded_filenames,
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs", "C:/dir1/file3.nxs"])
six.assertCountEqual(self, self.model.loaded_workspaces, [workspace_1, workspace_2, workspace_3])
six.assertCountEqual(self, self.model.loaded_runs, [[1234], [1235], [1236]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file3.nxs")
@run_test_with_and_without_threading
def test_that_loading_an_already_loaded_run_from_different_file_overwrites(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
workspace_3 = self.create_fake_workspace(3)
self.mock_loading_multiple_files_from_browse([1234, 1235], [workspace_1, workspace_2],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
# only checks runs, so can have a different file/workspace (this is why overwriting is
# the most useful behaviour in this situation).
self.mock_loading_multiple_files_from_browse([1234], [workspace_3], ["C:/dir2/file1.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
six.assertCountEqual(self, self.model.loaded_filenames, ["C:/dir2/file1.nxs", "C:/dir2/file2.nxs"])
six.assertCountEqual(self, self.model.loaded_workspaces, [workspace_3, workspace_2])
six.assertCountEqual(self, self.model.loaded_runs, [[1234], [1235]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir2/file1.nxs")
class LoadRunWidgetPresenterMultipleFileTest(unittest.TestCase):
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
ConfigService['default.instrument'] = 'MUSR'
setup_context_for_tests(self)
self.context.instrument = 'MUSR'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data, self.context)
self.view = LoadWidgetView(parent=self.obj, load_file_view=self.load_file_view,
load_run_view=self.load_run_view)
self.presenter = LoadWidgetPresenter(self.view, LoadWidgetModel(self.loaded_data, self.context))
self.presenter.set_load_file_widget(BrowseFileWidgetPresenter(self.load_file_view, self.load_file_model))
self.presenter.set_load_run_widget(LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.view.multiple_loading_check.setCheckState(1)
self.presenter.handle_multiple_files_option_changed()
self.runs = [15196, 15197]
self.workspaces = [self.create_fake_workspace(1) for _ in self.runs]
self.filenames = FileFinder.findRuns('MUSR00015196.nxs, MUSR00015197.nxs')
def tearDown(self):
self.obj = None
def assert_model_contains_correct_loaded_data(self):
# use sorted due to threads finishing at different times
self.assertEqual(sorted(self.presenter._model.filenames), ['Co-added'])
self.assertEqual(sorted(self.presenter._model.runs), [sorted(self.runs)])
def assert_interface_contains_correct_runs_and_files(self):
self.assertEqual(self.load_file_view.get_file_edit_text(), ";".join(['Co-added']))
self.assertEqual(self.load_run_view.get_run_edit_text(), "15196-15197")
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'MUSR'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock]}
# ------------------------------------------------------------------------------------------------------------------
# TESTS : The loading of multiple files is supported by the widget
# ------------------------------------------------------------------------------------------------------------------
def test_that_loading_multiple_files_via_browse_sets_model_and_interface_correctly(self):
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=self.filenames)
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_contains_correct_loaded_data()
self.assert_interface_contains_correct_runs_and_files()
def test_that_loading_multiple_files_via_user_input_files_sets_model_and_interface_correctly(self):
self.load_file_view.file_path_edit.setText(';'.join(self.filenames))
self.load_file_view.file_path_edit.returnPressed.emit()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_contains_correct_loaded_data()
self.assert_interface_contains_correct_runs_and_files()
def test_that_loading_multiple_files_via_user_input_runs_sets_model_and_interface_correctly(self):
self.load_run_view.run_edit.setText("15196-97")
self.load_run_view.run_edit.returnPressed.emit()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assert_model_contains_correct_loaded_data()
self.assert_interface_contains_correct_runs_and_files()
class LoadRunWidgetPresenterLoadFailTest(unittest.TestCase):
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'EMU'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock], 'MainFieldDirection': 'transverse'}
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
self.load_patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data, self.context)
self.model = LoadWidgetModel(self.loaded_data, self.context)
self.view = LoadWidgetView(parent=self.obj, load_run_view=self.load_run_view,
load_file_view=self.load_file_view)
self.presenter = LoadWidgetPresenter(view=self.view, model=self.model)
self.presenter.set_load_file_widget(BrowseFileWidgetPresenter(self.load_file_view, self.load_file_model))
self.presenter.set_load_run_widget(LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:\\dir1\\EMU0001234.nxs"])
self.workspace_mock = self.create_fake_workspace(1)
self.load_mock.return_value = (self.workspace_mock, 1234, "C:\\dir1\\EMU0001234.nxs")
self.load_run_mock.return_value = (self.workspace_mock, 1234, "C:\\dir1\\EMU0001234.nxs")
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.mock_loading_to_throw()
file_utils.get_current_run_filename = mock.Mock(return_value="EMU0001234.nxs")
def tearDown(self):
self.obj = None
def load_failure(self):
raise ValueError("Error text")
def mock_loading_to_throw(self):
self.load_mock.side_effect = self.load_failure
self.load_run_mock.side_effect = self.load_failure
def mock_disabling_buttons_in_run_and_file_widget(self):
self.load_run_view.disable_load_buttons = mock.Mock()
self.load_run_view.enable_load_buttons = mock.Mock()
self.load_file_view.disable_load_buttons = mock.Mock()
self.load_file_view.enable_load_buttons = mock.Mock()
def assert_model_unchanged(self):
self.assertEqual(self.presenter._model.filenames, ["C:\\dir1\\EMU0001234.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.workspace_mock])
self.assertEqual(self.presenter._model.runs, [[1234]])
def assert_interface_unchanged(self):
self.assertEqual(self.load_file_view.get_file_edit_text(), "C:\\dir1\\EMU0001234.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1234")
# ------------------------------------------------------------------------------------------------------------------
# TESTS : The interface should always revert to its previous state if a load fails from anywhere in the widget
# and a warning should be shown to the user.
# ------------------------------------------------------------------------------------------------------------------
def test_that_if_load_fails_from_browse_that_model_and_interface_are_unchanged_from_previous_state(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_user_file_entry_that_model_and_interface_are_unchanged_from_previous_state(self):
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_current_run_that_model_and_interface_are_unchanged_from_previous_state(self):
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_user_run_entry_that_model_and_interface_are_unchanged_from_previous_state(self):
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_browse_that_warning_is_displayed(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.presenter.load_file_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_user_file_entry_that_warning_is_displayed(self):
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.presenter.load_file_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_current_run_that_warning_is_displayed(self):
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.presenter.load_run_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_user_run_entry_that_warning_is_displayed(self):
self.presenter.load_run_widget._view.run_edit.setText('1239')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.presenter.load_run_widget._view.warning_popup.call_count, 1)
class LoadFileWidgetViewTest(unittest.TestCase):
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.view = BrowseFileWidgetView()
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
def test_view_initialized_with_empty_line_edit(self):
self.assertEqual(self.view.get_file_edit_text(), "No data loaded")
def test_reset_text_to_cache_resets_correctly(self):
text = "C:\dir1\dir2\EMU00012345.nxs;C:\dir1\dir2\EMU00012345.nxs"
self.view.set_file_edit(text)
# User may then overwrite the text in the LineEdit, causing a signal to be sent
# and the corresponding slot should implement reset_edit_to_cached_value()
self.view.reset_edit_to_cached_value()
self.assertEqual(self.view.get_file_edit_text(), text)
def test_text_clears_from_line_edit_correctly(self):
text = "C:\dir1\dir2\EMU00012345.nxs;C:\dir1\dir2\EMU00012345.nxs"
self.view.set_file_edit(text)
self.view.clear()
self.assertEqual(self.view.get_file_edit_text(), "No data loaded")
self.view.reset_edit_to_cached_value()
self.assertEqual(self.view.get_file_edit_text(), "No data loaded")
def test_text_stored_correctly_when_not_visible_in_line_edit(self):
# This feature is currently unused
text = "C:\dir1\dir2\EMU00012345.nxs;C:\dir1\dir2\EMU00012345.nxs"
self.view.set_file_edit(text, store=True)
self.assertEqual(self.view.get_file_edit_text(), text)
self.view.reset_edit_to_cached_value()
self.assertEqual(self.view.get_file_edit_text(), text)
class LoadRunWidgetPresenterTest(unittest.TestCase):
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
QApplication.instance().processEvents()
def setUp(self):
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QWidget()
setup_context_for_tests(self)
self.context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data, self.context)
self.presenter = LoadWidgetPresenter(
LoadWidgetView(parent=self.obj, load_file_view=self.load_file_view, load_run_view=self.load_run_view),
LoadWidgetModel(self.loaded_data, self.context))
self.presenter.set_load_file_widget(BrowseFileWidgetPresenter(self.load_file_view, self.load_file_model))
self.presenter.set_load_run_widget(LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.filepath = FileFinder.findRuns('MUSR00022725.nxs')[0]
self.load_patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
self.mock_workspace = self.create_fake_workspace(1)
self.mock_loading_from_browse(self.mock_workspace, "C:\dir1\dir2\dir3\EMU0001234.nxs", 1234)
file_utils.get_current_run_filename = mock.Mock(return_value="C:\dir1\dir2\dir3\EMU0001234.nxs")
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
def setGroupAndPairsToEmptyList(grouping_context):
grouping_context._groups = []
grouping_context._pairs = []
self.group_context.reset_group_and_pairs_to_default = mock.MagicMock(
side_effect=setGroupAndPairsToEmptyList(self.group_context))
def tearDown(self):
self.obj = None
def create_fake_workspace(self, name):
workspace_mock = CreateSampleWorkspace(StoreInADS=False)
LoadInstrument(Workspace=workspace_mock, InstrumentName='EMU', RewriteSpectraMap=False, StoreInADS=False)
return {'OutputWorkspace': [MuonWorkspaceWrapper(workspace_mock)], 'MainFieldDirection': 'transverse'}
def mock_loading_from_browse(self, workspace, filename, run):
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=[filename])
self.load_mock.return_value = (workspace, run, filename, False)
self.load_run_mock.return_value = (workspace, run, filename, False)
def mock_loading_from_current_run(self, workspace, filename, run):
file_utils.get_current_run_filename = mock.Mock(return_value=filename)
self.load_run_mock.return_value = (workspace, run, filename, False)
self.load_mock.return_value = (workspace, run, filename, False)
def mock_user_input_single_run(self, workspace, filename, run):
self.load_run_view.get_run_edit_text = mock.Mock(return_value=str(run))
self.load_run_mock.return_value = (workspace, run, filename, False)
self.load_mock.return_value = (workspace, run, filename, False)
def mock_user_input_single_file(self, workspace, filename, run):
self.load_run_mock.return_value = (workspace, run, filename, False)
self.load_mock.return_value = (workspace, run, filename, False)
self.load_file_view.get_file_edit_text = mock.Mock(
return_value=filename)
def mock_disabling_buttons_in_run_and_file_widget(self):
self.load_run_view.disable_load_buttons = mock.Mock()
self.load_run_view.enable_load_buttons = mock.Mock()
self.load_file_view.disable_load_buttons = mock.Mock()
self.load_file_view.enable_load_buttons = mock.Mock()
# ------------------------------------------------------------------------------------------------------------------
# TESTS : The interface should correctly load/display data via the run widget (load current run and user input run)
# and via the file widget (browse button or user enters file)
# ------------------------------------------------------------------------------------------------------------------
def test_that_loading_single_file_via_browse_sets_model_data(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["C:\dir1\dir2\dir3\EMU0001234.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[1234]])
def test_that_loading_single_file_via_browse_sets_run_and_file_view(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(), "C:\dir1\dir2\dir3\EMU0001234.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1234")
def test_that_loading_via_browse_disables_all_load_buttons(self):
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count, 2)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 2)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 1)
def test_that_loading_single_file_via_user_input_file_is_loaded_into_model_correctly(self):
self.mock_user_input_single_file(self.mock_workspace, "C:\dir1\dir2\dir3\EMU0001111.nxs", 1111)
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["C:\dir1\dir2\dir3\EMU0001111.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[1111]])
def test_that_loading_single_file_via_user_input_file_is_displayed_correctly_in_the_interface(self):
self.mock_user_input_single_file(self.mock_workspace, "C:\dir1\dir2\dir3\EMU0001111.nxs", 1111)
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(), "C:\dir1\dir2\dir3\EMU0001111.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1111")
def test_that_loading_via_user_input_file_disables_all_load_buttons(self):
self.mock_user_input_single_file([1], "C:\dir1\dir2\dir3\EMU0001111.nxs", 1111)
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count, 2)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 2)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 1)
def test_that_loading_via_load_current_run_is_loaded_into_model_correctly(self):
self.mock_loading_from_current_run(self.mock_workspace, "\\\\EMU\\data\\EMU0083420.nxs", 83420)
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["\\\\EMU\\data\\EMU0083420.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[83420]])
def test_that_loading_via_load_current_run_is_displayed_correctly_in_the_interface(self):
self.mock_loading_from_current_run(self.mock_workspace, "\\\\EMU\\data\\EMU0083420.nxs", 83420)
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(), "\\\\EMU\\data\\EMU0083420.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "83420")
def test_that_loading_via_load_current_run_disables_all_load_buttons(self):
self.mock_loading_from_current_run(self.mock_workspace, "\\\\EMU\\data\\EMU0083420.nxs", 83420)
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 3)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 2)
def test_that_user_input_run_is_loaded_into_model_correctly(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs", 12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["EMU00012345.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[12345]])
def test_that_user_input_run_is_displayed_correctly_in_the_interface(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs", 12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(), "EMU00012345.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "12345")
def test_that_loading_via_user_input_run_disables_all_load_buttons(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs", 12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 3)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 2)
def setUp(self):
self.view = BrowseFileWidgetView()
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.view = BrowseFileWidgetView()
class LoadRunWidgetPresenterTest(unittest.TestCase):
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
setup_context_for_tests(self)
self.context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data, self.context)
self.presenter = LoadWidgetPresenter(
LoadWidgetView(parent=self.obj, load_file_view=self.load_file_view, load_run_view=self.load_run_view),
LoadWidgetModel(self.loaded_data, self.context))
self.presenter.set_load_file_widget(BrowseFileWidgetPresenter(self.load_file_view, self.load_file_model))
self.presenter.set_load_run_widget(LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.filepath = FileFinder.findRuns('MUSR00022725.nxs')[0]
self.load_patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename')
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
self.mock_workspace = self.create_fake_workspace(1)
self.mock_loading_from_browse(self.mock_workspace, "C:\dir1\dir2\dir3\EMU0001234.nxs", 1234)
file_utils.get_current_run_filename = mock.Mock(return_value="C:\dir1\dir2\dir3\EMU0001234.nxs")
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
def tearDown(self):
self.obj = None
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'EMU'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock], 'MainFieldDirection': 'transverse'}
def mock_loading_from_browse(self, workspace, filename, run):
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=[filename])
self.load_mock.return_value = (workspace, run, filename)
self.load_run_mock.return_value = (workspace, run, filename)
def mock_loading_from_current_run(self, workspace, filename, run):
file_utils.get_current_run_filename = mock.Mock(return_value=filename)
self.load_run_mock.return_value = (workspace, run, filename)
self.load_mock.return_value = (workspace, run, filename)
def mock_user_input_single_run(self, workspace, filename, run):
self.load_run_view.get_run_edit_text = mock.Mock(return_value=str(run))
self.load_run_mock.return_value = (workspace, run, filename)
self.load_mock.return_value = (workspace, run, filename)
def mock_user_input_single_file(self, workspace, filename, run):
self.load_run_mock.return_value = (workspace, run, filename)
self.load_mock.return_value = (workspace, run, filename)
self.load_file_view.get_file_edit_text = mock.Mock(
return_value=filename)
def mock_disabling_buttons_in_run_and_file_widget(self):
self.load_run_view.disable_load_buttons = mock.Mock()
self.load_run_view.enable_load_buttons = mock.Mock()
self.load_file_view.disable_load_buttons = mock.Mock()
self.load_file_view.enable_load_buttons = mock.Mock()
# ------------------------------------------------------------------------------------------------------------------
# TESTS : The interface should correctly load/display data via the run widget (load current run and user input run)
# and via the file widget (browse button or user enters file)
# ------------------------------------------------------------------------------------------------------------------
def test_that_loading_single_file_via_browse_sets_model_data(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["C:\dir1\dir2\dir3\EMU0001234.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[1234]])
def test_that_loading_single_file_via_browse_sets_run_and_file_view(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(), "C:\dir1\dir2\dir3\EMU0001234.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1234")
def test_that_loading_via_browse_disables_all_load_buttons(self):
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count, 2)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 2)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 1)
def test_that_loading_single_file_via_user_input_file_is_loaded_into_model_correctly(self):
self.mock_user_input_single_file(self.mock_workspace, "C:\dir1\dir2\dir3\EMU0001111.nxs", 1111)
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["C:\dir1\dir2\dir3\EMU0001111.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[1111]])
def test_that_loading_single_file_via_user_input_file_is_displayed_correctly_in_the_interface(self):
self.mock_user_input_single_file(self.mock_workspace, "C:\dir1\dir2\dir3\EMU0001111.nxs", 1111)
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(), "C:\dir1\dir2\dir3\EMU0001111.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1111")
def test_that_loading_via_user_input_file_disables_all_load_buttons(self):
self.mock_user_input_single_file([1], "C:\dir1\dir2\dir3\EMU0001111.nxs", 1111)
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count, 2)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 2)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 1)
def test_that_loading_via_load_current_run_is_loaded_into_model_correctly(self):
self.mock_loading_from_current_run(self.mock_workspace, "\\\\EMU\\data\\EMU0083420.nxs", 83420)
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["\\\\EMU\\data\\EMU0083420.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[83420]])
def test_that_loading_via_load_current_run_is_displayed_correctly_in_the_interface(self):
self.mock_loading_from_current_run(self.mock_workspace, "\\\\EMU\\data\\EMU0083420.nxs", 83420)
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(), "\\\\EMU\\data\\EMU0083420.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "83420")
def test_that_loading_via_load_current_run_disables_all_load_buttons(self):
self.mock_loading_from_current_run(self.mock_workspace, "\\\\EMU\\data\\EMU0083420.nxs", 83420)
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 3)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 2)
def test_that_user_input_run_is_loaded_into_model_correctly(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs", 12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["EMU00012345.nxs"])
self.assertEqual(self.presenter._model.workspaces, [self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[12345]])
def test_that_user_input_run_is_displayed_correctly_in_the_interface(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs", 12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(), "EMU00012345.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "12345")
def test_that_loading_via_user_input_run_disables_all_load_buttons(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs", 12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 3)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 2)
class LoadFileWidgetPresenterTest(unittest.TestCase):
def run_test_with_and_without_threading(test_function):
def run_twice(self):
test_function(self)
self.setUp()
self.presenter._use_threading = False
test_function(self)
return run_twice
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.view = BrowseFileWidgetView()
self.view.on_browse_clicked = mock.Mock()
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.data = MuonLoadData()
self.model = BrowseFileWidgetModel(self.data)
self.model.exception_message_for_failed_files = mock.Mock()
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
self.presenter.enable_multiple_files(False)
patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def mock_browse_button_to_return_files(self, files):
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=files)
def mock_user_input_text(self, text):
self.view.get_file_edit_text = mock.Mock(return_value=text)
def mock_model_to_load_workspaces(self, workspaces, runs, filenames):
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=zip(workspaces, runs, filenames))
def load_workspaces_into_model_and_view_from_browse(
self, workspaces, runs, files):
self.mock_model_to_load_workspaces(workspaces, runs, files)
self.mock_browse_button_to_return_files(files)
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
def load_failure(self):
raise ValueError("Error text")
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
@run_test_with_and_without_threading
def test_browser_dialog_opens_when_browse_button_clicked(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(
self.view.show_file_browser_and_return_selection.call_count, 1)
@run_test_with_and_without_threading
def test_loading_not_initiated_if_no_file_selected_from_browser(self):
self.mock_model_to_load_workspaces([], [], [])
self.mock_browse_button_to_return_files([])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(
self.load_utils_patcher.load_workspace_from_filename.call_count, 0)
@run_test_with_and_without_threading
def test_buttons_disabled_while_load_thread_running(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.load_workspace_from_filename.assert_called_once_with(
"file.nxs")
self.assertEqual(self.view.disable_load_buttons.call_count, 1)
self.assertEqual(self.view.enable_load_buttons.call_count, 1)
@run_test_with_and_without_threading
def test_buttons_enabled_after_load_even_if_load_thread_throws(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.load_utils_patcher.load_workspace_from_filename.side_effect = self.load_failure
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.load_workspace_from_filename.assert_called_once_with(
"file.nxs")
self.assertEqual(self.view.disable_load_buttons.call_count, 1)
self.assertEqual(self.view.enable_load_buttons.call_count, 1)
@run_test_with_and_without_threading
def test_files_not_loaded_into_model_if_multiple_files_selected_from_browse_in_single_file_mode(
self):
self.mock_model_to_load_workspaces(
[[1], [2]], [1234, 1235],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.mock_browse_button_to_return_files(
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.model.execute = mock.Mock()
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.execute.call_count, 0)
self.assertEqual(self.view.disable_load_buttons.call_count, 0)
self.assertEqual(self.view.enable_load_buttons.call_count, 0)
@run_test_with_and_without_threading
def test_files_not_loaded_into_model_if_multiple_files_entered_by_user_in_single_file_mode(
self):
self.mock_user_input_text("C:/dir1/file1.nxs;C:/dir2/file2.nxs")
self.mock_model_to_load_workspaces(
[[1], [2]], [1234, 1235],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.model.execute = mock.Mock()
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.execute.call_count, 0)
self.assertEqual(self.view.disable_load_buttons.call_count, 0)
self.assertEqual(self.view.enable_load_buttons.call_count, 0)
@run_test_with_and_without_threading
def test_warning_shown_if_multiple_files_selected_from_browse_in_single_file_mode(
self):
self.mock_browse_button_to_return_files(
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.mock_model_to_load_workspaces(
[[1], [2]], [1234, 1235],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.warning_popup.call_count, 1)
@run_test_with_and_without_threading
def test_warning_shown_if_multiple_files_entered_by_user_in_single_file_mode(
self):
self.mock_user_input_text("C:/dir1/file1.nxs;C:/dir2/file2.nxs")
self.mock_model_to_load_workspaces(
[[1], [2]], [1234, 1235], ["C:/dir1/file1.nxs;C:/dir2/file2.nxs"])
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.warning_popup.call_count, 1)
@run_test_with_and_without_threading
def test_single_file_from_browse_loaded_into_model_and_view_in_single_file_mode(
self):
self.mock_browse_button_to_return_files(["C:/dir1/file1.nxs"])
self.mock_model_to_load_workspaces([[1]], [1234],
["C:/dir1/file1.nxs"])
self.view.set_file_edit = mock.Mock()
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs"])
self.assertEqual(self.model.loaded_workspaces, [[1]])
self.assertEqual(self.model.loaded_runs, [1234])
self.view.set_file_edit.assert_called_once_with(
"C:/dir1/file1.nxs", mock.ANY)
@run_test_with_and_without_threading
def test_single_file_from_user_input_loaded_into_model_and_view_in_single_file_mode(
self):
self.view.set_file_edit = mock.Mock()
self.mock_model_to_load_workspaces([[1]], [1234],
["C:/dir1/file1.nxs"])
self.mock_user_input_text("C:/dir1/file1.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs"])
self.assertEqual(self.model.loaded_workspaces, [[1]])
self.assertEqual(self.model.loaded_runs, [1234])
self.view.set_file_edit.assert_called_once_with(
"C:/dir1/file1.nxs", mock.ANY)
@run_test_with_and_without_threading
def test_that_if_invalid_file_selected_in_browser_view_does_not_change(
self):
self.mock_browse_button_to_return_files(["not_a_file"])
self.mock_model_to_load_workspaces([[1]], [1234], ["not_a_file"])
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=self.load_failure)
set_file_edit_count = self.view.set_file_edit.call_count
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.set_file_edit.call_count,
set_file_edit_count)
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 0)
@run_test_with_and_without_threading
def test_that_view_reverts_to_previous_text_if_users_supplies_invalid_text(
self):
self.load_workspaces_into_model_and_view_from_browse(
[[1]], [1234], ["C:/dir1/EMU0001234.nxs"])
invalid_user_input = ["some random text", "1+1=2", "..."]
call_count = self.view.reset_edit_to_cached_value.call_count
for invalid_text in invalid_user_input:
call_count += 1
self.view.get_file_edit_text = mock.Mock(return_value=invalid_text)
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.reset_edit_to_cached_value.call_count,
call_count)
@run_test_with_and_without_threading
def test_that_model_and_interface_revert_to_previous_values_if_load_fails_from_browse(
self):
self.load_workspaces_into_model_and_view_from_browse(
[[1]], [1234], ["C:/dir1/EMU0001234.nxs"])
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=self.load_failure)
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames,
["C:/dir1/EMU0001234.nxs"])
self.assertEqual(self.model.loaded_workspaces, [[1]])
self.assertEqual(self.model.loaded_runs, [1234])
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 0)
self.assertEqual(self.view.set_file_edit.call_args[0][0],
"C:/dir1/EMU0001234.nxs")
@run_test_with_and_without_threading
def test_that_model_and_interface_revert_to_previous_values_if_load_fails_from_user_input(
self):
self.load_workspaces_into_model_and_view_from_browse(
[[1]], [1234], ["C:/dir1/EMU0001234.nxs"])
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=self.load_failure)
self.view.set_file_edit("C:\dir2\EMU000123.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames,
["C:/dir1/EMU0001234.nxs"])
self.assertEqual(self.model.loaded_workspaces, [[1]])
self.assertEqual(self.model.loaded_runs, [1234])
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 1)
class LoadFileWidgetPresenterMultipleFileModeTest(unittest.TestCase):
def run_test_with_and_without_threading(test_function):
def run_twice(self):
test_function(self)
self.setUp()
self.presenter._use_threading = False
test_function(self)
return run_twice
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.data = MuonLoadData()
self.context = MuonDataContext
self.context.instrument = 'EMU'
self.view = BrowseFileWidgetView()
self.model = BrowseFileWidgetModel(self.data, self.context)
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def mock_loading_multiple_files_from_browse(self, runs, workspaces, filenames):
self.view.show_file_browser_and_return_selection = mock.Mock(return_value=filenames)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip(workspaces, runs, filenames))
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'EMU'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock]}
# ------------------------------------------------------------------------------------------------------------------
# TESTS : Multiple runs can be selected via browse and entered explicitly using the ";" separator
# ------------------------------------------------------------------------------------------------------------------
@run_test_with_and_without_threading
def test_that_cannot_load_same_file_twice_from_same_browse_even_if_filepaths_are_different(self):
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir2/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.load_utils_patcher.load_workspace_from_filename.call_count, 2)
self.load_utils_patcher.load_workspace_from_filename.assert_any_call("C:/dir1/file1.nxs")
self.load_utils_patcher.load_workspace_from_filename.assert_any_call("C:/dir2/file2.nxs")
@run_test_with_and_without_threading
def test_that_cannot_load_same_file_twice_from_user_input_even_if_filepaths_are_different(self):
self.view.set_file_edit("C:/dir1/file1.nxs;C:/dir2/file1.nxs;C:/dir2/file2.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.load_utils_patcher.load_workspace_from_filename.call_count, 2)
self.load_utils_patcher.load_workspace_from_filename.assert_any_call("C:/dir1/file1.nxs")
self.load_utils_patcher.load_workspace_from_filename.assert_any_call("C:/dir2/file2.nxs")
@run_test_with_and_without_threading
def test_that_cannot_browse_and_load_same_run_twice_even_if_filenames_are_different(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip([workspace_1, workspace_2],
[1234, 1234],
["C:/dir1/file1.nxs", "C:/dir1/file2.nxs"]))
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir1/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
# Load will take the last occurrence of the run from the list
six.assertCountEqual(self, self.model.loaded_filenames, ["C:/dir1/file2.nxs"])
six.assertCountEqual(self, self.model.loaded_workspaces, [workspace_2])
six.assertCountEqual(self, self.model.loaded_runs, [[1234]])
#
@run_test_with_and_without_threading
def test_that_cannot_input_and_load_same_run_twice_even_if_filenames_are_different(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip([workspace_1, workspace_2], [1234, 1234], ["C:/dir1/file1.nxs", "C:/dir1/file2.nxs"]))
self.view.set_file_edit("C:/dir1/file1.nxs;C:/dir1/file2.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
# Load will take the last occurrence of the run from the user input
six.assertCountEqual(self, self.model.loaded_filenames, ["C:/dir1/file2.nxs"])
six.assertCountEqual(self, self.model.loaded_workspaces, [workspace_2])
six.assertCountEqual(self, self.model.loaded_runs, [[1234]])
@run_test_with_and_without_threading
def test_that_loading_two_files_from_browse_sets_model_and_interface_correctly(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.mock_loading_multiple_files_from_browse([1234, 1235], [workspace_1, workspace_2],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.assertEqual(self.model.loaded_workspaces, [workspace_1, workspace_2])
self.assertEqual(self.model.loaded_runs, [[1234], [1235]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir2/file2.nxs")
@run_test_with_and_without_threading
def test_that_loading_two_files_from_user_input_sets_model_and_interface_correctly(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip([workspace_1, workspace_2],
[1234, 1235],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"]))
self.view.set_file_edit("C:/dir1/file1.nxs;C:/dir2/file2.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.assertEqual(self.model.loaded_workspaces, [workspace_1, workspace_2])
self.assertEqual(self.model.loaded_runs, [[1234], [1235]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir2/file2.nxs")
@run_test_with_and_without_threading
def test_that_loading_two_files_from_browse_sets_interface_alphabetically(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.mock_loading_multiple_files_from_browse([1234, 1235], [workspace_1, workspace_2],
["C:/dir1/file2.nxs", "C:/dir1/file1.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir1/file2.nxs")
@run_test_with_and_without_threading
def test_that_loading_two_files_from_user_input_sets_interface_alphabetically(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=zip([workspace_2, workspace_1], [1235, 1234], ["C:/dir1/file2.nxs", "C:/dir1/file1.nxs"]))
self.view.set_file_edit("C:/dir1/file2.nxs;C:/dir1/file1.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir1/file2.nxs")
@run_test_with_and_without_threading
def test_that_loading_multiple_files_from_browse_ignores_loads_which_throw(self):
workspace= self.create_fake_workspace(1)
files = ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs", "C:/dir2/file3.nxs"]
self.view.show_file_browser_and_return_selection = mock.Mock(return_value=files)
load_return_values = [(workspace, 1234 + i, filename) for i, filename in enumerate(files)]
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=iter(IteratorWithException(load_return_values, [1])))
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs", "C:/dir2/file3.nxs"])
self.assertEqual(self.model.loaded_runs, [[1234], [1236]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file1.nxs;C:/dir2/file3.nxs")
@run_test_with_and_without_threading
def test_that_browse_allows_loading_of_additional_files(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
workspace_3 = self.create_fake_workspace(3)
self.mock_loading_multiple_files_from_browse([1234, 1235], [workspace_1, workspace_2],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.mock_loading_multiple_files_from_browse([1236], [workspace_3], ["C:/dir1/file3.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
six.assertCountEqual(self, self.model.loaded_filenames,
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs", "C:/dir1/file3.nxs"])
six.assertCountEqual(self, self.model.loaded_workspaces, [workspace_1, workspace_2, workspace_3])
six.assertCountEqual(self, self.model.loaded_runs, [[1234], [1235], [1236]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir1/file3.nxs")
@run_test_with_and_without_threading
def test_that_loading_an_already_loaded_run_from_different_file_overwrites(self):
workspace_1 = self.create_fake_workspace(1)
workspace_2 = self.create_fake_workspace(2)
workspace_3 = self.create_fake_workspace(3)
self.mock_loading_multiple_files_from_browse([1234, 1235], [workspace_1, workspace_2],
["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
# only checks runs, so can have a different file/workspace (this is why overwriting is
# the most useful behaviour in this situation).
self.mock_loading_multiple_files_from_browse([1234], [workspace_3], ["C:/dir2/file1.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
six.assertCountEqual(self, self.model.loaded_filenames, ["C:/dir2/file1.nxs", "C:/dir2/file2.nxs"])
six.assertCountEqual(self, self.model.loaded_workspaces, [workspace_3, workspace_2])
six.assertCountEqual(self, self.model.loaded_runs, [[1234], [1235]])
self.assertEqual(self.view.get_file_edit_text(), "C:/dir2/file1.nxs")
class LoadFileWidgetPresenterTest(unittest.TestCase):
def run_test_with_and_without_threading(test_function):
def run_twice(self):
test_function(self)
self.setUp()
self.presenter._use_threading = False
test_function(self)
return run_twice
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
QApplication.sendPostedEvents()
def setUp(self):
self.view = BrowseFileWidgetView()
self.view.on_browse_clicked = mock.Mock()
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
self.model.exception_message_for_failed_files = mock.Mock()
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename'
)
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
self.load_utils_patcher.return_value = (self.create_fake_workspace(1),
'22222', 'filename')
def mock_browse_button_to_return_files(self, files):
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=files)
def mock_user_input_text(self, text):
self.view.get_file_edit_text = mock.Mock(return_value=text)
def mock_model_to_load_workspaces(self, workspaces, runs, filenames):
psi_data = [False] * len(filenames)
self.load_utils_patcher.side_effect = zip(workspaces, runs, filenames,
psi_data)
def load_workspaces_into_model_and_view_from_browse(
self, workspaces, runs, files):
self.mock_model_to_load_workspaces(workspaces, runs, files)
self.mock_browse_button_to_return_files(files)
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
def load_failure(self, unused_arg):
raise ValueError("Error text")
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'EMU'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock]}
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
@run_test_with_and_without_threading
def test_browser_dialog_opens_when_browse_button_clicked(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(
self.view.show_file_browser_and_return_selection.call_count, 1)
@run_test_with_and_without_threading
def test_loading_not_initiated_if_no_file_selected_from_browser(self):
self.mock_model_to_load_workspaces([], [], [])
self.mock_browse_button_to_return_files([])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.load_utils_patcher.call_count, 0)
@run_test_with_and_without_threading
def test_buttons_disabled_while_load_thread_running(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.assert_called_once_with("file.nxs")
self.assertEqual(self.view.disable_load_buttons.call_count, 1)
self.assertEqual(self.view.enable_load_buttons.call_count, 1)
@run_test_with_and_without_threading
def test_buttons_enabled_after_load_even_if_load_thread_throws(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.load_utils_patcher.side_effect = self.load_failure
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.assert_called_once_with("file.nxs")
self.assertEqual(self.view.disable_load_buttons.call_count, 1)
self.assertEqual(self.view.enable_load_buttons.call_count, 1)
@run_test_with_and_without_threading
def test_single_file_from_browse_loaded_into_model_and_view_in_single_file_mode(
self):
workspace = self.create_fake_workspace(1)
self.mock_browse_button_to_return_files(["C:/dir1/file1.nxs"])
self.mock_model_to_load_workspaces([workspace], [1234],
["C:/dir1/file1.nxs"])
self.view.set_file_edit = mock.Mock()
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs"])
self.assertEqual(self.model.loaded_workspaces, [workspace])
self.assertEqual(self.model.loaded_runs, [[1234]])
self.view.set_file_edit.assert_called_once_with(
"C:/dir1/file1.nxs", mock.ANY)
@run_test_with_and_without_threading
def test_single_file_from_user_input_loaded_into_model_and_view_in_single_file_mode(
self):
workspace = self.create_fake_workspace(1)
self.view.set_file_edit = mock.Mock()
self.mock_model_to_load_workspaces([workspace], [1234],
["C:/dir1/file1.nxs"])
self.mock_user_input_text("C:/dir1/file1.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs"])
self.assertEqual(self.model.loaded_workspaces, [workspace])
self.assertEqual(self.model.loaded_runs, [[1234]])
self.view.set_file_edit.assert_called_once_with(
"C:/dir1/file1.nxs", mock.ANY)
@run_test_with_and_without_threading
def test_that_if_invalid_file_selected_in_browser_view_does_not_change(
self):
workspace = self.create_fake_workspace(1)
self.mock_browse_button_to_return_files(["not_a_file"])
self.mock_model_to_load_workspaces([workspace], [1234], ["not_a_file"])
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.side_effect = self.load_failure
set_file_edit_count = self.view.set_file_edit.call_count
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.set_file_edit.call_count,
set_file_edit_count)
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 0)
@run_test_with_and_without_threading
def test_that_view_reverts_to_previous_text_if_users_supplies_invalid_text(
self):
workspace = self.create_fake_workspace(1)
self.load_workspaces_into_model_and_view_from_browse(
[workspace], [[1234]], ["C:/dir1/EMU0001234.nxs"])
invalid_user_input = ["some random text", "1+1=2", "..."]
call_count = self.view.reset_edit_to_cached_value.call_count
for invalid_text in invalid_user_input:
call_count += 1
self.view.get_file_edit_text = mock.Mock(return_value=invalid_text)
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.reset_edit_to_cached_value.call_count,
call_count)
@run_test_with_and_without_threading
def test_that_model_and_interface_revert_to_previous_values_if_load_fails_from_browse(
self):
workspace = self.create_fake_workspace(1)
self.load_workspaces_into_model_and_view_from_browse(
[workspace], [1234], ["C:/dir1/EMU0001234.nxs"])
self.load_utils_patcher.side_effect = self.load_failure
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames,
["C:/dir1/EMU0001234.nxs"])
self.assertEqual(self.model.loaded_workspaces, [workspace])
self.assertEqual(self.model.loaded_runs, [[1234]])
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 0)
self.assertEqual(self.view.set_file_edit.call_args[0][0],
"C:/dir1/EMU0001234.nxs")
@run_test_with_and_without_threading
def test_that_model_and_interface_revert_to_previous_values_if_load_fails_from_user_input(
self):
workspace = self.create_fake_workspace(1)
self.load_workspaces_into_model_and_view_from_browse(
[workspace], [1234], ["C:/dir1/EMU0001234.nxs"])
self.load_utils_patcher.side_effect = self.load_failure
self.view.set_file_edit(r"C:\dir2\EMU000123.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames,
["C:/dir1/EMU0001234.nxs"])
self.assertEqual(self.model.loaded_workspaces, [workspace])
self.assertEqual(self.model.loaded_runs, [[1234]])
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 1)
class LoadRunWidgetPresenterTest(unittest.TestCase):
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
self.data = MuonLoadData()
self.context = MuonDataContext(self.data)
self.context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.data, self.context)
self.load_run_model = LoadRunWidgetModel(self.data, self.context)
self.presenter = LoadWidgetPresenter(
LoadWidgetView(parent=self.obj,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view),
LoadWidgetModel(self.data, self.context))
self.presenter.set_load_file_widget(
BrowseFileWidgetPresenter(self.load_file_view,
self.load_file_model))
self.presenter.set_load_run_widget(
LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.filepath = FileFinder.findRuns('MUSR00022725.nxs')[0]
self.load_patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
self.mock_workspace = self.create_fake_workspace(1)
self.mock_loading_from_browse(self.mock_workspace,
"C:\dir1\dir2\dir3\EMU0001234.nxs", 1234)
file_utils.get_current_run_filename = mock.Mock(
return_value="C:\dir1\dir2\dir3\EMU0001234.nxs")
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
def tearDown(self):
self.obj = None
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'EMU'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock]}
def mock_loading_from_browse(self, workspace, filename, run):
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=[filename])
self.load_mock.return_value = (workspace, run, filename)
self.load_run_mock.return_value = (workspace, run, filename)
def mock_loading_from_current_run(self, workspace, filename, run):
file_utils.get_current_run_filename = mock.Mock(return_value=filename)
self.load_run_mock.return_value = (workspace, run, filename)
self.load_mock.return_value = (workspace, run, filename)
def mock_user_input_single_run(self, workspace, filename, run):
self.load_run_view.get_run_edit_text = mock.Mock(return_value=str(run))
self.load_run_mock.return_value = (workspace, run, filename)
self.load_mock.return_value = (workspace, run, filename)
def mock_user_input_single_file(self, workspace, filename, run):
self.load_run_mock.return_value = (workspace, run, filename)
self.load_mock.return_value = (workspace, run, filename)
self.load_file_view.get_file_edit_text = mock.Mock(
return_value=filename)
def mock_disabling_buttons_in_run_and_file_widget(self):
self.load_run_view.disable_load_buttons = mock.Mock()
self.load_run_view.enable_load_buttons = mock.Mock()
self.load_file_view.disable_load_buttons = mock.Mock()
self.load_file_view.enable_load_buttons = mock.Mock()
# ------------------------------------------------------------------------------------------------------------------
# TESTS : The interface should correctly load/display data via the run widget (load current run and user input run)
# and via the file widget (browse button or user enters file)
# ------------------------------------------------------------------------------------------------------------------
def test_that_loading_single_file_via_browse_sets_model_data(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.presenter._model.filenames,
["C:\dir1\dir2\dir3\EMU0001234.nxs"])
self.assertEqual(self.presenter._model.workspaces,
[self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[1234]])
def test_that_loading_single_file_via_browse_sets_run_and_file_view(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(),
"C:\dir1\dir2\dir3\EMU0001234.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1234")
def test_that_loading_via_browse_disables_all_load_buttons(self):
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count,
2)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 2)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 1)
def test_that_loading_single_file_via_user_input_file_is_loaded_into_model_correctly(
self):
self.mock_user_input_single_file(self.mock_workspace,
"C:\dir1\dir2\dir3\EMU0001111.nxs",
1111)
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.presenter._model.filenames,
["C:\dir1\dir2\dir3\EMU0001111.nxs"])
self.assertEqual(self.presenter._model.workspaces,
[self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[1111]])
def test_that_loading_single_file_via_user_input_file_is_displayed_correctly_in_the_interface(
self):
self.mock_user_input_single_file(self.mock_workspace,
"C:\dir1\dir2\dir3\EMU0001111.nxs",
1111)
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(),
"C:\dir1\dir2\dir3\EMU0001111.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1111")
def test_that_loading_via_user_input_file_disables_all_load_buttons(self):
self.mock_user_input_single_file([1],
"C:\dir1\dir2\dir3\EMU0001111.nxs",
1111)
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count,
2)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 2)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 1)
def test_that_loading_via_load_current_run_is_loaded_into_model_correctly(
self):
self.mock_loading_from_current_run(self.mock_workspace,
"\\\\EMU\\data\\EMU0083420.nxs",
83420)
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.presenter._model.filenames,
["\\\\EMU\\data\\EMU0083420.nxs"])
self.assertEqual(self.presenter._model.workspaces,
[self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[83420]])
def test_that_loading_via_load_current_run_is_displayed_correctly_in_the_interface(
self):
self.mock_loading_from_current_run(self.mock_workspace,
"\\\\EMU\\data\\EMU0083420.nxs",
83420)
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(),
"\\\\EMU\\data\\EMU0083420.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "83420")
def test_that_loading_via_load_current_run_disables_all_load_buttons(self):
self.mock_loading_from_current_run(self.mock_workspace,
"\\\\EMU\\data\\EMU0083420.nxs",
83420)
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count,
1)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 3)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 2)
def test_that_user_input_run_is_loaded_into_model_correctly(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs",
12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.presenter._model.filenames, ["EMU00012345.nxs"])
self.assertEqual(self.presenter._model.workspaces,
[self.mock_workspace])
self.assertEqual(self.presenter._model.runs, [[12345]])
def test_that_user_input_run_is_displayed_correctly_in_the_interface(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs",
12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.get_file_edit_text(),
"EMU00012345.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "12345")
def test_that_loading_via_user_input_run_disables_all_load_buttons(self):
self.mock_user_input_single_run(self.mock_workspace, "EMU00012345.nxs",
12345)
self.presenter.load_run_widget.set_current_instrument('EMU')
self.mock_disabling_buttons_in_run_and_file_widget()
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(self.load_file_view.disable_load_buttons.call_count,
1)
self.assertEqual(self.load_file_view.enable_load_buttons.call_count, 1)
self.assertEqual(self.load_run_view.disable_load_buttons.call_count, 3)
self.assertEqual(self.load_run_view.enable_load_buttons.call_count, 2)
class LoadRunWidgetPresenterLoadFailTest(unittest.TestCase):
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'EMU'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock]}
def setUp(self):
self._qapp = mock_widget.mockQapp()
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QtGui.QWidget()
self.popup_patcher = mock.patch('Muon.GUI.Common.thread_model.warning')
self.addCleanup(self.popup_patcher.stop)
self.popup_mock = self.popup_patcher.start()
self.load_patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_patcher.stop)
self.load_mock = self.load_patcher.start()
self.load_run_patcher = mock.patch(
'Muon.GUI.Common.load_run_widget.load_run_model.load_utils.load_workspace_from_filename'
)
self.addCleanup(self.load_run_patcher.stop)
self.load_run_mock = self.load_run_patcher.start()
self.data = MuonLoadData()
self.context = MuonDataContext(self.data)
self.context.instrument = 'EMU'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.data, self.context)
self.load_run_model = LoadRunWidgetModel(self.data, self.context)
self.model = LoadWidgetModel(self.data, self.context)
self.view = LoadWidgetView(parent=self.obj,
load_run_view=self.load_run_view,
load_file_view=self.load_file_view)
self.presenter = LoadWidgetPresenter(view=self.view, model=self.model)
self.presenter.set_load_file_widget(
BrowseFileWidgetPresenter(self.load_file_view,
self.load_file_model))
self.presenter.set_load_run_widget(
LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_run_widget.set_current_instrument('EMU')
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:\\dir1\\EMU0001234.nxs"])
self.workspace_mock = self.create_fake_workspace(1)
self.load_mock.return_value = (self.workspace_mock, 1234,
"C:\\dir1\\EMU0001234.nxs")
self.load_run_mock.return_value = (self.workspace_mock, 1234,
"C:\\dir1\\EMU0001234.nxs")
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.mock_loading_to_throw()
file_utils.get_current_run_filename = mock.Mock(
return_value="EMU0001234.nxs")
def tearDown(self):
self.obj = None
def load_failure(self):
raise ValueError("Error text")
def mock_loading_to_throw(self):
self.load_mock.side_effect = self.load_failure
self.load_run_mock.side_effect = self.load_failure
def mock_disabling_buttons_in_run_and_file_widget(self):
self.load_run_view.disable_load_buttons = mock.Mock()
self.load_run_view.enable_load_buttons = mock.Mock()
self.load_file_view.disable_load_buttons = mock.Mock()
self.load_file_view.enable_load_buttons = mock.Mock()
def assert_model_unchanged(self):
self.assertEqual(self.presenter._model.filenames,
["C:\\dir1\\EMU0001234.nxs"])
self.assertEqual(self.presenter._model.workspaces,
[self.workspace_mock])
self.assertEqual(self.presenter._model.runs, [[1234]])
def assert_interface_unchanged(self):
self.assertEqual(self.load_file_view.get_file_edit_text(),
"C:\\dir1\\EMU0001234.nxs")
self.assertEqual(self.load_run_view.get_run_edit_text(), "1234")
# ------------------------------------------------------------------------------------------------------------------
# TESTS : The interface should always revert to its previous state if a load fails from anywhere in the widget
# and a warning should be shown to the user.
# ------------------------------------------------------------------------------------------------------------------
def test_that_if_load_fails_from_browse_that_model_and_interface_are_unchanged_from_previous_state(
self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_user_file_entry_that_model_and_interface_are_unchanged_from_previous_state(
self):
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_current_run_that_model_and_interface_are_unchanged_from_previous_state(
self):
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_user_run_entry_that_model_and_interface_are_unchanged_from_previous_state(
self):
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assert_model_unchanged()
self.assert_interface_unchanged()
def test_that_if_load_fails_from_browse_that_warning_is_displayed(self):
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(
self.presenter.load_file_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_user_file_entry_that_warning_is_displayed(
self):
self.presenter.load_file_widget.handle_file_changed_by_user()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assertEqual(
self.presenter.load_file_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_current_run_that_warning_is_displayed(
self):
self.presenter.load_run_widget.handle_load_current_run()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(
self.presenter.load_run_widget._view.warning_popup.call_count, 1)
def test_that_if_load_fails_from_user_run_entry_that_warning_is_displayed(
self):
self.presenter.load_run_widget._view.run_edit.setText('1239')
self.presenter.load_run_widget.handle_run_changed_by_user()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assertEqual(
self.presenter.load_run_widget._view.warning_popup.call_count, 1)
class LoadRunWidgetPresenterMultipleFileTest(unittest.TestCase):
def wait_for_thread(self, thread_model):
while (thread_model._thread.isRunning()):
QApplication.instance().processEvents()
time.sleep(0.1)
QApplication.instance().processEvents()
def setUp(self):
# Store an empty widget to parent all the views, and ensure they are deleted correctly
self.obj = QWidget()
ConfigService['default.instrument'] = 'MUSR'
setup_context_for_tests(self)
self.context.instrument = 'MUSR'
self.load_file_view = BrowseFileWidgetView(self.obj)
self.load_run_view = LoadRunWidgetView(self.obj)
self.load_file_model = BrowseFileWidgetModel(self.loaded_data,
self.context)
self.load_run_model = LoadRunWidgetModel(self.loaded_data,
self.context)
self.view = LoadWidgetView(parent=self.obj,
load_file_view=self.load_file_view,
load_run_view=self.load_run_view)
self.presenter = LoadWidgetPresenter(
self.view, LoadWidgetModel(self.loaded_data, self.context))
self.presenter.set_load_file_widget(
BrowseFileWidgetPresenter(self.load_file_view,
self.load_file_model))
self.presenter.set_load_run_widget(
LoadRunWidgetPresenter(self.load_run_view, self.load_run_model))
self.presenter.load_file_widget._view.warning_popup = mock.MagicMock()
self.presenter.load_run_widget._view.warning_popup = mock.MagicMock()
self.view.multiple_loading_check.setCheckState(1)
self.presenter.handle_multiple_files_option_changed()
self.runs = [15196, 15197]
self.workspaces = [self.create_fake_workspace(1) for _ in self.runs]
self.filenames = FileFinder.findRuns(
'MUSR00015196.nxs, MUSR00015197.nxs')
def tearDown(self):
self.obj = None
def assert_model_contains_correct_loaded_data(self):
# use sorted due to threads finishing at different times
self.assertEqual(sorted(self.presenter._model.filenames), ['Co-added'])
self.assertEqual(sorted(self.presenter._model.runs),
[sorted(self.runs)])
def assert_interface_contains_correct_runs_and_files(self):
self.assertEqual(self.load_file_view.get_file_edit_text(),
";".join(['Co-added']))
self.assertEqual(self.load_run_view.get_run_edit_text(), "15196-15197")
def create_fake_workspace(self, name):
workspace_mock = mock.MagicMock()
instrument_mock = mock.MagicMock()
instrument_mock.getName.return_value = 'MUSR'
workspace_mock.workspace.getInstrument.return_value = instrument_mock
return {'OutputWorkspace': [workspace_mock]}
# ------------------------------------------------------------------------------------------------------------------
# TESTS : The loading of multiple files is supported by the widget
# ------------------------------------------------------------------------------------------------------------------
def test_that_loading_multiple_files_via_browse_sets_model_and_interface_correctly(
self):
self.load_file_view.show_file_browser_and_return_selection = mock.Mock(
return_value=self.filenames)
self.presenter.load_file_widget.on_browse_button_clicked()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_contains_correct_loaded_data()
self.assert_interface_contains_correct_runs_and_files()
def test_that_loading_multiple_files_via_user_input_files_sets_model_and_interface_correctly(
self):
self.load_file_view.file_path_edit.setText(';'.join(self.filenames))
self.load_file_view.file_path_edit.returnPressed.emit()
self.wait_for_thread(self.presenter.load_file_widget._load_thread)
self.assert_model_contains_correct_loaded_data()
self.assert_interface_contains_correct_runs_and_files()
def test_that_loading_multiple_files_via_user_input_runs_sets_model_and_interface_correctly(
self):
self.load_run_view.run_edit.setText("15196-97")
self.load_run_view.run_edit.returnPressed.emit()
self.wait_for_thread(self.presenter.load_run_widget._load_thread)
self.assert_model_contains_correct_loaded_data()
self.assert_interface_contains_correct_runs_and_files()
class LoadFileWidgetViewTest(unittest.TestCase):
def setUp(self):
self.view = BrowseFileWidgetView()
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
def test_view_initialized_with_empty_line_edit(self):
self.assertEqual(self.view.get_file_edit_text(), "No data loaded")
def test_reset_text_to_cache_resets_correctly(self):
text = "C:\dir1\dir2\EMU00012345.nxs;C:\dir1\dir2\EMU00012345.nxs"
self.view.set_file_edit(text)
# User may then overwrite the text in the LineEdit, causing a signal to be sent
# and the corresponding slot should implement reset_edit_to_cached_value()
self.view.reset_edit_to_cached_value()
self.assertEqual(self.view.get_file_edit_text(), text)
def test_text_clears_from_line_edit_correctly(self):
text = "C:\dir1\dir2\EMU00012345.nxs;C:\dir1\dir2\EMU00012345.nxs"
self.view.set_file_edit(text)
self.view.clear()
self.assertEqual(self.view.get_file_edit_text(), "No data loaded")
self.view.reset_edit_to_cached_value()
self.assertEqual(self.view.get_file_edit_text(), "No data loaded")
def test_text_stored_correctly_when_not_visible_in_line_edit(self):
# This feature is currently unused
text = "C:\dir1\dir2\EMU00012345.nxs;C:\dir1\dir2\EMU00012345.nxs"
self.view.set_file_edit(text, store=True)
self.assertEqual(self.view.get_file_edit_text(), text)
self.view.reset_edit_to_cached_value()
self.assertEqual(self.view.get_file_edit_text(), text)
class LoadFileWidgetPresenterTest(unittest.TestCase):
def run_test_with_and_without_threading(test_function):
def run_twice(self):
test_function(self)
self.setUp()
self.presenter._use_threading = False
test_function(self)
return run_twice
def wait_for_thread(self, thread_model):
if thread_model:
thread_model._thread.wait()
self._qapp.processEvents()
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.view = BrowseFileWidgetView()
self.view.on_browse_clicked = mock.Mock()
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.view.show_file_browser_and_return_selection = mock.Mock(
return_value=["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.data = MuonLoadData()
self.model = BrowseFileWidgetModel(self.data)
self.model.exception_message_for_failed_files = mock.Mock()
self.view.disable_load_buttons = mock.Mock()
self.view.enable_load_buttons = mock.Mock()
self.view.warning_popup = mock.Mock()
self.presenter = BrowseFileWidgetPresenter(self.view, self.model)
self.presenter.enable_multiple_files(False)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def mock_browse_button_to_return_files(self, files):
self.view.show_file_browser_and_return_selection = mock.Mock(return_value=files)
def mock_user_input_text(self, text):
self.view.get_file_edit_text = mock.Mock(return_value=text)
def mock_model_to_load_workspaces(self, workspaces, runs, filenames):
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=zip(workspaces, runs, filenames))
def load_workspaces_into_model_and_view_from_browse(self, workspaces, runs, files):
self.mock_model_to_load_workspaces(workspaces, runs, files)
self.mock_browse_button_to_return_files(files)
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
def load_failure(self):
raise ValueError("Error text")
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
@run_test_with_and_without_threading
def test_browser_dialog_opens_when_browse_button_clicked(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.show_file_browser_and_return_selection.call_count, 1)
@run_test_with_and_without_threading
def test_loading_not_initiated_if_no_file_selected_from_browser(self):
self.mock_model_to_load_workspaces([], [], [])
self.mock_browse_button_to_return_files([])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.load_utils_patcher.load_workspace_from_filename.call_count, 0)
@run_test_with_and_without_threading
def test_buttons_disabled_while_load_thread_running(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.load_workspace_from_filename.assert_called_once_with("file.nxs")
self.assertEqual(self.view.disable_load_buttons.call_count, 1)
self.assertEqual(self.view.enable_load_buttons.call_count, 1)
@run_test_with_and_without_threading
def test_buttons_enabled_after_load_even_if_load_thread_throws(self):
self.mock_browse_button_to_return_files(["file.nxs"])
self.load_utils_patcher.load_workspace_from_filename.side_effect = self.load_failure
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.load_workspace_from_filename.assert_called_once_with("file.nxs")
self.assertEqual(self.view.disable_load_buttons.call_count, 1)
self.assertEqual(self.view.enable_load_buttons.call_count, 1)
@run_test_with_and_without_threading
def test_files_not_loaded_into_model_if_multiple_files_selected_from_browse_in_single_file_mode(self):
self.mock_model_to_load_workspaces([[1], [2]], [1234, 1235], ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.mock_browse_button_to_return_files(["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.model.execute = mock.Mock()
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.execute.call_count, 0)
self.assertEqual(self.view.disable_load_buttons.call_count, 0)
self.assertEqual(self.view.enable_load_buttons.call_count, 0)
@run_test_with_and_without_threading
def test_files_not_loaded_into_model_if_multiple_files_entered_by_user_in_single_file_mode(self):
self.mock_user_input_text("C:/dir1/file1.nxs;C:/dir2/file2.nxs")
self.mock_model_to_load_workspaces([[1], [2]], [1234, 1235], ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.model.execute = mock.Mock()
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.execute.call_count, 0)
self.assertEqual(self.view.disable_load_buttons.call_count, 0)
self.assertEqual(self.view.enable_load_buttons.call_count, 0)
@run_test_with_and_without_threading
def test_warning_shown_if_multiple_files_selected_from_browse_in_single_file_mode(self):
self.mock_browse_button_to_return_files(["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.mock_model_to_load_workspaces([[1], [2]], [1234, 1235], ["C:/dir1/file1.nxs", "C:/dir2/file2.nxs"])
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.warning_popup.call_count, 1)
@run_test_with_and_without_threading
def test_warning_shown_if_multiple_files_entered_by_user_in_single_file_mode(self):
self.mock_user_input_text("C:/dir1/file1.nxs;C:/dir2/file2.nxs")
self.mock_model_to_load_workspaces([[1], [2]], [1234, 1235], ["C:/dir1/file1.nxs;C:/dir2/file2.nxs"])
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.warning_popup.call_count, 1)
@run_test_with_and_without_threading
def test_single_file_from_browse_loaded_into_model_and_view_in_single_file_mode(self):
self.mock_browse_button_to_return_files(["C:/dir1/file1.nxs"])
self.mock_model_to_load_workspaces([[1]], [1234], ["C:/dir1/file1.nxs"])
self.view.set_file_edit = mock.Mock()
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs"])
self.assertEqual(self.model.loaded_workspaces, [[1]])
self.assertEqual(self.model.loaded_runs, [1234])
self.view.set_file_edit.assert_called_once_with("C:/dir1/file1.nxs", mock.ANY)
@run_test_with_and_without_threading
def test_single_file_from_user_input_loaded_into_model_and_view_in_single_file_mode(self):
self.view.set_file_edit = mock.Mock()
self.mock_model_to_load_workspaces([[1]], [1234], ["C:/dir1/file1.nxs"])
self.mock_user_input_text("C:/dir1/file1.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/file1.nxs"])
self.assertEqual(self.model.loaded_workspaces, [[1]])
self.assertEqual(self.model.loaded_runs, [1234])
self.view.set_file_edit.assert_called_once_with("C:/dir1/file1.nxs", mock.ANY)
@run_test_with_and_without_threading
def test_that_if_invalid_file_selected_in_browser_view_does_not_change(self):
self.mock_browse_button_to_return_files(["not_a_file"])
self.mock_model_to_load_workspaces([[1]], [1234], ["not_a_file"])
self.view.set_file_edit = mock.Mock()
self.view.reset_edit_to_cached_value = mock.Mock()
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=self.load_failure)
set_file_edit_count = self.view.set_file_edit.call_count
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.set_file_edit.call_count, set_file_edit_count)
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 0)
@run_test_with_and_without_threading
def test_that_view_reverts_to_previous_text_if_users_supplies_invalid_text(self):
self.load_workspaces_into_model_and_view_from_browse([[1]], [1234], ["C:/dir1/EMU0001234.nxs"])
invalid_user_input = ["some random text", "1+1=2", "..."]
call_count = self.view.reset_edit_to_cached_value.call_count
for invalid_text in invalid_user_input:
call_count += 1
self.view.get_file_edit_text = mock.Mock(return_value=invalid_text)
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, call_count)
@run_test_with_and_without_threading
def test_that_model_and_interface_revert_to_previous_values_if_load_fails_from_browse(self):
self.load_workspaces_into_model_and_view_from_browse([[1]], [1234], ["C:/dir1/EMU0001234.nxs"])
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=self.load_failure)
self.presenter.on_browse_button_clicked()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/EMU0001234.nxs"])
self.assertEqual(self.model.loaded_workspaces, [[1]])
self.assertEqual(self.model.loaded_runs, [1234])
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 0)
self.assertEqual(self.view.set_file_edit.call_args[0][0], "C:/dir1/EMU0001234.nxs")
@run_test_with_and_without_threading
def test_that_model_and_interface_revert_to_previous_values_if_load_fails_from_user_input(self):
self.load_workspaces_into_model_and_view_from_browse([[1]], [1234], ["C:/dir1/EMU0001234.nxs"])
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=self.load_failure)
self.view.set_file_edit("C:\dir2\EMU000123.nxs")
self.presenter.handle_file_changed_by_user()
self.wait_for_thread(self.presenter._load_thread)
self.assertEqual(self.model.loaded_filenames, ["C:/dir1/EMU0001234.nxs"])
self.assertEqual(self.model.loaded_workspaces, [[1]])
self.assertEqual(self.model.loaded_runs, [1234])
self.assertEqual(self.view.reset_edit_to_cached_value.call_count, 1)
def setUp(self):
self._qapp = mock_widget.mockQapp()
self.view = BrowseFileWidgetView()
