def test_image_pull_secrets_workflow():
resp = requests.get(IMAGE_PULL_SECRETS)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_retry_container_to_completion_workflow():
resp = requests.get(RETRY_CONTAINER_TO_COMPLETION_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_retry_on_error_workflow():
resp = requests.get(RETRY_ON_ERROR_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_script_python_workflow():
resp = requests.get(SCRIPTS_PYTHON_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_corn_flip_workflow():
resp = requests.get(CORN_FLIP_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_input_artifact_s3_workflow():
resp = requests.get(INPUTS_ARTIFACT_S3_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_loops_sequence_workflow():
resp = requests.get(LOOPS_SEQUENCE_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_artifact_path_placeholders_workflow():
resp = requests.get(ARTIFACT_PATH_PLACEHOLDERS_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_artifact_repo_ref_workflow():
resp = requests.get(ARTIFACT_REPO_REF_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_workflow_template_ref():
resp = requests.get(WORKFLOW_TEMPLATE_REF_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_artifact_disable_archive_workflow():
resp = requests.get(ARTIFACT_DISABLE_ARCHIVE_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def register_template_workflow():
resp = requests.get(TEMPLATES_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflowtemplate(NAMESPACE, manifest)
def test_submit_parallelism_template_limit_workflow():
resp = requests.get(PARALLELSIM_TEMPLATE_LIMIT_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_parallelism_agg_dag_workflow():
resp = requests.get(PARALLELISM_AGG_DAG_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_exit_handler_step_level_workflow():
resp = requests.get(EXIT_HANDLERS_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_dag_continue_on_fail_workflow():
resp = requests.get(DAG_CONTINUE_ON_FAIL_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_global_parameters_workflow():
resp = requests.get(GLOBAL_PARAMETERS_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_dag_diamond_steps_workflow():
resp = requests.get(DAG_DIAMOND_STEPS_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_loops_param_result_workflow():
resp = requests.get(LOOPS_PARAM_RESULT_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_dag_disable_failfast_workflow():
resp = requests.get(DAG_DISABLE_FAILFAST_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_parameter_aggregation_workflow():
resp = requests.get(PARAMETER_AGGREGATION_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_dag_enhanced_depends_workflow():
resp = requests.get(DAG_ENHANCED_DEPENDS_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_hello_world_workflow():
resp = requests.get(HELLO_WORLD_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_dag_multi_root_workflow():
resp = requests.get(DAG_MULTI_ROOT_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_resubmit_workflow():
resp = requests.get(RESUBMIT_WORKFLOW_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_dag_targets_workflow():
resp = requests.get(DAG_TARGETS_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_retry_with_steps_workflow():
resp = requests.get(RETRY_WITH_STEPS_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit_exit_handler_dag_level_workflow():
resp = requests.get(EXIT_CODE_OUTPUT_VARIABLE)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)
def test_submit(self, api: V1alpha1Api, wf: Workflow) -> None:
"""Test `Workflow.submit` method."""
fake_workflow_name = "test"
flexmock(V1alpha1Api).should_receive("create_namespaced_workflow").and_return(
fake_workflow_name
)
# submit w/o parameters
workflow_name: str = wf.submit(client=V1alpha1Api(), namespace="test")
assert isinstance(workflow_name, str)
assert workflow_name == "test"
# submit w/ parameters
with pytest.raises(AttributeError) as exc:
# Expected ValueError due to undefined parameter
workflow_name: str = wf.submit(
client=api, namespace="test", parameters={"param": "test"}
)
wf.spec.arguments = V1alpha1Arguments(
parameters=[V1alpha1Parameter(name="param")]
)
workflow_result: str = wf.submit(
client=api, namespace="test", parameters={"param": "test"}
)
assert isinstance(workflow_result, V1alpha1Workflow)
assert isinstance(workflow_result.metadata.name, str)
assert len(workflow_result.spec.arguments.parameters) == 1
assert workflow_result.spec.arguments.parameters[0].name == 'param'
assert workflow_result.spec.arguments.parameters[0].value == 'test'
assert workflow_result.metadata.name == "test"
def test_init_container_workflow():
resp = requests.get(INIT_CONTAINER_YML)
resp.raise_for_status()
manifest: dict = yaml.safe_load(resp.text)
v1alpha1 = V1alpha1Api()
v1alpha1.create_namespaced_workflow(NAMESPACE, manifest)