Ejemplo n.º 1
0
    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])
Ejemplo n.º 2
0
    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])
Ejemplo n.º 3
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)
Ejemplo n.º 4
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)
Ejemplo n.º 5
0
    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)