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
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
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
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
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
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"}
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
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"}
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
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