class VOSpaceErrorHandlerTest(unittest.TestCase): def setUp(self): app = Mock(spec=ValidationApplication) view = Mock(spec=ApplicationView) app.get_view.return_value = view self.downloadable_item = Mock(spec=DownloadRequest) self.error_handler = DownloadErrorHandler(app) self.view = view def test_handle_certificate_problem(self): message = "Your certificate is expired." error = OSError(message) error.errno = errno.EACCES self.error_handler.handle_error(error, self.downloadable_item) self.view.show_certificate_dialog.assert_called_once_with( self.error_handler, message) def test_handle_connection_timeout(self): message = "Connection timed out." error = IOError(message) error.errno = errno.ECONNREFUSED self.error_handler.handle_error(error, self.downloadable_item) self.view.show_retry_download_dialog.assert_called_once_with( self.error_handler, message)