def test_get_rpm_units_skip(self):
"""
Test that when rpm types are in the skip list, they are actually skipped
"""
# Setup
mock_conduit = mock.Mock(spec=RepoPublishConduit)
skip_list = (ids.TYPE_ID_RPM, ids.TYPE_ID_SRPM, ids.TYPE_ID_DRPM)
# Test
export_utils.get_rpm_units(mock_conduit, skip_list)
self.assertEqual(0, mock_conduit.get_units.call_count)
def test_get_rpm_units(self):
"""
Test that the conduit is called for each type id
"""
# Setup
mock_conduit = mock.Mock(spec=RepoPublishConduit)
mock_conduit.get_units.return_value = ['unit']
# Test
rpm_units = export_utils.get_rpm_units(mock_conduit)
call_list = mock_conduit.get_units.call_args_list
self.assertEqual(3, len(rpm_units))
self.assertEqual(3, mock_conduit.get_units.call_count)
self.assertEqual([models.RPM.TYPE], call_list[0][1]['criteria'].type_ids)
self.assertEqual([models.SRPM.TYPE], call_list[1][1]['criteria'].type_ids)
self.assertEqual([models.DRPM.TYPE], call_list[2][1]['criteria'].type_ids)
