def test_execute_run_api_grpc_python_handle():
with instance_for_test() as instance:
with get_foo_pipeline_handle() as pipeline_handle:
pipeline_run = instance.create_run(
pipeline_name="foo",
run_id=None,
run_config={},
mode="default",
solids_to_execute=None,
step_keys_to_execute=None,
status=None,
tags=None,
root_run_id=None,
parent_run_id=None,
pipeline_snapshot=None,
execution_plan_snapshot=None,
parent_pipeline_snapshot=None,
)
loadable_target_origin = (pipeline_handle.get_external_origin(
).external_repository_origin.repository_location_origin.
loadable_target_origin)
server_process = GrpcServerProcess(loadable_target_origin,
max_workers=2)
with server_process.create_ephemeral_client() as api_client:
events = [
event for event in sync_execute_run_grpc(
api_client=api_client,
instance_ref=instance.get_ref(),
pipeline_origin=pipeline_handle.get_external_origin(),
pipeline_run=pipeline_run,
)
]
assert len(events) == 17
assert [event.event_type_value for event in events] == [
"ENGINE_EVENT",
"ENGINE_EVENT",
"PIPELINE_START",
"ENGINE_EVENT",
"STEP_START",
"STEP_OUTPUT",
"OBJECT_STORE_OPERATION",
"STEP_SUCCESS",
"STEP_START",
"OBJECT_STORE_OPERATION",
"STEP_INPUT",
"STEP_OUTPUT",
"OBJECT_STORE_OPERATION",
"STEP_SUCCESS",
"ENGINE_EVENT",
"PIPELINE_SUCCESS",
"ENGINE_EVENT",
]
server_process.wait()
def test_execute_run_api_grpc_python_handle(pipeline_handle):
with seven.TemporaryDirectory() as temp_dir:
instance = DagsterInstance.local_temp(temp_dir)
pipeline_run = instance.create_run(
pipeline_name="foo",
run_id=None,
run_config={},
mode="default",
solids_to_execute=None,
step_keys_to_execute=None,
status=None,
tags=None,
root_run_id=None,
parent_run_id=None,
pipeline_snapshot=None,
execution_plan_snapshot=None,
parent_pipeline_snapshot=None,
)
loadable_target_origin = LoadableTargetOrigin.from_python_origin(
pipeline_handle.get_origin().repository_origin
)
with GrpcServerProcess(
loadable_target_origin, max_workers=2
).create_ephemeral_client() as api_client:
events = [
event
for event in sync_execute_run_grpc(
api_client=api_client,
instance_ref=instance.get_ref(),
pipeline_origin=pipeline_handle.get_origin(),
pipeline_run=pipeline_run,
)
]
assert len(events) == 14
assert [event.event_type_value for event in events] == [
"ENGINE_EVENT",
"ENGINE_EVENT",
"PIPELINE_START",
"ENGINE_EVENT",
"STEP_START",
"STEP_OUTPUT",
"STEP_SUCCESS",
"STEP_START",
"STEP_INPUT",
"STEP_OUTPUT",
"STEP_SUCCESS",
"ENGINE_EVENT",
"PIPELINE_SUCCESS",
"ENGINE_EVENT",
]
def execute_pipeline(
self, instance, external_pipeline, pipeline_run,
):
from dagster.api.execute_run import sync_execute_run_grpc
check.inst_param(instance, "instance", DagsterInstance)
check.inst_param(external_pipeline, "external_pipeline", ExternalPipeline)
check.inst_param(pipeline_run, "pipeline_run", PipelineRun)
event_list = sync_execute_run_grpc(
self._handle.client, instance.get_ref(), external_pipeline.get_origin(), pipeline_run
)
return ExternalPipelineExecutionResult(event_list=event_list)
def test_execute_run_api_grpc_server_handle():
with seven.TemporaryDirectory() as temp_dir:
instance = DagsterInstance.local_temp(temp_dir)
with get_foo_grpc_pipeline_handle() as pipeline_handle:
pipeline_run = instance.create_run(
pipeline_name="foo",
run_id=None,
run_config={},
mode="default",
solids_to_execute=None,
step_keys_to_execute=None,
status=None,
tags=None,
root_run_id=None,
parent_run_id=None,
pipeline_snapshot=None,
execution_plan_snapshot=None,
parent_pipeline_snapshot=None,
)
events = [
event for event in sync_execute_run_grpc(
api_client=pipeline_handle.repository_handle.
repository_location_handle.client,
instance_ref=instance.get_ref(),
pipeline_origin=pipeline_handle.get_origin(),
pipeline_run=pipeline_run,
)
]
assert len(events) == 17
assert [event.event_type_value for event in events] == [
"ENGINE_EVENT",
"ENGINE_EVENT",
"PIPELINE_START",
"ENGINE_EVENT",
"STEP_START",
"STEP_OUTPUT",
"OBJECT_STORE_OPERATION",
"STEP_SUCCESS",
"STEP_START",
"OBJECT_STORE_OPERATION",
"STEP_INPUT",
"STEP_OUTPUT",
"OBJECT_STORE_OPERATION",
"STEP_SUCCESS",
"ENGINE_EVENT",
"PIPELINE_SUCCESS",
"ENGINE_EVENT",
]
def test_execute_run_api_grpc_server_handle(pipeline_handle):
with seven.TemporaryDirectory() as temp_dir:
instance = DagsterInstance.local_temp(temp_dir)
pipeline_run = instance.create_run(
pipeline_name='foo',
run_id=None,
run_config={},
mode='default',
solids_to_execute=None,
step_keys_to_execute=None,
status=None,
tags=None,
root_run_id=None,
parent_run_id=None,
pipeline_snapshot=None,
execution_plan_snapshot=None,
parent_pipeline_snapshot=None,
)
events = [
event for event in sync_execute_run_grpc(
api_client=pipeline_handle.repository_handle.
repository_location_handle.client,
instance_ref=instance.get_ref(),
pipeline_origin=pipeline_handle.get_origin(),
pipeline_run=pipeline_run,
)
]
assert len(events) == 14
assert [event.event_type_value for event in events] == [
'ENGINE_EVENT',
'ENGINE_EVENT',
'PIPELINE_START',
'ENGINE_EVENT',
'STEP_START',
'STEP_OUTPUT',
'STEP_SUCCESS',
'STEP_START',
'STEP_INPUT',
'STEP_OUTPUT',
'STEP_SUCCESS',
'ENGINE_EVENT',
'PIPELINE_SUCCESS',
'ENGINE_EVENT',
]
