Added purge script for soft deleted objects
This patchset implements the purge script as specified in its related blueprint: - The '--age-in-days' option allows to specify the number of days before expiry - The '--max-number' option allows us to specify a limit on the number of objects to delete - The '--audit-template' option allows you to only delete objects related to the specified audit template UUID or name - The '--dry-run' option to go through the purge procedure without actually deleting anything - The '--exclude-orphans' option which allows you to exclude from the purge any object that does not have a parent (i.e. and audit without a related audit template) A prompt has been added to also propose to narrow down the number of deletions to be below the specified limit. Change-Id: I3ce83ab95277c109df67a6b5b920a878f6e59d3f Implements: blueprint db-purge-engine
This commit is contained in:
Vincent Françoise
@@ -1,5 +1,5 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
# Copyright (c) 2015 b<>com
|
||||
# Copyright (c) 2016 b<>com
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
@@ -14,15 +14,18 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from mock import Mock
|
||||
from mock import patch
|
||||
import sys
|
||||
|
||||
import mock
|
||||
from oslo_config import cfg
|
||||
|
||||
from watcher.cmd import dbmanage
|
||||
from watcher.db import migration
|
||||
from watcher.tests.base import TestCase
|
||||
from watcher.db import purge
|
||||
from watcher.tests import base
|
||||
|
||||
|
||||
class TestDBManageRunApp(TestCase):
|
||||
class TestDBManageRunApp(base.TestCase):
|
||||
|
||||
scenarios = (
|
||||
("upgrade", {"command": "upgrade", "expected": "upgrade"}),
|
||||
@@ -32,15 +35,16 @@ class TestDBManageRunApp(TestCase):
|
||||
("version", {"command": "version", "expected": "version"}),
|
||||
("create_schema", {"command": "create_schema",
|
||||
"expected": "create_schema"}),
|
||||
("purge", {"command": "purge", "expected": "purge"}),
|
||||
("no_param", {"command": None, "expected": "upgrade"}),
|
||||
)
|
||||
|
||||
@patch.object(dbmanage, "register_sub_command_opts", Mock())
|
||||
@patch("watcher.cmd.dbmanage.service.prepare_service")
|
||||
@patch("watcher.cmd.dbmanage.sys")
|
||||
@mock.patch.object(dbmanage, "register_sub_command_opts", mock.Mock())
|
||||
@mock.patch("watcher.cmd.dbmanage.service.prepare_service")
|
||||
@mock.patch("watcher.cmd.dbmanage.sys")
|
||||
def test_run_db_manage_app(self, m_sys, m_prepare_service):
|
||||
# Patch command function
|
||||
m_func = Mock()
|
||||
m_func = mock.Mock()
|
||||
cfg.CONF.register_opt(cfg.SubCommandOpt("command"))
|
||||
cfg.CONF.command.func = m_func
|
||||
|
||||
@@ -53,9 +57,9 @@ class TestDBManageRunApp(TestCase):
|
||||
["watcher-db-manage", self.expected])
|
||||
|
||||
|
||||
class TestDBManageRunCommand(TestCase):
|
||||
class TestDBManageRunCommand(base.TestCase):
|
||||
|
||||
@patch.object(migration, "upgrade")
|
||||
@mock.patch.object(migration, "upgrade")
|
||||
def test_run_db_upgrade(self, m_upgrade):
|
||||
cfg.CONF.register_opt(cfg.StrOpt("revision"), group="command")
|
||||
cfg.CONF.set_default("revision", "dummy", group="command")
|
||||
@@ -63,7 +67,7 @@ class TestDBManageRunCommand(TestCase):
|
||||
|
||||
m_upgrade.assert_called_once_with("dummy")
|
||||
|
||||
@patch.object(migration, "downgrade")
|
||||
@mock.patch.object(migration, "downgrade")
|
||||
def test_run_db_downgrade(self, m_downgrade):
|
||||
cfg.CONF.register_opt(cfg.StrOpt("revision"), group="command")
|
||||
cfg.CONF.set_default("revision", "dummy", group="command")
|
||||
@@ -71,7 +75,7 @@ class TestDBManageRunCommand(TestCase):
|
||||
|
||||
m_downgrade.assert_called_once_with("dummy")
|
||||
|
||||
@patch.object(migration, "revision")
|
||||
@mock.patch.object(migration, "revision")
|
||||
def test_run_db_revision(self, m_revision):
|
||||
cfg.CONF.register_opt(cfg.StrOpt("message"), group="command")
|
||||
cfg.CONF.register_opt(cfg.StrOpt("autogenerate"), group="command")
|
||||
@@ -87,14 +91,85 @@ class TestDBManageRunCommand(TestCase):
|
||||
"dummy_message", "dummy_autogenerate"
|
||||
)
|
||||
|
||||
@patch.object(migration, "stamp")
|
||||
@mock.patch.object(migration, "stamp")
|
||||
def test_run_db_stamp(self, m_stamp):
|
||||
cfg.CONF.register_opt(cfg.StrOpt("revision"), group="command")
|
||||
cfg.CONF.set_default("revision", "dummy", group="command")
|
||||
dbmanage.DBCommand.stamp()
|
||||
|
||||
@patch.object(migration, "version")
|
||||
@mock.patch.object(migration, "version")
|
||||
def test_run_db_version(self, m_version):
|
||||
dbmanage.DBCommand.version()
|
||||
|
||||
self.assertEqual(1, m_version.call_count)
|
||||
|
||||
@mock.patch.object(purge, "PurgeCommand")
|
||||
def test_run_db_purge(self, m_purge_cls):
|
||||
m_purge = mock.Mock()
|
||||
m_purge_cls.return_value = m_purge
|
||||
m_purge_cls.get_audit_template_uuid.return_value = 'Some UUID'
|
||||
cfg.CONF.register_opt(cfg.IntOpt("age_in_days"), group="command")
|
||||
cfg.CONF.register_opt(cfg.IntOpt("max_number"), group="command")
|
||||
cfg.CONF.register_opt(cfg.StrOpt("audit_template"), group="command")
|
||||
cfg.CONF.register_opt(cfg.BoolOpt("exclude_orphans"), group="command")
|
||||
cfg.CONF.register_opt(cfg.BoolOpt("dry_run"), group="command")
|
||||
cfg.CONF.set_default("age_in_days", None, group="command")
|
||||
cfg.CONF.set_default("max_number", None, group="command")
|
||||
cfg.CONF.set_default("audit_template", None, group="command")
|
||||
cfg.CONF.set_default("exclude_orphans", True, group="command")
|
||||
cfg.CONF.set_default("dry_run", False, group="command")
|
||||
|
||||
dbmanage.DBCommand.purge()
|
||||
|
||||
m_purge_cls.assert_called_once_with(
|
||||
None, None, 'Some UUID', True, False)
|
||||
m_purge.execute.assert_called_once_with()
|
||||
|
||||
@mock.patch.object(sys, "exit")
|
||||
@mock.patch.object(purge, "PurgeCommand")
|
||||
def test_run_db_purge_negative_max_number(self, m_purge_cls, m_exit):
|
||||
m_purge = mock.Mock()
|
||||
m_purge_cls.return_value = m_purge
|
||||
m_purge_cls.get_audit_template_uuid.return_value = 'Some UUID'
|
||||
cfg.CONF.register_opt(cfg.IntOpt("age_in_days"), group="command")
|
||||
cfg.CONF.register_opt(cfg.IntOpt("max_number"), group="command")
|
||||
cfg.CONF.register_opt(cfg.StrOpt("audit_template"), group="command")
|
||||
cfg.CONF.register_opt(cfg.BoolOpt("exclude_orphans"), group="command")
|
||||
cfg.CONF.register_opt(cfg.BoolOpt("dry_run"), group="command")
|
||||
cfg.CONF.set_default("age_in_days", None, group="command")
|
||||
cfg.CONF.set_default("max_number", -1, group="command")
|
||||
cfg.CONF.set_default("audit_template", None, group="command")
|
||||
cfg.CONF.set_default("exclude_orphans", True, group="command")
|
||||
cfg.CONF.set_default("dry_run", False, group="command")
|
||||
|
||||
dbmanage.DBCommand.purge()
|
||||
|
||||
self.assertEqual(0, m_purge_cls.call_count)
|
||||
self.assertEqual(0, m_purge.execute.call_count)
|
||||
self.assertEqual(0, m_purge.do_delete.call_count)
|
||||
self.assertEqual(1, m_exit.call_count)
|
||||
|
||||
@mock.patch.object(sys, "exit")
|
||||
@mock.patch.object(purge, "PurgeCommand")
|
||||
def test_run_db_purge_dry_run(self, m_purge_cls, m_exit):
|
||||
m_purge = mock.Mock()
|
||||
m_purge_cls.return_value = m_purge
|
||||
m_purge_cls.get_audit_template_uuid.return_value = 'Some UUID'
|
||||
cfg.CONF.register_opt(cfg.IntOpt("age_in_days"), group="command")
|
||||
cfg.CONF.register_opt(cfg.IntOpt("max_number"), group="command")
|
||||
cfg.CONF.register_opt(cfg.StrOpt("audit_template"), group="command")
|
||||
cfg.CONF.register_opt(cfg.BoolOpt("exclude_orphans"), group="command")
|
||||
cfg.CONF.register_opt(cfg.BoolOpt("dry_run"), group="command")
|
||||
cfg.CONF.set_default("age_in_days", None, group="command")
|
||||
cfg.CONF.set_default("max_number", None, group="command")
|
||||
cfg.CONF.set_default("audit_template", None, group="command")
|
||||
cfg.CONF.set_default("exclude_orphans", True, group="command")
|
||||
cfg.CONF.set_default("dry_run", True, group="command")
|
||||
|
||||
dbmanage.DBCommand.purge()
|
||||
|
||||
m_purge_cls.assert_called_once_with(
|
||||
None, None, 'Some UUID', True, True)
|
||||
self.assertEqual(1, m_purge.execute.call_count)
|
||||
self.assertEqual(0, m_purge.do_delete.call_count)
|
||||
self.assertEqual(0, m_exit.call_count)
|
||||
|
||||
372
watcher/tests/db/test_purge.py
Normal file
372
watcher/tests/db/test_purge.py
Normal file
@@ -0,0 +1,372 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
# Copyright (c) 2016 b<>com
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
# implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import uuid
|
||||
|
||||
import freezegun
|
||||
import mock
|
||||
|
||||
from watcher.common import context as watcher_context
|
||||
from watcher.db import purge
|
||||
from watcher.db.sqlalchemy import api as dbapi
|
||||
from watcher.tests.db import base
|
||||
from watcher.tests.objects import utils as obj_utils
|
||||
|
||||
|
||||
class TestPurgeCommand(base.DbTestCase):
|
||||
|
||||
def setUp(self):
|
||||
super(TestPurgeCommand, self).setUp()
|
||||
self.cmd = purge.PurgeCommand()
|
||||
token_info = {
|
||||
'token': {
|
||||
'project': {
|
||||
'id': 'fake_project'
|
||||
},
|
||||
'user': {
|
||||
'id': 'fake_user'
|
||||
}
|
||||
}
|
||||
}
|
||||
self.context = watcher_context.RequestContext(
|
||||
auth_token_info=token_info,
|
||||
project_id='fake_project',
|
||||
user_id='fake_user',
|
||||
show_deleted=True,
|
||||
)
|
||||
|
||||
self.fake_today = '2016-02-24T09:52:05.219414+00:00'
|
||||
self.expired_date = '2016-01-24T09:52:05.219414+00:00'
|
||||
|
||||
self.m_input = mock.Mock()
|
||||
p = mock.patch("watcher.db.purge.input", self.m_input)
|
||||
self.m_input.return_value = 'y'
|
||||
p.start()
|
||||
self.addCleanup(p.stop)
|
||||
|
||||
self._id_generator = None
|
||||
self._data_setup()
|
||||
|
||||
def _generate_id(self):
|
||||
if self._id_generator is None:
|
||||
self._id_generator = self._get_id_generator()
|
||||
return next(self._id_generator)
|
||||
|
||||
def _get_id_generator(self):
|
||||
seed = 1
|
||||
while True:
|
||||
yield seed
|
||||
seed += 1
|
||||
|
||||
def _data_setup(self):
|
||||
# All the 1's are soft_deleted and are expired
|
||||
# All the 2's are soft_deleted but are not expired
|
||||
# All the 3's are *not* soft_deleted
|
||||
|
||||
# Number of days we want to keep in DB (no purge for them)
|
||||
self.cmd.age_in_days = 10
|
||||
self.cmd.max_number = None
|
||||
self.cmd.orphans = True
|
||||
gen_name = lambda: "Audit Template %s" % uuid.uuid4()
|
||||
self.audit_template1_name = gen_name()
|
||||
self.audit_template2_name = gen_name()
|
||||
self.audit_template3_name = gen_name()
|
||||
|
||||
with freezegun.freeze_time(self.expired_date):
|
||||
self.audit_template1 = obj_utils.create_test_audit_template(
|
||||
self.context, name=self.audit_template1_name,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.audit_template2 = obj_utils.create_test_audit_template(
|
||||
self.context, name=self.audit_template2_name,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.audit_template3 = obj_utils.create_test_audit_template(
|
||||
self.context, name=self.audit_template3_name,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.audit_template1.soft_delete()
|
||||
|
||||
with freezegun.freeze_time(self.expired_date):
|
||||
self.audit1 = obj_utils.create_test_audit(
|
||||
self.context, audit_template_id=self.audit_template1.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.audit2 = obj_utils.create_test_audit(
|
||||
self.context, audit_template_id=self.audit_template2.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.audit3 = obj_utils.create_test_audit(
|
||||
self.context, audit_template_id=self.audit_template3.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.audit1.soft_delete()
|
||||
|
||||
with freezegun.freeze_time(self.expired_date):
|
||||
self.action_plan1 = obj_utils.create_test_action_plan(
|
||||
self.context, audit_id=self.audit1.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.action_plan2 = obj_utils.create_test_action_plan(
|
||||
self.context, audit_id=self.audit2.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.action_plan3 = obj_utils.create_test_action_plan(
|
||||
self.context, audit_id=self.audit3.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
|
||||
self.action1 = obj_utils.create_test_action(
|
||||
self.context, action_plan_id=self.action_plan1.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.action2 = obj_utils.create_test_action(
|
||||
self.context, action_plan_id=self.action_plan2.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.action3 = obj_utils.create_test_action(
|
||||
self.context, action_plan_id=self.action_plan3.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
self.action_plan1.soft_delete()
|
||||
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action_plan")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit_template")
|
||||
def test_execute_max_number_exceeded(self, m_destroy_audit_template,
|
||||
m_destroy_audit,
|
||||
m_destroy_action_plan,
|
||||
m_destroy_action):
|
||||
self.cmd.age_in_days = None
|
||||
self.cmd.max_number = 5
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
self.audit_template2.soft_delete()
|
||||
self.audit2.soft_delete()
|
||||
self.action_plan2.soft_delete()
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
self.cmd.execute()
|
||||
|
||||
# The 1's and the 2's are purgeable (due to age of day set to 0),
|
||||
# but max_number = 5, and because of no Db integrity violation, we
|
||||
# should be able to purge only 4 objects.
|
||||
self.assertEqual(m_destroy_audit_template.call_count, 1)
|
||||
self.assertEqual(m_destroy_audit.call_count, 1)
|
||||
self.assertEqual(m_destroy_action_plan.call_count, 1)
|
||||
self.assertEqual(m_destroy_action.call_count, 1)
|
||||
|
||||
def test_find_deleted_entries(self):
|
||||
self.cmd.age_in_days = None
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
objects_map = self.cmd.find_objects_to_delete()
|
||||
|
||||
self.assertEqual(len(objects_map.audit_templates), 1)
|
||||
self.assertEqual(len(objects_map.audits), 1)
|
||||
self.assertEqual(len(objects_map.action_plans), 1)
|
||||
self.assertEqual(len(objects_map.actions), 1)
|
||||
|
||||
def test_find_deleted_and_expired_entries(self):
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
self.audit_template2.soft_delete()
|
||||
self.audit2.soft_delete()
|
||||
self.action_plan2.soft_delete()
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
objects_map = self.cmd.find_objects_to_delete()
|
||||
|
||||
# The 1's are purgeable (due to age of day set to 10)
|
||||
self.assertEqual(len(objects_map.audit_templates), 1)
|
||||
self.assertEqual(len(objects_map.audits), 1)
|
||||
self.assertEqual(len(objects_map.action_plans), 1)
|
||||
self.assertEqual(len(objects_map.actions), 1)
|
||||
|
||||
def test_find_deleted_and_nonexpired_related_entries(self):
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
# orphan audit
|
||||
audit4 = obj_utils.create_test_audit(
|
||||
self.context, audit_template_id=404, # Does not exist
|
||||
id=self._generate_id(), uuid=None)
|
||||
action_plan4 = obj_utils.create_test_action_plan(
|
||||
self.context, audit_id=audit4.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
action4 = obj_utils.create_test_action(
|
||||
self.context, action_plan_id=action_plan4.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
|
||||
audit5 = obj_utils.create_test_audit(
|
||||
self.context, audit_template_id=self.audit_template1.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
action_plan5 = obj_utils.create_test_action_plan(
|
||||
self.context, audit_id=audit5.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
action5 = obj_utils.create_test_action(
|
||||
self.context, action_plan_id=action_plan5.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
|
||||
self.audit_template2.soft_delete()
|
||||
self.audit2.soft_delete()
|
||||
self.action_plan2.soft_delete()
|
||||
|
||||
# All the 4's should be purged as well because they are orphans
|
||||
# even though they were not deleted
|
||||
|
||||
# All the 5's should be purged as well even though they are not
|
||||
# expired because their related audit template is itself expired
|
||||
audit5.soft_delete()
|
||||
action_plan5.soft_delete()
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
objects_map = self.cmd.find_objects_to_delete()
|
||||
|
||||
self.assertEqual(len(objects_map.audit_templates), 1)
|
||||
self.assertEqual(len(objects_map.audits), 3)
|
||||
self.assertEqual(len(objects_map.action_plans), 3)
|
||||
self.assertEqual(len(objects_map.actions), 3)
|
||||
self.assertEqual(
|
||||
set([self.action1.id, action4.id, action5.id]),
|
||||
set([entry.id for entry in objects_map.actions]))
|
||||
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action_plan")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit_template")
|
||||
def test_purge_command(self, m_destroy_audit_template,
|
||||
m_destroy_audit, m_destroy_action_plan,
|
||||
m_destroy_action):
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
self.cmd.execute()
|
||||
|
||||
m_destroy_audit_template.assert_called_once_with(
|
||||
self.audit_template1.uuid)
|
||||
m_destroy_audit.assert_called_once_with(
|
||||
self.audit1.uuid)
|
||||
m_destroy_action_plan.assert_called_once_with(
|
||||
self.action_plan1.uuid)
|
||||
m_destroy_action.assert_called_once_with(
|
||||
self.action1.uuid)
|
||||
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action_plan")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit_template")
|
||||
def test_purge_command_with_nonexpired_related_entries(
|
||||
self, m_destroy_audit_template, m_destroy_audit,
|
||||
m_destroy_action_plan, m_destroy_action):
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
# orphan audit
|
||||
audit4 = obj_utils.create_test_audit(
|
||||
self.context, audit_template_id=404, # Does not exist
|
||||
id=self._generate_id(), uuid=None)
|
||||
action_plan4 = obj_utils.create_test_action_plan(
|
||||
self.context, audit_id=audit4.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
action4 = obj_utils.create_test_action(
|
||||
self.context, action_plan_id=action_plan4.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
|
||||
audit5 = obj_utils.create_test_audit(
|
||||
self.context, audit_template_id=self.audit_template1.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
action_plan5 = obj_utils.create_test_action_plan(
|
||||
self.context, audit_id=audit5.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
action5 = obj_utils.create_test_action(
|
||||
self.context, action_plan_id=action_plan5.id,
|
||||
id=self._generate_id(), uuid=None)
|
||||
|
||||
self.audit_template2.soft_delete()
|
||||
self.audit2.soft_delete()
|
||||
self.action_plan2.soft_delete()
|
||||
|
||||
# All the 4's should be purged as well because they are orphans
|
||||
# even though they were not deleted
|
||||
|
||||
# All the 5's should be purged as well even though they are not
|
||||
# expired because their related audit template is itself expired
|
||||
audit5.soft_delete()
|
||||
action_plan5.soft_delete()
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
self.cmd.execute()
|
||||
|
||||
self.assertEqual(m_destroy_audit_template.call_count, 1)
|
||||
self.assertEqual(m_destroy_audit.call_count, 3)
|
||||
self.assertEqual(m_destroy_action_plan.call_count, 3)
|
||||
self.assertEqual(m_destroy_action.call_count, 3)
|
||||
|
||||
m_destroy_audit_template.assert_any_call(self.audit_template1.uuid)
|
||||
m_destroy_audit.assert_any_call(self.audit1.uuid)
|
||||
m_destroy_audit.assert_any_call(audit4.uuid)
|
||||
m_destroy_action_plan.assert_any_call(self.action_plan1.uuid)
|
||||
m_destroy_action_plan.assert_any_call(action_plan4.uuid)
|
||||
m_destroy_action_plan.assert_any_call(action_plan5.uuid)
|
||||
m_destroy_action.assert_any_call(self.action1.uuid)
|
||||
m_destroy_action.assert_any_call(action4.uuid)
|
||||
m_destroy_action.assert_any_call(action5.uuid)
|
||||
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action_plan")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit_template")
|
||||
def test_purge_command_with_audit_template_ok(
|
||||
self, m_destroy_audit_template, m_destroy_audit,
|
||||
m_destroy_action_plan, m_destroy_action):
|
||||
self.cmd.orphans = False
|
||||
self.cmd.uuid = self.audit_template1.uuid
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
self.cmd.execute()
|
||||
|
||||
self.assertEqual(m_destroy_audit_template.call_count, 1)
|
||||
self.assertEqual(m_destroy_audit.call_count, 1)
|
||||
self.assertEqual(m_destroy_action_plan.call_count, 1)
|
||||
self.assertEqual(m_destroy_action.call_count, 1)
|
||||
|
||||
m_destroy_audit_template.assert_called_once_with(
|
||||
self.audit_template1.uuid)
|
||||
m_destroy_audit.assert_called_once_with(
|
||||
self.audit1.uuid)
|
||||
m_destroy_action_plan.assert_called_once_with(
|
||||
self.action_plan1.uuid)
|
||||
m_destroy_action.assert_called_once_with(
|
||||
self.action1.uuid)
|
||||
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action_plan")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit_template")
|
||||
def test_purge_command_with_audit_template_not_expired(
|
||||
self, m_destroy_audit_template, m_destroy_audit,
|
||||
m_destroy_action_plan, m_destroy_action):
|
||||
self.cmd.orphans = False
|
||||
self.cmd.uuid = self.audit_template2.uuid
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
self.cmd.execute()
|
||||
|
||||
self.assertEqual(m_destroy_audit_template.call_count, 0)
|
||||
self.assertEqual(m_destroy_audit.call_count, 0)
|
||||
self.assertEqual(m_destroy_action_plan.call_count, 0)
|
||||
self.assertEqual(m_destroy_action.call_count, 0)
|
||||
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_action_plan")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit")
|
||||
@mock.patch.object(dbapi.Connection, "destroy_audit_template")
|
||||
def test_purge_command_with_audit_template_not_soft_deleted(
|
||||
self, m_destroy_audit_template, m_destroy_audit,
|
||||
m_destroy_action_plan, m_destroy_action):
|
||||
self.cmd.orphans = False
|
||||
self.cmd.uuid = self.audit_template3.uuid
|
||||
|
||||
with freezegun.freeze_time(self.fake_today):
|
||||
self.cmd.execute()
|
||||
|
||||
self.assertEqual(m_destroy_audit_template.call_count, 0)
|
||||
self.assertEqual(m_destroy_audit.call_count, 0)
|
||||
self.assertEqual(m_destroy_action_plan.call_count, 0)
|
||||
self.assertEqual(m_destroy_action.call_count, 0)
|
||||
Reference in New Issue
Block a user