def test_cleanup(self, mock_cleanup):
audit_templates_mocks = [mock.Mock() for i in range(2)]
self.context.update({
"admin": {
"credential": mock.MagicMock()
},
"audit_templates": audit_templates_mocks
})
audit_templates_ctx = audit_templates.AuditTemplateGenerator(
self.context)
audit_templates_ctx.cleanup()
mock_cleanup.assert_called_once_with(
names=["watcher.action_plan", "watcher.audit_template"],
admin=self.context["admin"])
def test_setup(self, mock_osclients, mock_watcher_goal_transform,
mock_watcher_strategy_transform,
mock_watcher_scenario__create_audit_template):
users = [{"id": 1, "tenant_id": 1, "credential": mock.MagicMock()}]
self.context.update({
"config": {
"audit_templates": {
"audit_templates_per_admin":
1,
"fill_strategy":
"random",
"params": [{
"goal": {
"name": "workload_balancing"
},
"strategy": {
"name": "workload_stabilization"
},
"extra": {}
}, {
"goal": {
"name": "workload_balancing"
},
"strategy": {
"name": "workload_stabilization"
},
"extra": {}
}]
},
},
"admin": {
"credential": mock.MagicMock()
},
"users": users
})
audit_template = audit_templates.AuditTemplateGenerator(self.context)
audit_template.setup()
goal_id = mock_watcher_goal_transform.return_value
strategy_id = mock_watcher_strategy_transform.return_value
mock_calls = [mock.call(goal_id, strategy_id, {})]
mock_watcher_scenario__create_audit_template.assert_has_calls(
mock_calls)