def test_hydrate_registration_parameters__workflow_nothing_set():
workflow = _workflow_pb2.WorkflowSpec(
template=_core_workflow_pb2.WorkflowTemplate(
id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.WORKFLOW,
name="name",
),
nodes=[
_core_workflow_pb2.Node(
id="foo",
task_node=_core_workflow_pb2.TaskNode(
reference_id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK, name="task1")),
)
],
))
identifier, entity = hydrate_registration_parameters(
_identifier_pb2.WORKFLOW, "project", "domain", "12345", workflow)
assert (identifier == _identifier_pb2.Identifier(
resource_type=_identifier_pb2.WORKFLOW,
project="project",
domain="domain",
name="name",
version="12345",
) == entity.template.id)
assert len(workflow.template.nodes) == 1
assert workflow.template.nodes[
0].task_node.reference_id == _identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK,
project="project",
domain="domain",
name="task1",
version="12345",
)
def to_flyte_idl(self):
"""
:rtype: flyteidl.admin.workflow_pb2.WorkflowSpec
"""
return _admin_workflow.WorkflowSpec(
template=self._template.to_flyte_idl(),
sub_workflows=[s.to_flyte_idl() for s in self._sub_workflows],
)
def test_hydrate_registration_parameters__workflow_already_set():
workflow = _workflow_pb2.WorkflowSpec(
template=_core_workflow_pb2.WorkflowTemplate(
id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.WORKFLOW,
project="project2",
domain="domain2",
name="name",
version="abc",
), ))
identifier, entity = hydrate_registration_parameters(
_identifier_pb2.WORKFLOW, "project", "domain", "12345", workflow)
assert (identifier == _identifier_pb2.Identifier(
resource_type=_identifier_pb2.WORKFLOW,
project="project2",
domain="domain2",
name="name",
version="abc",
) == entity.template.id)
def test_hydrate_registration_parameters__subworkflows():
workflow_template = _core_workflow_pb2.WorkflowTemplate()
workflow_template.id.CopyFrom(
_identifier_pb2.Identifier(resource_type=_identifier_pb2.WORKFLOW,
name="workflow"))
sub_workflow_template = _core_workflow_pb2.WorkflowTemplate()
sub_workflow_template.id.CopyFrom(
_identifier_pb2.Identifier(resource_type=_identifier_pb2.WORKFLOW,
name="subworkflow"))
sub_workflow_template.nodes.append(
_core_workflow_pb2.Node(
id="task_node",
task_node=_core_workflow_pb2.TaskNode(
reference_id=_identifier_pb2.Identifier(
resource_type=_identifier_pb2.TASK)),
))
workflow_spec = _workflow_pb2.WorkflowSpec(template=workflow_template)
workflow_spec.sub_workflows.append(sub_workflow_template)
identifier, entity = hydrate_registration_parameters(
_identifier_pb2.WORKFLOW, "project", "domain", "12345", workflow_spec)
assert (identifier == _identifier_pb2.Identifier(
resource_type=_identifier_pb2.WORKFLOW,
project="project",
domain="domain",
name="workflow",
version="12345",
) == entity.template.id)
assert entity.sub_workflows[0].id == _identifier_pb2.Identifier(
resource_type=_identifier_pb2.WORKFLOW,
project="project",
domain="domain",
name="subworkflow",
version="12345",
)
def to_flyte_idl(self):
"""
:rtype: flyteidl.admin.workflow_pb2.WorkflowSpec
"""
return _admin_workflow.WorkflowSpec(
template=self._template.to_flyte_idl())