def setUp(self):
self.view = mock.Mock(spec=PlotSelectorView)
self.view.get_filter_text = mock.Mock(return_value="")
self.model = mock.Mock(spec=PlotSelectorModel)
self.model.get_plot_list = mock.Mock(return_value=[0, 1, 2, 42])
self.model.get_plot_name_from_number = mock.Mock(side_effect=self.side_effect_plot_name)
self.presenter = PlotSelectorPresenter(None, self.view, self.model)
self.presenter.widget = self.view
self.presenter.model = self.model
# Ignore calls during the setup
self.view.reset_mock()
self.model.reset_mock()
def setUp(self):
self.widget = mock.Mock(spec=PlotSelectorView)
self.widget.get_filter_text = mock.Mock(return_value="")
self.model = mock.Mock(spec=PlotSelectorModel)
self.model.configure_mock(
plot_list=["Plot1", "Plot2", "Plot3", "Graph99"])
self.presenter = PlotSelectorPresenter(None, self.widget, self.model)
self.presenter.widget = self.widget
self.presenter.model = self.model
# Ignore calls during the setup
self.widget.reset_mock()
self.model.reset_mock()
def __init__(self, parent):
super(PlotSelector, self).__init__(parent)
plot_selector_presenter = PlotSelectorPresenter(GlobalFigureManager)
# layout
self.plot_selector_widget = plot_selector_presenter.widget
layout = QVBoxLayout()
layout.addWidget(self.plot_selector_widget)
self.setLayout(layout)
def setUp(self):
self.view = mock.Mock(spec=PlotSelectorView)
self.view.get_filter_text = mock.Mock(return_value="")
self.model = mock.Mock(spec=PlotSelectorModel)
self.model.get_plot_list = mock.Mock(return_value=[0, 1, 2, 42])
self.model.get_plot_name_from_number = mock.Mock(side_effect=self.side_effect_plot_name)
self.presenter = PlotSelectorPresenter(None, self.view, self.model)
self.presenter.widget = self.view
self.presenter.model = self.model
# Ignore calls during the setup
self.view.reset_mock()
self.model.reset_mock()
class PlotSelectorPresenterTest(unittest.TestCase):
def side_effect_plot_name(self, plot_number):
if plot_number in [0, 101, 102, 103]:
return "Plot1"
if plot_number == 1:
return "Plot2"
if plot_number == 2:
return "Plot3"
if plot_number == 42:
return "Graph99"
return None
def setUp(self):
self.view = mock.Mock(spec=PlotSelectorView)
self.view.get_filter_text = mock.Mock(return_value="")
self.model = mock.Mock(spec=PlotSelectorModel)
self.model.get_plot_list = mock.Mock(return_value=[0, 1, 2, 42])
self.model.get_plot_name_from_number = mock.Mock(side_effect=self.side_effect_plot_name)
self.presenter = PlotSelectorPresenter(None, self.view, self.model)
self.presenter.widget = self.view
self.presenter.model = self.model
# Ignore calls during the setup
self.view.reset_mock()
self.model.reset_mock()
def convert_list_to_calls(self, list_to_convert):
call_list = []
for item in list_to_convert:
call_list.append(mock.call(item))
return call_list
# ------------------------ Plot Updates ------------------------
def test_plot_list_update(self):
self.presenter.update_plot_list()
self.assertEqual(self.model.get_plot_list.call_count, 1)
self.view.set_plot_list.assert_called_once_with([0, 1, 2, 42])
def test_append_to_plot_list_calls_update_in_model_and_view(self):
self.presenter.append_to_plot_list(42)
self.view.append_to_plot_list.assert_called_once_with(42)
def test_remove_from_plot_list_calls_update_in_model_and_view(self):
self.presenter.remove_from_plot_list(42)
self.view.remove_from_plot_list.assert_called_once_with(42)
def test_rename_in_plot_list_calls_update_in_model_and_view(self):
self.presenter.rename_in_plot_list(42, "NewName")
self.view.rename_in_plot_list.assert_called_once_with(42, "NewName")
# ----------------------- Plot Filtering ------------------------
def test_no_filtering_displays_all_plots(self):
self.presenter.filter_text_changed()
self.view.unhide_all_plots.assert_called_once_with()
def filtering_calls_filter_on_view(self):
self.view.get_filter_text = mock.Mock(return_value="Plot1")
self.presenter.filter_text_changed()
self.view.filter_plot_list.assert_called_once_with("Plot1")
def test_plots_filtered_on_full_name(self):
self.view.get_filter_text = mock.Mock(return_value="Plot1")
self.assertTrue(self.presenter.is_shown_by_filter(0)) # Plot1
self.assertFalse(self.presenter.is_shown_by_filter(1)) # Plot 2
def test_plots_filtered_on_substring(self):
self.view.get_filter_text = mock.Mock(return_value="lot")
self.assertTrue(self.presenter.is_shown_by_filter(0)) # Plot1
self.assertFalse(self.presenter.is_shown_by_filter(42)) # Graph99
def test_filtering_case_invariant(self):
self.view.get_filter_text = mock.Mock(return_value="pLOT1")
self.assertTrue(self.presenter.is_shown_by_filter(0)) # Plot1
self.assertFalse(self.presenter.is_shown_by_filter(1)) # Plot2
# ------------------------ Plot Showing ------------------------
def test_show_single_plot_shows_it(self):
self.view.get_currently_selected_plot_number = mock.Mock(return_value=1)
self.presenter.show_single_selected()
self.model.show_plot.assert_called_once_with(1)
def test_show_multiple_plots_shows_them(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[1, 2])
self.presenter.show_multiple_selected()
self.assertEqual(self.model.show_plot.mock_calls[0], mock.call(1))
self.assertEqual(self.model.show_plot.mock_calls[1], mock.call(2))
def test_set_active_font_sets_active_font_in_view(self):
self.view.active_plot_number = 1
self.presenter.set_active_font(2)
# 2 calls, one to set the plot number 1 to normal, one to
# set plot number 2 to bold
self.assertEqual(self.view.set_active_font.mock_calls[0], mock.call(1, False))
self.assertEqual(self.view.set_active_font.mock_calls[1], mock.call(2, True))
self.assertEqual(self.view.active_plot_number, 2)
# ------------------------ Plot Hiding -------------------------
def test_hide_multiple_plots_calls_hide_in_model(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[1, 2])
self.presenter.hide_selected_plots()
self.assertEquals(self.model.hide_plot.mock_calls[0], mock.call(1))
self.assertEquals(self.model.hide_plot.mock_calls[1], mock.call(2))
def test_toggle_plot_visibility_for_visible_plot(self):
self.model.is_visible = mock.Mock(return_value=True)
self.presenter.toggle_plot_visibility(42)
self.model.hide_plot.assert_called_once_with(42)
self.view.set_visibility_icon.assert_called_once_with(42, True)
def test_toggle_plot_visibility_for_hidden_plot(self):
self.model.is_visible = mock.Mock(return_value=False)
self.presenter.toggle_plot_visibility(42)
self.model.show_plot.assert_called_once_with(42)
self.view.set_visibility_icon.assert_called_once_with(42, False)
# ------------------------ Plot Renaming ------------------------
def test_rename_figure_calls_rename_in_model(self):
self.presenter.rename_figure(0, "NewName")
self.model.rename_figure.assert_called_once_with(0, "NewName")
def test_rename_figure_raising_a_value_error_undoes_rename_in_view(self):
self.model.rename_figure.side_effect = ValueError("Some problem")
self.presenter.rename_figure(0, "NewName")
self.model.rename_figure.assert_called_once_with(0, "NewName")
self.view.rename_in_plot_list.assert_called_once_with(0, "NewName")
# ------------------------ Plot Closing -------------------------
def test_close_action_single_plot(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[42])
self.presenter.close_action_called()
self.model.close_plot.assert_called_once_with(42)
def test_close_action_multiple_plots(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[42, 43])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 2)
self.model.close_plot.assert_has_calls(self.convert_list_to_calls([42, 43]),
any_order=True)
def test_close_action_with_model_call_raising_value_error(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[42])
self.model.close_plot.side_effect = ValueError("Some problem")
self.presenter.close_action_called()
self.model.close_plot.assert_called_once_with(42)
def test_close_action_with_no_plots_open(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[])
self.model.configure_mock(plot_list=[])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 0)
def test_close_action_with_no_plots_selected(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 0)
def test_close_single_plot_called(self):
self.presenter.close_single_plot("Plot2")
self.model.close_plot.assert_called_once_with("Plot2")
# ----------------------- Plot Sorting --------------------------
def test_set_sort_order_to_ascending_calls_view_update(self):
self.presenter.set_sort_order(is_ascending=True)
self.view.set_sort_order.assert_called_once_with(True)
def test_set_sort_order_to_descending_calls_view_update(self):
self.presenter.set_sort_order(is_ascending=False)
self.view.set_sort_order.assert_called_once_with(False)
def test_set_sort_type_to_name(self):
self.view.sort_type = mock.Mock(return_value=Column.Name)
self.presenter.set_sort_type(Column.Name)
self.view.set_sort_type.assert_called_once_with(Column.Name)
self.model.last_active_values.assert_not_called()
self.view.set_last_active_values.assert_not_called()
def test_set_sort_type_to_last_active(self):
self.model.last_active_values = mock.Mock(return_value={0: 1, 1: 2})
self.view.sort_type = mock.Mock(return_value=Column.LastActive)
self.presenter.set_sort_type(Column.LastActive)
self.view.set_sort_type.assert_called_once_with(Column.LastActive)
self.view.set_last_active_values.assert_called_once_with({0: 1,
1: 2})
def test_set_last_active_values_with_sorting_by_last_active(self):
self.model.last_active_values = mock.Mock(return_value={0: 1, 1: 2})
self.view.sort_type = mock.Mock(return_value=Column.LastActive)
self.presenter.update_last_active_order()
self.view.set_last_active_values.assert_called_once_with({0: 1,
1: 2})
def test_set_last_active_values_with_sorting_by_name_does_nothing(self):
self.model.last_active_order = mock.Mock(return_value={0: 1, 1: 2})
self.view.sort_type = mock.Mock(return_value=Column.Name)
self.presenter.update_last_active_order()
self.model.last_active_order.assert_not_called()
self.view.set_last_active_values.assert_not_called()
def test_get_initial_last_active_value(self):
self.assertEqual(self.presenter.get_initial_last_active_value(0), "_Plot1")
def test_get_renamed_last_active_value_for_numeric_old_value(self):
self.assertEqual(self.presenter.get_renamed_last_active_value(1, "23"), "23")
def test_get_renamed_last_active_value_for_never_shown_value(self):
self.assertEqual(self.presenter.get_renamed_last_active_value(0, "_Plot1"), "_Plot1")
# ---------------------- Plot Exporting -------------------------
def test_exporting_single_plot_generates_correct_filename(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[0])
self.view.get_file_name_for_saving = mock.Mock(return_value='/home/Documents/Plot1')
self.presenter.export_plots_called('.xyz')
self.model.export_plot.assert_called_once_with(0, '/home/Documents/Plot1.xyz')
def test_exporting_single_plot_with_extension_given_in_file_name(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[0])
self.view.get_file_name_for_saving = mock.Mock(return_value='/home/Documents/Plot1.xyz')
self.presenter.export_plots_called('.xyz')
self.model.export_plot.assert_called_once_with(0, '/home/Documents/Plot1.xyz')
def test_exporting_multiple_plots_generates_correct_filename(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[0, 1, 2])
self.view.get_directory_name_for_saving = mock.Mock(return_value='/home/Documents')
self.presenter.export_plots_called('.xyz')
for i in range(len(self.model.export_plot.mock_calls)):
self.assertEqual(self.model.export_plot.mock_calls[i],
mock.call(i, os.path.join('/home/Documents', 'Plot{}.xyz'.format(i+1))))
def test_exporting_multiple_plots_with_repeated_plot_names_generates_unique_names(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[0, 101, 102, 103])
self.view.get_directory_name_for_saving = mock.Mock(return_value='/home/Documents')
self.presenter.export_plots_called('.xyz')
self.assertEqual(self.model.export_plot.mock_calls[0],
mock.call(0, os.path.join('/home/Documents', 'Plot1.xyz')))
for i in range(1, len(self.model.export_plot.mock_calls)):
self.assertEqual(self.model.export_plot.mock_calls[i],
mock.call(100 + i, os.path.join('/home/Documents', 'Plot1 ({}).xyz'.format(i))))
def test_exporting_multiple_plots_with_special_characters_in_file_name(self):
for character in '<>:"/|\\?*':
self.run_special_character_test(character)
def run_special_character_test(self, special_character):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[0, 1])
self.model.get_plot_name_from_number = mock.Mock(return_value='Plot' + special_character + '1')
self.view.get_directory_name_for_saving = mock.Mock(return_value='/home/Documents')
self.presenter.export_plots_called('.xyz')
self.assertEqual(self.model.export_plot.mock_calls[0],
mock.call(0, os.path.join('/home/Documents', 'Plot-1.xyz')))
self.assertEqual(self.model.export_plot.mock_calls[1],
mock.call(1, os.path.join('/home/Documents', 'Plot-1 (1).xyz')))
class PlotSelectorPresenterTest(unittest.TestCase):
def side_effect_plot_name(self, plot_number):
if plot_number in [0, 101, 102, 103]:
return "Plot1"
if plot_number == 1:
return "Plot2"
if plot_number == 2:
return "Plot3"
if plot_number == 42:
return "Graph99"
return None
def setUp(self):
self.view = mock.Mock(spec=PlotSelectorView)
self.view.get_filter_text = mock.Mock(return_value="")
self.model = mock.Mock(spec=PlotSelectorModel)
self.model.get_plot_list = mock.Mock(return_value=[0, 1, 2, 42])
self.model.get_plot_name_from_number = mock.Mock(
side_effect=self.side_effect_plot_name)
self.presenter = PlotSelectorPresenter(None, self.view, self.model)
self.presenter.widget = self.view
self.presenter.model = self.model
# Ignore calls during the setup
self.view.reset_mock()
self.model.reset_mock()
def convert_list_to_calls(self, list_to_convert):
call_list = []
for item in list_to_convert:
call_list.append(mock.call(item))
return call_list
# ------------------------ Plot Updates ------------------------
def test_plot_list_update(self):
self.presenter.update_plot_list()
self.assertEqual(self.model.get_plot_list.call_count, 1)
self.view.set_plot_list.assert_called_once_with([0, 1, 2, 42])
def test_append_to_plot_list_calls_update_in_model_and_view(self):
self.presenter.append_to_plot_list(42)
self.view.append_to_plot_list.assert_called_once_with(42)
def test_remove_from_plot_list_calls_update_in_model_and_view(self):
self.presenter.remove_from_plot_list(42)
self.view.remove_from_plot_list.assert_called_once_with(42)
def test_rename_in_plot_list_calls_update_in_model_and_view(self):
self.presenter.rename_in_plot_list(42, "NewName")
self.view.rename_in_plot_list.assert_called_once_with(42, "NewName")
# ----------------------- Plot Filtering ------------------------
def test_no_filtering_displays_all_plots(self):
self.presenter.filter_text_changed()
self.view.unhide_all_plots.assert_called_once_with()
def filtering_calls_filter_on_view(self):
self.view.get_filter_text = mock.Mock(return_value="Plot1")
self.presenter.filter_text_changed()
self.view.filter_plot_list.assert_called_once_with("Plot1")
def test_plots_filtered_on_full_name(self):
self.view.get_filter_text = mock.Mock(return_value="Plot1")
self.assertTrue(self.presenter.is_shown_by_filter(0)) # Plot1
self.assertFalse(self.presenter.is_shown_by_filter(1)) # Plot 2
def test_plots_filtered_on_substring(self):
self.view.get_filter_text = mock.Mock(return_value="lot")
self.assertTrue(self.presenter.is_shown_by_filter(0)) # Plot1
self.assertFalse(self.presenter.is_shown_by_filter(42)) # Graph99
def test_filtering_case_invariant(self):
self.view.get_filter_text = mock.Mock(return_value="pLOT1")
self.assertTrue(self.presenter.is_shown_by_filter(0)) # Plot1
self.assertFalse(self.presenter.is_shown_by_filter(1)) # Plot2
# ------------------------ Plot Showing ------------------------
def test_show_single_plot_shows_it(self):
self.view.get_currently_selected_plot_number = mock.Mock(
return_value=1)
self.presenter.show_single_selected()
self.model.show_plot.assert_called_once_with(1)
def test_show_multiple_plots_shows_them(self):
self.view.get_all_selected_plot_numbers = mock.Mock(
return_value=[1, 2])
self.presenter.show_multiple_selected()
self.assertEqual(self.model.show_plot.mock_calls[0], mock.call(1))
self.assertEqual(self.model.show_plot.mock_calls[1], mock.call(2))
def test_set_active_font_sets_active_font_in_view(self):
self.view.active_plot_number = 1
self.presenter.set_active_font(2)
# 2 calls, one to set the plot number 1 to normal, one to
# set plot number 2 to bold
self.assertEqual(self.view.set_active_font.mock_calls[0],
mock.call(1, False))
self.assertEqual(self.view.set_active_font.mock_calls[1],
mock.call(2, True))
self.assertEqual(self.view.active_plot_number, 2)
# ------------------------ Plot Hiding -------------------------
def test_hide_multiple_plots_calls_hide_in_model(self):
self.view.get_all_selected_plot_numbers = mock.Mock(
return_value=[1, 2])
self.presenter.hide_selected_plots()
self.assertEqual(self.model.hide_plot.mock_calls[0], mock.call(1))
self.assertEqual(self.model.hide_plot.mock_calls[1], mock.call(2))
def test_toggle_plot_visibility_for_visible_plot(self):
self.model.is_visible = mock.Mock(return_value=True)
self.presenter.toggle_plot_visibility(42)
self.model.hide_plot.assert_called_once_with(42)
self.view.set_visibility_icon.assert_called_once_with(42, True)
def test_toggle_plot_visibility_for_hidden_plot(self):
self.model.is_visible = mock.Mock(return_value=False)
self.presenter.toggle_plot_visibility(42)
self.model.show_plot.assert_called_once_with(42)
self.view.set_visibility_icon.assert_called_once_with(42, False)
# ------------------------ Plot Renaming ------------------------
def test_rename_figure_calls_rename_in_model(self):
self.presenter.rename_figure(0, "NewName")
self.model.rename_figure.assert_called_once_with(0, "NewName")
def test_rename_figure_raising_a_value_error_undoes_rename_in_view(self):
self.model.rename_figure.side_effect = ValueError("Some problem")
self.presenter.rename_figure(0, "NewName")
self.model.rename_figure.assert_called_once_with(0, "NewName")
self.view.rename_in_plot_list.assert_called_once_with(0, "NewName")
# ------------------------ Plot Closing -------------------------
def test_close_action_single_plot(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[42])
self.presenter.close_action_called()
self.model.close_plot.assert_called_once_with(42)
def test_close_action_multiple_plots(self):
self.view.get_all_selected_plot_numbers = mock.Mock(
return_value=[42, 43])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 2)
self.model.close_plot.assert_has_calls(self.convert_list_to_calls(
[42, 43]),
any_order=True)
def test_close_action_with_model_call_raising_value_error(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[42])
self.model.close_plot.side_effect = ValueError("Some problem")
self.presenter.close_action_called()
self.model.close_plot.assert_called_once_with(42)
def test_close_action_with_no_plots_open(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[])
self.model.configure_mock(plot_list=[])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 0)
def test_close_action_with_no_plots_selected(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 0)
def test_close_single_plot_called(self):
self.presenter.close_single_plot("Plot2")
self.model.close_plot.assert_called_once_with("Plot2")
# ----------------------- Plot Sorting --------------------------
def test_set_sort_order_to_ascending_calls_view_update(self):
self.presenter.set_sort_order(is_ascending=True)
self.view.set_sort_order.assert_called_once_with(True)
def test_set_sort_order_to_descending_calls_view_update(self):
self.presenter.set_sort_order(is_ascending=False)
self.view.set_sort_order.assert_called_once_with(False)
def test_set_sort_type_to_name(self):
self.view.sort_type = mock.Mock(return_value=Column.Name)
self.presenter.set_sort_type(Column.Name)
self.view.set_sort_type.assert_called_once_with(Column.Name)
self.model.last_active_values.assert_not_called()
self.view.set_last_active_values.assert_not_called()
def test_set_sort_type_to_last_active(self):
self.model.last_active_values = mock.Mock(return_value={0: 1, 1: 2})
self.view.sort_type = mock.Mock(return_value=Column.LastActive)
self.presenter.set_sort_type(Column.LastActive)
self.view.set_sort_type.assert_called_once_with(Column.LastActive)
self.view.set_last_active_values.assert_called_once_with({0: 1, 1: 2})
def test_set_last_active_values_with_sorting_by_last_active(self):
self.model.last_active_values = mock.Mock(return_value={0: 1, 1: 2})
self.view.sort_type = mock.Mock(return_value=Column.LastActive)
self.presenter.update_last_active_order()
self.view.set_last_active_values.assert_called_once_with({0: 1, 1: 2})
def test_set_last_active_values_with_sorting_by_name_does_nothing(self):
self.model.last_active_order = mock.Mock(return_value={0: 1, 1: 2})
self.view.sort_type = mock.Mock(return_value=Column.Name)
self.presenter.update_last_active_order()
self.model.last_active_order.assert_not_called()
self.view.set_last_active_values.assert_not_called()
def test_get_initial_last_active_value(self):
self.assertEqual(self.presenter.get_initial_last_active_value(0),
"_Plot1")
def test_get_renamed_last_active_value_for_numeric_old_value(self):
self.assertEqual(self.presenter.get_renamed_last_active_value(1, "23"),
"23")
def test_get_renamed_last_active_value_for_never_shown_value(self):
self.assertEqual(
self.presenter.get_renamed_last_active_value(0, "_Plot1"),
"_Plot1")
# ---------------------- Plot Exporting -------------------------
def test_exporting_single_plot_generates_correct_filename(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[0])
self.view.get_file_name_for_saving = mock.Mock(
return_value='/home/Documents/Plot1')
self.presenter.export_plots_called('.xyz')
self.model.export_plot.assert_called_once_with(
0, '/home/Documents/Plot1.xyz')
def test_exporting_single_plot_with_extension_given_in_file_name(self):
self.view.get_all_selected_plot_numbers = mock.Mock(return_value=[0])
self.view.get_file_name_for_saving = mock.Mock(
return_value='/home/Documents/Plot1.xyz')
self.presenter.export_plots_called('.xyz')
self.model.export_plot.assert_called_once_with(
0, '/home/Documents/Plot1.xyz')
def test_exporting_multiple_plots_generates_correct_filename(self):
self.view.get_all_selected_plot_numbers = mock.Mock(
return_value=[0, 1, 2])
self.view.get_directory_name_for_saving = mock.Mock(
return_value='/home/Documents')
self.presenter.export_plots_called('.xyz')
for i in range(len(self.model.export_plot.mock_calls)):
self.assertEqual(
self.model.export_plot.mock_calls[i],
mock.call(
i,
os.path.join('/home/Documents',
'Plot{}.xyz'.format(i + 1))))
def test_exporting_multiple_plots_with_repeated_plot_names_generates_unique_names(
self):
self.view.get_all_selected_plot_numbers = mock.Mock(
return_value=[0, 101, 102, 103])
self.view.get_directory_name_for_saving = mock.Mock(
return_value='/home/Documents')
self.presenter.export_plots_called('.xyz')
self.assertEqual(
self.model.export_plot.mock_calls[0],
mock.call(0, os.path.join('/home/Documents', 'Plot1.xyz')))
for i in range(1, len(self.model.export_plot.mock_calls)):
self.assertEqual(
self.model.export_plot.mock_calls[i],
mock.call(
100 + i,
os.path.join('/home/Documents',
'Plot1 ({}).xyz'.format(i))))
def test_exporting_multiple_plots_with_special_characters_in_file_name(
self):
for character in '<>:"/|\\?*':
self.run_special_character_test(character)
def run_special_character_test(self, special_character):
self.view.get_all_selected_plot_numbers = mock.Mock(
return_value=[0, 1])
self.model.get_plot_name_from_number = mock.Mock(
return_value='Plot' + special_character + '1')
self.view.get_directory_name_for_saving = mock.Mock(
return_value='/home/Documents')
self.presenter.export_plots_called('.xyz')
self.assertEqual(
self.model.export_plot.mock_calls[0],
mock.call(0, os.path.join('/home/Documents', 'Plot-1.xyz')))
self.assertEqual(
self.model.export_plot.mock_calls[1],
mock.call(1, os.path.join('/home/Documents', 'Plot-1 (1).xyz')))
class PlotSelectorPresenterTest(unittest.TestCase):
def setUp(self):
self.widget = mock.Mock(spec=PlotSelectorView)
self.widget.get_filter_text = mock.Mock(return_value="")
self.model = mock.Mock(spec=PlotSelectorModel)
self.model.configure_mock(
plot_list=["Plot1", "Plot2", "Plot3", "Graph99"])
self.presenter = PlotSelectorPresenter(None, self.widget, self.model)
self.presenter.widget = self.widget
self.presenter.model = self.model
# Ignore calls during the setup
self.widget.reset_mock()
self.model.reset_mock()
def convert_list_to_calls(self, list_to_convert):
call_list = []
for item in list_to_convert:
call_list.append(mock.call(item))
return call_list
def test_plot_list_update(self):
self.presenter.update_plot_list()
self.assertEqual(self.model.update_plot_list.call_count, 1)
self.widget.set_plot_list.assert_called_once_with(
["Plot1", "Plot2", "Plot3", "Graph99"])
def test_plot_list_update_with_filter_set(self):
self.widget.get_filter_text = mock.Mock(return_value="Graph99")
self.presenter.update_plot_list()
self.assertEqual(self.model.update_plot_list.call_count, 1)
self.widget.set_plot_list.assert_called_once_with(["Graph99"])
# ------------------------ Plot Closing -------------------------
def test_close_action_single_plot(self):
self.widget.get_all_selected_plot_names = mock.Mock(
return_value=["Plot1"])
self.presenter.close_action_called()
self.model.close_plot.assert_called_once_with("Plot1")
def test_close_action_multiple_plots(self):
self.widget.get_all_selected_plot_names = mock.Mock(
return_value=["Plot1", "Plot3"])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 2)
self.model.close_plot.assert_has_calls(self.convert_list_to_calls(
["Plot1", "Plot3"]),
any_order=True)
def test_close_action_with_no_plots_open(self):
self.widget.get_all_selected_plot_names = mock.Mock(return_value=[])
self.model.configure_mock(plot_list=[])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 0)
def test_close_action_with_no_plots_selected(self):
self.widget.get_all_selected_plot_names = mock.Mock(return_value=[])
self.presenter.close_action_called()
self.assertEqual(self.model.close_plot.call_count, 0)
# ----------------------- Plot Filtering ------------------------
def test_no_filtering_displays_all_plots(self):
self.presenter.filter_text_changed()
self.widget.set_plot_list.assert_called_once_with(
["Plot1", "Plot2", "Plot3", "Graph99"])
def test_plots_filtered_on_full_name(self):
self.widget.get_filter_text = mock.Mock(return_value="Plot1")
self.presenter.filter_text_changed()
self.widget.set_plot_list.assert_called_once_with(["Plot1"])
def test_plots_filtered_on_substring(self):
self.widget.get_filter_text = mock.Mock(return_value="lot")
self.presenter.filter_text_changed()
self.widget.set_plot_list.assert_called_once_with(
["Plot1", "Plot2", "Plot3"])
def test_filtering_case_invariant(self):
self.widget.get_filter_text = mock.Mock(return_value="pLOT1")
self.presenter.filter_text_changed()
self.widget.set_plot_list.assert_called_once_with(["Plot1"])
# ----------------------- Plot Selection ------------------------
def test_double_clicking_plot_brings_to_front(self):
self.widget.get_currently_selected_plot_name = mock.Mock(
return_value="Plot2")
self.presenter.list_double_clicked()
self.model.make_plot_active.assert_called_once_with("Plot2")