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
def workflow_job_template_unit(): return WorkflowJobTemplate(name='workflow')