add unit tests

This commit is contained in:
Chris Meyers
2016-09-13 13:31:10 -04:00
parent 9c12b234b1
commit 32461574ae
15 changed files with 617 additions and 295 deletions

View File

@@ -1,6 +1,6 @@
# AWX
from awx.main.models import (
WorkflowNode,
WorkflowJobTemplateNode,
WorkflowJobTemplate,
)
from awx.main.models.jobs import JobTemplate
@@ -10,17 +10,17 @@ def do_init_workflow(job_template_success, job_template_fail, job_template_never
wfjt.delete()
wfjt, created = WorkflowJobTemplate.objects.get_or_create(name="parallel workflow")
print(wfjt.id)
WorkflowNode.objects.all().delete()
WorkflowJobTemplateNode.objects.all().delete()
if created:
node_success = WorkflowNode.objects.create(workflow_job_template=wfjt, unified_job_template=job_template_success)
node_success = WorkflowJobTemplateNode.objects.create(workflow_job_template=wfjt, unified_job_template=job_template_success)
nodes_never = []
for x in range(0, 3):
nodes_never.append(WorkflowNode.objects.create(workflow_job_template=wfjt, unified_job_template=job_template_never))
nodes_never.append(WorkflowJobTemplateNode.objects.create(workflow_job_template=wfjt, unified_job_template=job_template_never))
nodes_parallel = []
for jt in jts_parallel:
nodes_parallel.append(WorkflowNode.objects.create(workflow_job_template=wfjt, unified_job_template=jt))
nodes_parallel.append(WorkflowJobTemplateNode.objects.create(workflow_job_template=wfjt, unified_job_template=jt))
node_success.success_nodes.add(nodes_parallel[0])
node_success.success_nodes.add(nodes_parallel[1])