refactor(native): raise the shared native stage mismatch error for model config

Author: kvlonge

src/sqlbuild/compiler/compile/_helpers/attachment/model_config.py

@@ -26,7 +26,7 @@
     MODEL_AUDIT_OVERRIDE_KEYS,
     MODEL_HEADER_METADATA_KEYS,
 )
-from sqlbuild.compiler.compile.exceptions import CompileInputError, NativeModelConfigMismatchError
+from sqlbuild.compiler.compile.exceptions import CompileInputError
 from sqlbuild.compiler.compile.models import (
     CachedModelHeaderColumns,
     CompileModelConfig,
@@ -42,6 +42,7 @@
     DiscoveredSqlModelFile,
     ModelSchemaDeclaration,
 )
+from sqlbuild.compiler.frontier.exceptions import NativeStageMismatchError
 from sqlbuild.compiler.model_config.main._native_config_contains_macro_call import (
     native_config_contains_macro_call,
 )
@@ -359,7 +360,7 @@ def build_model_header_schema_entry(
         )
     )
     if native is not None and native.invalid:
-        raise NativeModelConfigMismatchError(
+        raise NativeStageMismatchError(
             f"{file_path}: native model config rejected MODEL columns or audits that the Python "
             "model config accepts; run with SQLBUILD_COMPILER_ENGINE=python"
         )

src/sqlbuild/compiler/compile/_helpers/render/context_templates.py

@@ -8,8 +8,8 @@
 
 from sqlbuild.compiler.compile._helpers.render.templating import expand_template_data
 from sqlbuild.compiler.compile.constants import COMPILE_INPUT_READS, PRESERVE_TARGET_VALUE
-from sqlbuild.compiler.compile.exceptions import NativeModelConfigMismatchError
 from sqlbuild.compiler.compile.types import CompileContextKey
+from sqlbuild.compiler.frontier.exceptions import NativeStageMismatchError
 from sqlbuild.compiler.model_config.constants import ENVIRONMENT_READ, INVALID_OUTCOME
 from sqlbuild.compiler.model_config.main._expand_native_config_templates import (
     expand_native_config_templates,
@@ -278,7 +278,7 @@ def expand_config_templates(
         preserve_unknown_context=preserve_unknown_context,
     )
     if outcome == INVALID_OUTCOME:
-        raise NativeModelConfigMismatchError(
+        raise NativeStageMismatchError(
             f"Native model config rejected {context_label} templates that the Python model "
             "config expands; run with SQLBUILD_COMPILER_ENGINE=python"
         )

src/sqlbuild/compiler/compile/exceptions.py

@@ -29,7 +29,3 @@ class CompactAnalysisInputError(CompileInputError):
 
 class AnalysisCacheEntryError(ValueError):
     """Raised when a persisted model analysis cache entry is invalid."""
-
-
-class NativeModelConfigMismatchError(RuntimeError):
-    """Raised when native model config rejects a value the Python model config accepts."""

tests/integration/src/sqlbuild/compiler/model_config/helpers.py

@@ -21,13 +21,14 @@
     expand_template_data,
 )
 from sqlbuild.compiler.compile.constants import COMPILE_INPUT_READS, MACRO_CALL_PATTERN
-from sqlbuild.compiler.compile.exceptions import CompileInputError, NativeModelConfigMismatchError
+from sqlbuild.compiler.compile.exceptions import CompileInputError
 from sqlbuild.compiler.compile.main._build_compile_inputs import build_compile_inputs
 from sqlbuild.compiler.compile.models import CompileAdapterContext, CompileProjectInputs
 from sqlbuild.compiler.discovery.main._model_schema_columns import parse_schema_columns
 from sqlbuild.compiler.discovery.main.discover import discover_project_inputs
 from sqlbuild.compiler.discovery.models import DiscoveredProjectInputs, DiscoveredSqlModelFile
 from sqlbuild.compiler.frontier.constants import COMPILER_ENGINE_ENV_VAR
+from sqlbuild.compiler.frontier.exceptions import NativeStageMismatchError
 from sqlbuild.compiler.model_config.constants import INVALID_OUTCOME, UNSUPPORTED_OUTCOME
 from sqlbuild.compiler.model_config.main._native_config_contains_macro_call import (
     native_config_contains_macro_call,
@@ -227,7 +228,7 @@ def native_rejection_error(
         _ = model_config_engine_outcome(
             project_dir=project_dir, engine="native-preview", monkeypatch=monkeypatch
         )
-    except NativeModelConfigMismatchError as error:
+    except NativeStageMismatchError as error:
         return str(error)
     return "no error"
 
@@ -332,7 +333,7 @@ def _native_expansion(value: object, flags: TemplateFlags) -> object:
             )
         except CompileInputError:
             return _RAISES
-        except NativeModelConfigMismatchError as error:
+        except NativeStageMismatchError as error:
             return str(error)
     return (_expansion_shape(result), reads.environment_names, reads.read_run_id)