def test_error_callback(self): exception_string = "test string" fig = figure() fig.canvas.draw = Mock() mock_view = Mock() presenter = PlotConfigDialogPresenter(fig, mock_view) presenter.error_callback(exception_string) mock_view.set_error_text.assert_called_with(exception_string) self.assertTrue(presenter.error_state)
def test_apply_properties_calls_success_callback_on_canvas_draw_success( self): fig = figure() fig.canvas.draw = Mock() mock_view = Mock() presenter = PlotConfigDialogPresenter(fig, mock_view) presenter.success_callback = Mock() presenter.error_callback = Mock() presenter.apply_properties() presenter.success_callback.assert_called() presenter.error_callback.assert_not_called()
def test_apply_properties_calls_error_callback_when_exception_raised_in_canvas_draw( self): canvas_draw_exception = Exception("Exception in canvas.draw") def raise_(): raise canvas_draw_exception fig = figure() fig.canvas.draw = Mock(side_effect=lambda: raise_()) mock_view = Mock() presenter = PlotConfigDialogPresenter(fig, mock_view) presenter.success_callback = Mock() presenter.error_callback = Mock() mock_presenters = [Mock(), Mock(), Mock(), Mock()] presenter.tab_widget_presenters = mock_presenters presenter.apply_properties() presenter.error_callback.assert_called_with(str(canvas_draw_exception)) presenter.success_callback.assert_not_called() for mock in presenter.tab_widget_presenters: mock.update_view.assert_not_called()