Differential D5228 Diff 26378 python_modules/dagster/dagster_tests/api_tests/test_api_snapshot_sensor.py
Changeset View
Changeset View
Standalone View
Standalone View
python_modules/dagster/dagster_tests/api_tests/test_api_snapshot_sensor.py
Show All 9 Lines | |||||
def test_external_sensor_grpc(): | def test_external_sensor_grpc(): | ||||
with get_bar_repo_handle() as repository_handle: | with get_bar_repo_handle() as repository_handle: | ||||
with instance_for_test() as instance: | with instance_for_test() as instance: | ||||
result = sync_get_external_sensor_execution_data_ephemeral_grpc( | result = sync_get_external_sensor_execution_data_ephemeral_grpc( | ||||
instance, repository_handle, "sensor_foo", None | instance, repository_handle, "sensor_foo", None | ||||
) | ) | ||||
assert isinstance(result, ExternalSensorExecutionData) | assert isinstance(result, ExternalSensorExecutionData) | ||||
assert len(result.run_params) == 2 | assert len(result.run_requests) == 2 | ||||
run_params = result.run_params[0] | run_request = result.run_requests[0] | ||||
assert run_params.run_config == {"foo": "FOO"} | assert run_request.run_config == {"foo": "FOO"} | ||||
assert run_params.tags == {"foo": "foo_tag"} | assert run_request.tags == {"foo": "foo_tag"} | ||||
def test_external_sensor_error(): | def test_external_sensor_error(): | ||||
with get_bar_repo_handle() as repository_handle: | with get_bar_repo_handle() as repository_handle: | ||||
with instance_for_test() as instance: | with instance_for_test() as instance: | ||||
result = sync_get_external_sensor_execution_data_ephemeral_grpc( | result = sync_get_external_sensor_execution_data_ephemeral_grpc( | ||||
instance, repository_handle, "sensor_error", None | instance, repository_handle, "sensor_error", None | ||||
) | ) | ||||
assert isinstance(result, ExternalSensorExecutionErrorData) | assert isinstance(result, ExternalSensorExecutionErrorData) | ||||
assert "womp womp" in result.error.to_string() | assert "womp womp" in result.error.to_string() |