def test_set_progress_with_exception(self, mock_call):
# Setup
self.report_id = 'test-report'
self.mixin = mixins.StatusMixin(self.report_id, mixins.ImporterConduitException)
self.mixin.task_id = 'test_id'
mock_call.side_effect = Exception()
# Test
self.assertRaises(mixins.ImporterConduitException, self.mixin.set_progress, 'foo')
def test_set_progress_no_task(self, mock_get_task_id, mock_task_status_objects):
# Setup
mock_get_task_id.return_value = None
self.mixin = mixins.StatusMixin('', mixins.ImporterConduitException)
# Test
status = 'status'
self.mixin.set_progress(status)
# Verify
self.assertFalse(mock_task_status_objects.called)
def test_set_progress(self, mock_get_task_id, mock_task_status_objects):
# Setup
self.report_id = 'test-report'
task_id = 'test-id'
mock_get_task_id.return_value = task_id
test_task_documents = mock.Mock()
mock_task_status_objects.return_value = test_task_documents
self.mixin = mixins.StatusMixin(self.report_id, mixins.ImporterConduitException)
# Test
status = 'status'
self.mixin.set_progress(status)
# Verify
mock_task_status_objects.assert_called_with(task_id=task_id)
self.assertEqual(1, test_task_documents.update_one.call_count)
test_task_documents.update_one.assert_called_with(
set__progress_report={'test-report': 'status'})
def test_set_progress(self, mock_get_task_id, mock_update):
# Setup
self.report_id = 'test-report'
task_id = 'test-id'
mock_get_task_id.return_value = task_id
self.mixin = mixins.StatusMixin(self.report_id, mixins.ImporterConduitException)
# Test
status = 'status'
self.mixin.set_progress(status)
# Verify
expected_report = {
self.report_id : status,
}
self.assertEqual(1, mock_update.call_count)
delta = {'progress_report': expected_report}
call_args = mock_update.call_args[0]
self.assertEqual(call_args[0], task_id)
self.assertEqual(call_args[1], delta)
def setUp(self):
manager_factory.initialize()
self.report_id = 'test-report'
self.mixin = mixins.StatusMixin(self.report_id,
mixins.ImporterConduitException)
