mirror of
https://github.com/ansible/awx.git
synced 2026-02-24 06:26:00 -03:30
instantiate db with sample workflows
This commit is contained in:
41
awx/main/tests/manual/workflows/linear.py
Normal file
41
awx/main/tests/manual/workflows/linear.py
Normal file
@@ -0,0 +1,41 @@
|
||||
# AWX
|
||||
from awx.main.models import (
|
||||
WorkflowNode,
|
||||
WorkflowJobTemplate,
|
||||
WorkflowJob,
|
||||
)
|
||||
from awx.main.models.jobs import JobTemplate
|
||||
|
||||
def do_init_workflow(job_template_success, job_template_fail, job_template_never):
|
||||
wfjt, created = WorkflowJobTemplate.objects.get_or_create(name="linear workflow")
|
||||
wfjt.delete()
|
||||
wfjt, created = WorkflowJobTemplate.objects.get_or_create(name="linear workflow")
|
||||
print(wfjt.id)
|
||||
WorkflowNode.objects.all().delete()
|
||||
if created:
|
||||
nodes_success = []
|
||||
nodes_fail = []
|
||||
nodes_never = []
|
||||
for i in range(0, 2):
|
||||
nodes_success.append(WorkflowNode.objects.create(workflow_job_template=wfjt, unified_job_template=job_template_success))
|
||||
nodes_fail.append(WorkflowNode.objects.create(workflow_job_template=wfjt, unified_job_template=job_template_fail))
|
||||
nodes_never.append(WorkflowNode.objects.create(workflow_job_template=wfjt, unified_job_template=job_template_never))
|
||||
nodes_never.append(WorkflowNode.objects.create(workflow_job_template=wfjt, unified_job_template=job_template_never))
|
||||
nodes_fail[1].delete()
|
||||
|
||||
nodes_success[0].success_nodes.add(nodes_fail[0])
|
||||
nodes_success[0].failure_nodes.add(nodes_never[0])
|
||||
|
||||
nodes_fail[0].failure_nodes.add(nodes_success[1])
|
||||
nodes_fail[0].success_nodes.add(nodes_never[1])
|
||||
|
||||
nodes_success[1].failure_nodes.add(nodes_never[2])
|
||||
|
||||
def do_init():
|
||||
jt_success = JobTemplate.objects.get(id=5)
|
||||
jt_fail= JobTemplate.objects.get(id=9)
|
||||
jt_never= JobTemplate.objects.get(id=11)
|
||||
do_init_workflow(jt_success, jt_fail, jt_never)
|
||||
|
||||
if __name__ == "__main__":
|
||||
do_init()
|
||||
Reference in New Issue
Block a user