add back mutation tests for sensor/schedules with default status (#6721)

* update default status tests * add partition set snapshot
dagster-io · Feb 18, 2022 · 19b106c · 19b106c
1 parent 9eba0d0
commit 19b106c
Show file tree

Hide file tree

Showing 5 changed files with 400 additions and 0 deletions.
diff --git a/python_modules/dagster-graphql/dagster_graphql_tests/graphql/setup.py b/python_modules/dagster-graphql/dagster_graphql_tests/graphql/setup.py
@@ -17,6 +17,8 @@
     AssetObservation,
     Bool,
     DagsterInstance,
+    DefaultScheduleStatus,
+    DefaultSensorStatus,
     DynamicOutput,
     DynamicOutputDefinition,
     Enum,
@@ -1078,6 +1080,15 @@ def define_schedules():
     def partition_based_decorator(_date):
         return {}
 
+    @daily_schedule(
+        pipeline_name="no_config_pipeline",
+        start_date=today_at_midnight().subtract(days=1),
+        execution_time=(datetime.datetime.now() + datetime.timedelta(hours=2)).time(),
+        default_status=DefaultScheduleStatus.RUNNING,
+    )
+    def running_in_code_schedule(_date):
+        return {}
+
     @daily_schedule(
         pipeline_name="multi_mode_with_loggers",
         start_date=today_at_midnight().subtract(days=1),
@@ -1194,6 +1205,7 @@ def timezone_schedule(_date):
         tags_error_schedule,
         timezone_schedule,
         invalid_config_schedule,
+        running_in_code_schedule,
     ]
 
 
@@ -1260,12 +1272,24 @@ def custom_interval_sensor(_):
             tags={"test": "1234"},
         )
 
+    @sensor(
+        pipeline_name="no_config_pipeline",
+        mode="default",
+        default_status=DefaultSensorStatus.RUNNING,
+    )
+    def running_in_code_sensor(_):
+        return RunRequest(
+            run_key=None,
+            tags={"test": "1234"},
+        )
+
     return [
         always_no_config_sensor,
         once_no_config_sensor,
         never_no_config_sensor,
         multi_no_config_sensor,
         custom_interval_sensor,
+        running_in_code_sensor,
     ]
 
 

diff --git a/...dules/dagster-graphql/dagster_graphql_tests/graphql/snapshots/snap_test_partition_sets.py b/...dules/dagster-graphql/dagster_graphql_tests/graphql/snapshots/snap_test_partition_sets.py
@@ -1256,6 +1256,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1322,6 +1328,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1388,6 +1400,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1454,6 +1472,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1520,6 +1544,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1586,6 +1616,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1652,6 +1688,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1718,6 +1760,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1784,6 +1832,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',
@@ -1850,6 +1904,12 @@
                 'pipelineName': 'no_config_pipeline',
                 'solidSelection': None
             },
+            {
+                'mode': 'default',
+                'name': 'running_in_code_schedule_partitions',
+                'pipelineName': 'no_config_pipeline',
+                'solidSelection': None
+            },
             {
                 'mode': 'default',
                 'name': 'scheduled_integer_partitions',