Ejemplos de sync_get_external_schedule_execution_data_ephemeral_grpc en Python

Ejemplo n.º 1

def test_external_partitions_api_grpc():
    repository_handle = get_bar_repo_handle()
    with seven.TemporaryDirectory() as temp_dir:
        instance = DagsterInstance.local_temp(temp_dir)
        execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
            instance, repository_handle, 'foo_schedule'
        )
        assert isinstance(execution_data, ExternalScheduleExecutionData)
        assert execution_data.run_config is not None

Ejemplo n.º 2

Archivo: test_api_snapshot_schedule_execution_data.py Proyecto: zuik/dagster

def test_external_schedule_execution_data_api_never_execute_grpc():
    with instance_for_test() as instance:
        with get_bar_repo_handle() as repository_handle:
            execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
                instance,
                repository_handle,
                "foo_schedule_never_execute",
                None,
            )
            assert isinstance(execution_data, ExternalScheduleExecutionData)
            assert len(execution_data.run_requests) == 0

Ejemplo n.º 3

def test_external_schedule_execution_data_api_preview_grpc(schedule_name):
    repository_handle = get_bar_repo_handle()
    with seven.TemporaryDirectory() as temp_dir:
        instance = DagsterInstance.local_temp(temp_dir)
        execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
            instance, repository_handle, schedule_name, ScheduleExecutionDataMode.PREVIEW, None,
        )
        assert isinstance(execution_data, ExternalScheduleExecutionData)

        assert execution_data.run_config == {"fizz": "buzz"}
        assert execution_data.tags == {"dagster/schedule_name": schedule_name}
        assert execution_data.should_execute is None

Ejemplo n.º 4

Archivo: test_api_snapshot_schedule_execution_data.py Proyecto: xaniasd/dagster

def test_external_schedule_execution_data_api_never_execute_grpc():
    with instance_for_test() as instance:
        with get_bar_repo_handle() as repository_handle:
            execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
                instance,
                repository_handle,
                "foo_schedule_never_execute",
                None,
            )
            assert isinstance(execution_data, ExternalScheduleExecutionData)
            assert execution_data.run_config == {}
            assert execution_data.tags == {}
            assert execution_data.should_execute == False

Ejemplo n.º 5

def test_external_schedule_execution_data_api_grpc():
    repository_handle = get_bar_repo_handle()
    with seven.TemporaryDirectory() as temp_dir:
        instance = DagsterInstance.local_temp(temp_dir)
        execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
            instance,
            repository_handle,
            "foo_schedule",
            ScheduleExecutionDataMode.LAUNCH_SCHEDULED_EXECUTION,
        )
        assert isinstance(execution_data, ExternalScheduleExecutionData)
        assert execution_data.run_config == {"fizz": "buzz"}
        assert execution_data.tags == {"dagster/schedule_name": "foo_schedule"}
        assert execution_data.should_execute == True

Ejemplo n.º 6

Archivo: test_api_snapshot_schedule_execution_data.py Proyecto: zuik/dagster

def test_external_schedule_execution_data_api_grpc():
    with instance_for_test() as instance:
        with get_bar_repo_handle() as repository_handle:
            execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
                instance,
                repository_handle,
                "foo_schedule",
                None,
            )
            assert isinstance(execution_data, ExternalScheduleExecutionData)
            assert len(execution_data.run_requests) == 1
            to_launch = execution_data.run_requests[0]
            assert to_launch.run_config == {"fizz": "buzz"}
            assert to_launch.tags == {"dagster/schedule_name": "foo_schedule"}

Ejemplo n.º 7

Archivo: test_api_snapshot_schedule_execution_data.py Proyecto: jmsanders/dagster

def test_external_schedule_execution_data_api_grpc():
    with instance_for_test() as instance:
        with get_bar_repo_handle() as repository_handle:
            execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
                instance,
                repository_handle,
                "foo_schedule",
                ScheduleExecutionDataMode.LAUNCH_SCHEDULED_EXECUTION,
                None,
            )
            assert isinstance(execution_data, ExternalScheduleExecutionData)
            assert execution_data.run_config == {"fizz": "buzz"}
            assert execution_data.tags == {
                "dagster/schedule_name": "foo_schedule"
            }
            assert execution_data.should_execute == True

Ejemplo n.º 8

Archivo: test_api_snapshot_schedule_execution_data.py Proyecto: trevenrawr/dagster

def test_include_execution_time_grpc():
    with instance_for_test() as instance:
        with get_bar_repo_handle(instance) as repository_handle:
            execution_time = get_current_datetime_in_utc()
            execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
                instance,
                repository_handle,
                "foo_schedule_echo_time",
                execution_time,
            )

            assert isinstance(execution_data, ScheduleExecutionData)
            assert len(execution_data.run_requests) == 1
            to_launch = execution_data.run_requests[0]
            assert to_launch.run_config == {"passed_in_time": execution_time.isoformat()}
            assert to_launch.tags == {"dagster/schedule_name": "foo_schedule_echo_time"}

Ejemplo n.º 9

Archivo: test_api_snapshot_schedule_execution_data.py Proyecto: jmsanders/dagster

def test_external_schedule_execution_data_api_preview_grpc(schedule_name):
    with instance_for_test() as instance:
        with get_bar_repo_handle() as repository_handle:
            execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
                instance,
                repository_handle,
                schedule_name,
                ScheduleExecutionDataMode.PREVIEW,
                None,
            )
            assert isinstance(execution_data, ExternalScheduleExecutionData)

            assert execution_data.run_config == {"fizz": "buzz"}
            assert execution_data.tags == {
                "dagster/schedule_name": schedule_name
            }
            assert execution_data.should_execute is None

Ejemplo n.º 10

def test_include_execution_time_grpc():
    repository_handle = get_bar_repo_handle()

    execution_time = get_current_datetime_in_utc()
    with instance_for_test() as instance:
        execution_data = sync_get_external_schedule_execution_data_ephemeral_grpc(
            instance,
            repository_handle,
            "foo_schedule_echo_time",
            ScheduleExecutionDataMode.LAUNCH_SCHEDULED_EXECUTION,
            execution_time,
        )

        assert isinstance(execution_data, ExternalScheduleExecutionData)
        assert execution_data.run_config == {"passed_in_time": execution_time.isoformat()}
        assert execution_data.tags == {"dagster/schedule_name": "foo_schedule_echo_time"}
        assert execution_data.should_execute == True