def setup(self): watcher_scenario = watcher_utils.WatcherScenario({ "admin": self.context["admin"], "task": self.context["task"], "owner_id": self.context["owner_id"], "config": { "api_versions": self.context["config"].get("api_versions", []) } }) clients = osclients.Clients(self.context["admin"]["credential"]) self.context["audit_templates"] = [] for i in six.moves.range(self.config["audit_templates_per_admin"]): cfg_size = len(self.config["params"]) if self.config["fill_strategy"] == "round_robin": audit_params = self.config["params"][i % cfg_size] elif self.config["fill_strategy"] == "random": audit_params = random.choice(self.config["params"]) goal_id = types.WatcherGoal.transform( clients=clients, resource_config=audit_params["goal"]) strategy_id = types.WatcherStrategy.transform( clients=clients, resource_config=audit_params["strategy"]) audit_template = watcher_scenario._create_audit_template( goal_id, strategy_id) self.context["audit_templates"].append(audit_template.uuid)
def test_delete_audit(self): mock_audit = mock.Mock() watcher_scenario = utils.WatcherScenario(self.context) watcher_scenario._delete_audit(mock_audit) self.admin_clients("watcher").audit.delete.assert_called_once_with( mock_audit.uuid) self._test_atomic_action_timer(watcher_scenario.atomic_actions(), "watcher.delete_audit")
def test_delete_audit_template(self): watcher_scenario = utils.WatcherScenario(self.context) watcher_scenario._delete_audit_template("fake_audit_template") self.admin_clients( "watcher").audit_template.delete.assert_called_once_with( "fake_audit_template") self._test_atomic_action_timer(watcher_scenario.atomic_actions(), "watcher.delete_audit_template")
def test_list_audit_templates(self): audit_templates_list = [] watcher_scenario = utils.WatcherScenario(self.context) self.admin_clients( "watcher").audit_template.list.return_value = audit_templates_list return_audit_templates_list = watcher_scenario._list_audit_templates() self.assertEqual(audit_templates_list, return_audit_templates_list) self._test_atomic_action_timer(watcher_scenario.atomic_actions(), "watcher.list_audit_templates")
def test_create_audit_template(self): watcher_scenario = utils.WatcherScenario(self.context) watcher_scenario.generate_random_name = mock.MagicMock( return_value="mock_name") watcher_scenario._create_audit_template("fake_goal", "fake_strategy") self.admin_clients( "watcher").audit_template.create.assert_called_once_with( goal="fake_goal", strategy="fake_strategy", name="mock_name") self._test_atomic_action_timer(watcher_scenario.atomic_actions(), "watcher.create_audit_template")
def test_create_audit(self): mock_audit_template = mock.Mock() watcher_scenario = utils.WatcherScenario(self.context) audit = watcher_scenario._create_audit(mock_audit_template) self.mock_wait_for_status.mock.assert_called_once_with( audit, ready_statuses=["SUCCEEDED"], failure_statuses=["FAILED"], status_attr="state", update_resource=self.mock_get_from_manager.mock.return_value, check_interval=CONF.openstack.watcher_audit_launch_poll_interval, timeout=CONF.openstack.watcher_audit_launch_timeout, id_attr="uuid") self.mock_get_from_manager.mock.assert_called_once_with() self.admin_clients("watcher").audit.create.assert_called_once_with( audit_template_uuid=mock_audit_template, audit_type="ONESHOT") self._test_atomic_action_timer(watcher_scenario.atomic_actions(), "watcher.create_audit")