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)