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