Move test-mode-related parts from paths/schema.py to separate file

Author: bbliem

kausal_common

@@ -122,28 +122,21 @@ def node(self, info: SBInfo, id: str) -> SBNode:
         return SBNode(id=cast(sb.ID, node.id))
 
 
-SB_MUTATION_TYPES: list[type] = []
-if test_mode_enabled():
-    SB_MUTATION_TYPES.append(TestModeMutations)
-
-SBMutation: type | None = None
-if SB_MUTATION_TYPES:
-    SBMutation = merge_types('Mutation', tuple(SB_MUTATION_TYPES))
-
-
-def generate_strawberry_schema() -> sb.Schema:
+def generate_strawberry_schema(query: type, mutation: type | None = None) -> sb.Schema:
     from kausal_common.strawberry.registry import strawberry_types
 
     sb_schema = sb.Schema(
-        query=SBQuery, mutation=SBMutation, types=strawberry_types, directives=[context_directive]
+        # TODO: Add DjangoOptimizerExtension?
+        # https://strawberry.rocks/docs/django/guide/optimizer
+        query=query, mutation=mutation, types=strawberry_types, directives=[context_directive]
     )
     return sb_schema
 
 
-def generate_schema() -> tuple[sb.Schema, CombinedSchema]:
+def generate_schema(sb_query: type, sb_mutation: type | None = None) -> tuple[sb.Schema, CombinedSchema]:
     # We generate the Strawberry schema just to be able to utilize the
     # resolved GraphQL types directly in the Graphene schema.
-    sb_schema = generate_strawberry_schema()
+    sb_schema = generate_strawberry_schema(sb_query, sb_mutation)
 
     schema = CombinedSchema(
         sb_schema=sb_schema,
@@ -155,4 +148,4 @@ def generate_schema() -> tuple[sb.Schema, CombinedSchema]:
     return sb_schema, schema
 
 
-sb_schema, schema = generate_schema()
+sb_schema, schema = generate_schema(SBQuery)

paths/schema.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+from strawberry.tools import merge_types
+
+from kausal_common.deployment import test_mode_enabled
+from kausal_common.testing.schema import TestModeMutations
+
+from paths.schema import SBQuery, generate_schema
+
+SB_MUTATION_TYPES: list[type] = []
+if test_mode_enabled():
+    SB_MUTATION_TYPES.append(TestModeMutations)
+
+SBMutation: type | None = None
+if SB_MUTATION_TYPES:
+    SBMutation = merge_types('Mutation', tuple(SB_MUTATION_TYPES))
+
+sb_schema, schema = generate_schema(SBQuery, SBMutation)

paths/schema_test_mode.py

@@ -15,7 +15,7 @@
 from corsheaders.defaults import default_headers as default_cors_headers
 
 from kausal_common import ENV_SCHEMA as COMMON_ENV_SCHEMA, register_settings as register_common_settings
-from kausal_common.deployment import set_secret_file_vars
+from kausal_common.deployment import set_secret_file_vars, test_mode_enabled
 from kausal_common.deployment.http import get_allowed_cors_headers
 from kausal_common.sentry.init import init_sentry
 
@@ -290,8 +290,12 @@
 SESSION_COOKIE_SAMESITE = 'None'
 SESSION_COOKIE_SECURE = True
 
+if test_mode_enabled():
+    GRAPHENE_SCHEMA = f'{PROJECT_NAME}.schema_test_mode.schema'
+else:
+    GRAPHENE_SCHEMA = f'{PROJECT_NAME}.schema.schema'
 GRAPHENE = {
-    'SCHEMA': f'{PROJECT_NAME}.schema.schema',
+    'SCHEMA': GRAPHENE_SCHEMA,
 }
 GRAPPLE = {
     'APPS': ['pages'],