def test_hydrate_workflow_template__branch_node():
workflow_template = _core_workflow_pb2.WorkflowTemplate()
branch_node = _core_workflow_pb2.Node(
id="branch_node",
branch_node=_core_workflow_pb2.BranchNode(
if_else=_core_workflow_pb2.IfElseBlock(
case=_core_workflow_pb2.IfBlock(
then_node=_core_workflow_pb2.Node(
task_node=_core_workflow_pb2.TaskNode(
reference_id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK,
name="if_case"), ), )),
else_node=_core_workflow_pb2.Node(
task_node=_core_workflow_pb2.TaskNode(
reference_id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK,
name="else_node"), ), ),
), ),
)
def to_flyte_idl(self):
"""
:rtype: flyteidl.core.workflow_pb2.IfBlock
"""
return _core_workflow.IfBlock(condition=self.condition.to_flyte_idl(), then_node=self.then_node.to_flyte_idl())
then_node=_core_workflow_pb2.Node(
task_node=_core_workflow_pb2.TaskNode(
reference_id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK,
name="if_case"), ), )),
else_node=_core_workflow_pb2.Node(
task_node=_core_workflow_pb2.TaskNode(
reference_id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK,
name="else_node"), ), ),
), ),
)
branch_node.branch_node.if_else.other.extend([
_core_workflow_pb2.IfBlock(then_node=_core_workflow_pb2.Node(
task_node=_core_workflow_pb2.TaskNode(
reference_id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK,
name="other_1"), ), ), ),
_core_workflow_pb2.IfBlock(then_node=_core_workflow_pb2.Node(
task_node=_core_workflow_pb2.TaskNode(
reference_id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK,
name="other_2"), ), ), ),
])
workflow_template.nodes.append(branch_node)
hydrated_workflow_template = _hydrate_workflow_template_nodes(
"project", "domain", "12345", workflow_template)
if_case_id = hydrated_workflow_template.nodes[
0].branch_node.if_else.case.then_node.task_node.reference_id
assert if_case_id.project == "project"
assert if_case_id.domain == "domain"
