def test_wfjt_unique_together_with_org(self, organization):
wfjt1 = WorkflowJobTemplate(name='foo', organization=organization)
wfjt1.save()
wfjt2 = WorkflowJobTemplate(name='foo', organization=organization)
with pytest.raises(ValidationError):
wfjt2.validate_unique()
wfjt2 = WorkflowJobTemplate(name='foo', organization=None)
wfjt2.validate_unique()
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 workflow_job_template(organization):
wjt = WorkflowJobTemplate(name='test-workflow_job_template',
organization=organization)
wjt.save()
return wjt