def _create_tick(graphql_context):
with create_test_daemon_workspace(graphql_context.process_context.
workspace_load_target) as workspace:
list(
execute_sensor_iteration(graphql_context.instance,
get_default_daemon_logger("SensorDaemon"),
workspace))
def instance_for_context(external_repo_context, overrides=None):
with instance_for_test(overrides) as instance:
with create_test_daemon_workspace(
workspace_load_target=workspace_load_target()
) as workspace:
with external_repo_context() as external_repo:
yield (instance, workspace, external_repo)
def _test_launch_sensor_runs_in_subprocess(instance_ref, execution_datetime,
debug_crash_flags):
with DagsterInstance.from_ref(instance_ref) as instance:
try:
with pendulum.test(execution_datetime
), create_test_daemon_workspace() as workspace:
list(
execute_sensor_iteration(
instance,
get_default_daemon_logger("SensorDaemon"),
workspace,
debug_crash_flags=debug_crash_flags,
))
finally:
cleanup_test_instance(instance)
def _test_launch_scheduled_runs_in_subprocess(instance_ref, execution_datetime,
debug_crash_flags):
with DagsterInstance.from_ref(instance_ref) as instance:
try:
with create_test_daemon_workspace() as workspace:
with pendulum.test(execution_datetime):
list(
launch_scheduled_runs(
instance,
workspace,
logger(),
pendulum.now("UTC"),
debug_crash_flags=debug_crash_flags,
))
finally:
cleanup_test_instance(instance)
def _test_backfill_in_subprocess(instance_ref, debug_crash_flags):
execution_datetime = to_timezone(
create_pendulum_time(
year=2021,
month=2,
day=17,
),
"US/Central",
)
with DagsterInstance.from_ref(instance_ref) as instance:
try:
with pendulum.test(execution_datetime), create_test_daemon_workspace() as workspace:
list(
execute_backfill_iteration(
instance,
workspace,
get_default_daemon_logger("BackfillDaemon"),
debug_crash_flags=debug_crash_flags,
)
)
finally:
cleanup_test_instance(instance)
def workspace_fixture():
with create_test_daemon_workspace(workspace_load_target=EmptyWorkspaceTarget()) as workspace:
yield workspace
def workspace():
with create_test_daemon_workspace() as workspace:
yield workspace
def _create_tick(instance):
with create_test_daemon_workspace() as workspace:
list(
execute_sensor_iteration(instance,
get_default_daemon_logger("SensorDaemon"),
workspace))
def workspace_fixture(instance_session_scoped): # pylint: disable=unused-argument
with create_test_daemon_workspace(
workspace_load_target=workspace_load_target(), instance=instance_session_scoped
) as workspace:
yield workspace
def workspace_fixture(instance):
with create_test_daemon_workspace(
workspace_load_target=EmptyWorkspaceTarget(), instance=instance
) as workspace:
yield workspace
