def test_rpm_two(self): self.conduit.get_units = mock.MagicMock( spec_set=self.conduit.get_units, side_effect=lambda criteria: self.rpms if models.RPM.TYPE in criteria.type_ids else []) self.conduit.remove_unit = mock.MagicMock(spec_set=self.conduit.remove_unit) purge.remove_old_versions(2, self.conduit) self.conduit.remove_unit.assert_called_once_with(self.rpms[0])
def test_rpm_one(self): self.conduit.get_units = mock.MagicMock( spec_set=self.conduit.get_units, side_effect=lambda criteria: self.rpms if models.RPM.TYPE in criteria.type_ids else []) self.conduit.remove_unit = mock.MagicMock(spec_set=self.conduit.remove_unit) purge.remove_old_versions(1, self.conduit) self.conduit.remove_unit.assert_any_call(self.rpms[0]) self.conduit.remove_unit.assert_any_call(self.rpms[1]) self.assertEqual(self.conduit.remove_unit.call_count, 2)
def test_drpm_one(self): catalog = mock.Mock() self.conduit.get_units = mock.MagicMock( spec_set=self.conduit.get_units, side_effect=lambda criteria: self.drpms if ids.TYPE_ID_DRPM in criteria.type_ids else [] ) self.conduit.remove_unit = mock.MagicMock(spec_set=self.conduit.remove_unit) purge.remove_old_versions(1, self.conduit, catalog) self.conduit.remove_unit.assert_any_call(self.drpms[0]) self.conduit.remove_unit.assert_any_call(self.drpms[1]) self.assertEqual(self.conduit.remove_unit.call_count, 2)