def spawn_fan(pnode, wid, fan, depth):
if depth == 0:
return
for _ in range(fan):
node = WorkflowJobTemplateNode()
node.unified_job_template_id = 10
node.workflow_job_template_id = wid
node.save()
pnode.success_nodes.add(node)
spawn_fan(node, wid, fan, depth - 1)
pnode.save()
def spawn_workflow_fan(depth, fan):
w = WorkflowJobTemplate()
w.name = "w2"
w.save()
node = WorkflowJobTemplateNode()
node.unified_job_template_id = 10
node.workflow_job_template_id = w.id
node.save()
w.workflow_job_template_nodes.add(node)
w.save()
spawn_fan(node, w.id, fan, depth)
def spawn_workflow(num):
w = WorkflowJobTemplate()
w.name = "w1"
w.save()
prev_node = None
for _ in range(num):
node = WorkflowJobTemplateNode()
# 10 was a job template I had created previously
node.unified_job_template_id = 10
node.workflow_job_template_id = w.id
node.save()
if prev_node:
prev_node.success_nodes.add(node)
prev_node.save()
else:
w.workflow_job_template_nodes.add(node)
w.save()
prev_node = node
def job_template_nodes(self):
return [WorkflowJobTemplateNode(id=i) for i in range(0, 20)]
def job_template_nodes(self, job_templates):
return [
WorkflowJobTemplateNode(unified_job_template=job_templates[i])
for i in range(0, 10)
]
def test_node_getter_and_setters():
node = WorkflowJobTemplateNode()
node.job_type = 'check'
assert node.char_prompts['job_type'] == 'check'
assert node.job_type == 'check'
def wfjt_node_no_prompts(workflow_job_template_unit, jt_ask):
node = WorkflowJobTemplateNode(
workflow_job_template=workflow_job_template_unit,
unified_job_template=jt_ask)
return node
