def test_import_upload_step_failed(self, render_failure_message): """ Assert that the CLI properly informs the user when an importer reports failure upon importing a unit to the repository. """ response_body = {'success_flag': False, 'summary': 'An error message.'} response = mock.MagicMock() response.response_body = response_body response.is_async = mock.MagicMock(return_value=False) upload_manager = mock.MagicMock() upload_manager.import_upload = mock.MagicMock(return_value=response) upload_ids = ['an_id'] upload.perform_upload(self.context, upload_manager, upload_ids) render_failure_message.assert_called_once_with('... failed: %s' % response_body['summary'])
def test_import_upload_step_succeeded(self, write, render_failure_message): """ Assert that the CLI properly informs the user when an importer reports success upon importing a unit to the repository. """ response_body = {'success_flag': True, 'summary': None} response = mock.MagicMock() response.response_body = response_body response.is_async = mock.MagicMock(return_value=False) upload_manager = mock.MagicMock() upload_manager.import_upload = mock.MagicMock(return_value=response) upload_ids = ['an_id'] upload.perform_upload(self.context, upload_manager, upload_ids) write.assert_any_call('... completed', tag='import_upload_success') # No errors should have been rendered self.assertEqual(render_failure_message.call_count, 0)