def test_get_total_ignore_filter(self):
step = UnitPublishStep("foo", ['bar', 'baz'])
step.association_filters = {'foo': 'bar'}
step.parent = Mock()
step.parent.repo.content_unit_counts.get.return_value = 1
total = step._get_total(ignore_filter=True)
self.assertEquals(2, total)
def test_get_with_association_filter(self, mock_manager_factory):
step = UnitPublishStep("foo", ['bar', 'baz'])
step.association_filters = {'foo': 'bar'}
find_by_criteria = mock_manager_factory.repo_unit_association_query_manager.return_value.\
find_by_criteria
find_by_criteria.return_value.count.return_value = 5
total = step._get_total()
criteria_object = find_by_criteria.call_args[0][0]
compare_dict(criteria_object.filters, {'foo': 'bar',
'unit_type_id': {'$in': ['bar', 'baz']}})
self.assertEquals(5, total)
def test_process_step_cancelled_mid_unit_processing(self, mock_get_units):
self.publisher.repo.content_unit_counts = {'FOO_TYPE': 2}
mock_get_units.return_value = ['cancel', 'bar_unit']
step = UnitPublishStep('foo_step', 'FOO_TYPE')
self.publisher.add_child(step)
step.process_unit = self._step_canceler
step.process()
self.assertEquals(step.state, reporting_constants.STATE_CANCELLED)
self.assertEquals(step.progress_successes, 1)
self.assertEquals(step.progress_failures, 0)
self.assertEquals(step.total_units, 2)
def test_process_step_single_unit_exception(self, mock_get_units):
self.publisher.repo.content_unit_counts = {'FOO_TYPE': 1}
mock_method = Mock(side_effect=Exception())
mock_get_units.return_value = ['mock_unit']
step = UnitPublishStep('foo_step', 'FOO_TYPE')
step.parent = self.publisher
step.process_unit = mock_method
self.assertRaises(Exception, step.process)
self.assertEquals(step.state, reporting_constants.STATE_FAILED)
self.assertEquals(step.progress_successes, 0)
self.assertEquals(step.progress_failures, 1)
self.assertEquals(step.total_units, 1)
mock_method.assert_called_once_with('mock_unit')
def test_process_step_no_units(self, mock_get_units):
self.publisher.repo.content_unit_counts = {'FOO_TYPE': 0}
mock_method = Mock()
mock_get_units.return_value = []
step = UnitPublishStep('foo_step', 'FOO_TYPE')
step.parent = self.publisher
step.process_unit = mock_method
step.process()
self.assertEquals(step.state, reporting_constants.STATE_COMPLETE)
self.assertFalse(mock_method.called)
def test_process_step_single_unit(self, mock_get_units):
self.publisher.repo.content_unit_counts = {'FOO_TYPE': 1}
mock_method = Mock()
mock_get_units.return_value = ['mock_unit']
step = UnitPublishStep('foo_step', 'FOO_TYPE')
step.parent = self.publisher
step.process_unit = mock_method
step.process()
self.assertEquals(step.state, reporting_constants.STATE_COMPLETE)
self.assertEquals(step.progress_successes, 1)
self.assertEquals(step.progress_failures, 0)
self.assertEquals(step.total_units, 1)
mock_method.assert_called_once_with('mock_unit')
def test_process_unit_with_no_work(self):
# Run the blank process unit to ensure no exceptions are raised
step = UnitPublishStep("foo", ['bar', 'baz'])
step.process_unit('foo')
def test_get_total_for_none(self):
step = UnitPublishStep("foo", ['bar', 'baz'])
step.parent = Mock()
step.parent.repo.content_unit_counts.get.return_value = 0
total = step._get_total()
self.assertEquals(0, total)
def test_is_skipped_dict_not_skipped(self):
step = UnitPublishStep("foo", 'bar')
step.config = PluginCallConfiguration(None, None)
self.assertFalse(step.is_skipped())
def test_is_skipped_dict(self):
step = UnitPublishStep("foo", 'bar')
step.config = PluginCallConfiguration(None, {'skip': {'bar': True, 'baz': True}})
self.assertTrue(step.is_skipped())
def test_process_step_skip_units(self):
self.publisher.config = PluginCallConfiguration(None, {'skip': ['FOO']})
step = UnitPublishStep('foo_step', 'FOO')
step.parent = self.publisher
step.process()
self.assertEquals(step.state, reporting_constants.STATE_SKIPPED)
def test_get_with_association_filter(self):
step = UnitPublishStep("foo", ['bar', 'baz'])
step.association_filters = {'foo': 'bar'}
total = step._get_total()
self.assertEquals(1, total)