sqlbuild 0.22.0__tar.gz → 0.22.1__tar.gz

sqlbuild-0.22.1/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.22.1"
+}

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## [0.22.1](https://github.com/chio-labs/sqlbuild/compare/v0.22.0...v0.22.1) (2026-05-28)
+
+
+### Bug Fixes
+
+* gate virtual operations on state status ([a89f806](https://github.com/chio-labs/sqlbuild/commit/a89f8063a04c54174e67dcd0f812dae968cd01f1))
+* gate virtual operations on state status ([c2edad2](https://github.com/chio-labs/sqlbuild/commit/c2edad2c79a9e3806d3294b234f35676271a03b8))
+
 ## [0.22.0](https://github.com/chio-labs/sqlbuild/compare/v0.21.0...v0.22.0) (2026-05-28)
 
 

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlbuild
-Version: 0.22.0
+Version: 0.22.1
 Summary: Typed, test-first SQL pipelines with local E2E testing
 License-File: LICENSE
 Requires-Python: >=3.12

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "sqlbuild"
-version = "0.22.0"
+version = "0.22.1"
 description = "Typed, test-first SQL pipelines with local E2E testing"
 readme = "README.md"
 requires-python = ">=3.12"

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/src/sqlbuild/cli/commands/main/diff.py

@@ -219,7 +219,15 @@ def _run_virtual_diff_cli(
         stream=sys.stdout,
         use_color=use_color,
     )
-    result, selected_names, skipped_names, from_stale, to_stale = run_virtual_diff(
+    (
+        result,
+        selected_names,
+        skipped_names,
+        from_stale,
+        to_stale,
+        from_working,
+        to_working,
+    ) = run_virtual_diff(
         project_dir=project_dir,
         discovered_inputs=discovered_inputs,
         adapter=adapter,
@@ -255,6 +263,8 @@ def _run_virtual_diff_cli(
         skipped_names=skipped_names,
         from_stale=from_stale,
         to_stale=to_stale,
+        from_working=from_working,
+        to_working=to_working,
         allow_partial_diff=allow_partial_diff,
         verbose=verbose,
         use_color=use_color,

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/src/sqlbuild/cli/commands/main/helpers/diff/virtual_output.py

@@ -13,6 +13,8 @@ def format_virtual_diff_header(
     skipped_names: tuple[str, ...],
     from_stale: tuple[str, ...],
     to_stale: tuple[str, ...],
+    from_working: bool,
+    to_working: bool,
     allow_partial_diff: bool,
     verbose: bool,
     use_color: bool,
@@ -27,8 +29,9 @@ def format_virtual_diff_header(
         len(selected_names) - len(skipped_names), use_color=use_color
     )
     skipped_count: str = _format_skipped_count(len(skipped_names), use_color=use_color)
-    working_label: str = "yes" if from_stale or to_stale else "no"
-    if from_stale or to_stale:
+    has_working_vde: bool = from_working or to_working
+    working_label: str = "yes" if has_working_vde else "no"
+    if has_working_vde:
         working_value: str = yellow_bold(working_label) if use_color else working_label
         if allow_partial_diff:
             suffix: str = dim(" (partial allowed)") if use_color else " (partial allowed)"

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/src/sqlbuild/virtual/diff/helpers/diff.py

@@ -15,7 +15,12 @@ from sqlbuild.compiler.pipeline.models import ProjectGraph
 from sqlbuild.compiler.planner.main.selection import resolve_project_selectors
 from sqlbuild.virtual.executor.main.rewrite import rewrite_virtual_project_model_targets
 from sqlbuild.virtual.planner.main.targets import build_virtual_target_from_physical_relation
-from sqlbuild.virtual.state.models import PhysicalRelationRecord, VirtualEnvironmentRefRecord
+from sqlbuild.virtual.state.models import (
+    PhysicalRelationRecord,
+    VirtualEnvironmentRecord,
+    VirtualEnvironmentRefRecord,
+)
+from sqlbuild.virtual.state.types import VirtualEnvironmentStatus
 
 
 def resolve_virtual_diff_model_names(
@@ -98,3 +103,17 @@ def rewrite_project_to_physical_relations(
         if model.name in relations
     }
     return rewrite_virtual_project_model_targets(project=project, rewritten_targets=targets)
+
+
+def non_finalized_environment_names(
+    environments: tuple[tuple[str, VirtualEnvironmentRecord | None], ...],
+) -> tuple[str, ...]:
+    return tuple(
+        name
+        for name, environment in environments
+        if environment is None or environment.status != VirtualEnvironmentStatus.FINALIZED
+    )
+
+
+def is_working_environment(environment: VirtualEnvironmentRecord | None) -> bool:
+    return environment is None or environment.status != VirtualEnvironmentStatus.FINALIZED

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/src/sqlbuild/virtual/diff/main/diff.py

@@ -18,6 +18,8 @@ from sqlbuild.executor.diff.models import DiffExecutionResult
 from sqlbuild.shared.types import ExternalSqlReferenceResolver
 from sqlbuild.virtual.diff.helpers.diff import (
     filter_models_with_changed_virtual_refs,
+    is_working_environment,
+    non_finalized_environment_names,
     read_physical_relations_for_refs,
     resolve_virtual_diff_model_names,
     rewrite_project_to_physical_relations,
@@ -28,6 +30,7 @@ from sqlbuild.virtual.state.main.runtime import build_state_runtime
 from sqlbuild.virtual.state.models import (
     ModelVersionRecord,
     PhysicalRelationRecord,
+    VirtualEnvironmentRecord,
     VirtualEnvironmentRefRecord,
 )
 
@@ -61,6 +64,8 @@ def run_virtual_diff(
     tuple[str, ...],
     tuple[str, ...],
     tuple[str, ...],
+    bool,
+    bool,
 ]:
     """Run a diff between two VDEs in the active physical environment."""
 
@@ -92,6 +97,16 @@ def run_virtual_diff(
         inspect_start: float = time.perf_counter()
         if on_progress is not None:
             on_progress("Inspecting virtual state...")
+        from_environment: VirtualEnvironmentRecord | None = backend.get_virtual_environment(
+            state_connection,
+            schema=config.schema,
+            virtual_environment_name=from_virtual_environment_name,
+        )
+        to_environment: VirtualEnvironmentRecord | None = backend.get_virtual_environment(
+            state_connection,
+            schema=config.schema,
+            virtual_environment_name=to_virtual_environment_name,
+        )
         from_refs: tuple[VirtualEnvironmentRefRecord, ...] = backend.get_virtual_environment_refs(
             state_connection,
             schema=config.schema,
@@ -135,13 +150,16 @@ def run_virtual_diff(
             bound_model_versions=to_model_versions,
         )
         if not select and not allow_partial_diff:
-            stale: tuple[str, ...] = tuple(
-                sorted({*from_semantics.stale_model_names, *to_semantics.stale_model_names})
+            non_finalized: tuple[str, ...] = non_finalized_environment_names(
+                (
+                    (from_virtual_environment_name, from_environment),
+                    (to_virtual_environment_name, to_environment),
+                )
             )
-            if stale:
+            if non_finalized:
                 raise PlannerInputError(
-                    "whole-VDE virtual diff requires finalized VDEs; stale models: "
-                    + ", ".join(stale),
+                    "whole-VDE virtual diff requires finalized VDEs; non-finalized VDEs: "
+                    + ", ".join(non_finalized),
                     code="S012",
                     help="Re-run with --allow-partial-diff to inspect a working VDE.",
                 )
@@ -185,6 +203,8 @@ def run_virtual_diff(
             skipped_names,
             from_semantics.stale_model_names,
             to_semantics.stale_model_names,
+            is_working_environment(from_environment),
+            is_working_environment(to_environment),
         )
 
     left_project: CompiledProject = rewrite_project_to_physical_relations(
@@ -230,6 +250,8 @@ def run_virtual_diff(
         skipped_names,
         from_semantics.stale_model_names,
         to_semantics.stale_model_names,
+        is_working_environment(from_environment),
+        is_working_environment(to_environment),
     )
 
 

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/src/sqlbuild/virtual/executor/main/promote.py

@@ -218,14 +218,6 @@ def run_virtual_promote(
                     code="S018",
                     help="Use --select for a coherent partial promotion from a working source VDE.",
                 )
-            if source_semantics.stale_model_names:
-                raise PlannerInputError(
-                    "whole-VDE promotion requires a source virtual environment that is current "
-                    "with the workspace; stale models: "
-                    + ", ".join(source_semantics.stale_model_names),
-                    code="S018",
-                    help="Build the source VDE or use --select for a coherent partial promotion.",
-                )
         source_ref_map: dict[str, str] = {ref.model_name: ref.version_hash for ref in source_refs}
         missing_source_refs: tuple[str, ...] = tuple(
             model_name for model_name in selected_model_names if model_name not in source_ref_map
@@ -247,6 +239,8 @@ def run_virtual_promote(
             if final_version_hashes.get(model.name)
             != target_semantics.expected_version_hashes.get(model.name)
         )
+        if not select:
+            stale_after = ()
         stale_upstreams: tuple[str, ...] = build_virtual_stale_required_upstream_closure(
             graph=graph,
             selected_model_names=selected_model_names,

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/src/sqlbuild/virtual/executor/main/rollback.py

@@ -21,7 +21,6 @@ from sqlbuild.virtual.executor.helpers.rollback import (
     guard_partial_rollback_scope,
     publish_function_versions,
     read_function_versions,
-    read_model_versions,
     read_physical_relations,
     resolve_selected_model_names,
     resolve_target_checkpoint,
@@ -29,8 +28,6 @@ from sqlbuild.virtual.executor.helpers.rollback import (
     validate_physical_relations_exist,
 )
 from sqlbuild.virtual.executor.main.views import refresh_logical_vde_views
-from sqlbuild.virtual.planner.main.semantics import build_virtual_plan_semantics
-from sqlbuild.virtual.planner.models import VirtualPlanSemantics
 from sqlbuild.virtual.state.main.locks import acquire_virtual_environment_lease
 from sqlbuild.virtual.state.main.release_lock import release_state_lease
 from sqlbuild.virtual.state.main.runtime import build_state_runtime
@@ -168,17 +165,6 @@ def run_virtual_rollback(
             schema=config.schema,
             checkpoint_id=target_checkpoint.checkpoint_id,
         )
-        current_model_versions: dict[str, Any] = read_model_versions(
-            backend=backend,
-            state_connection=state_connection,
-            schema=config.schema,
-            refs=current_refs,
-        )
-        current_semantics: VirtualPlanSemantics = build_virtual_plan_semantics(
-            graph=graph,
-            bound_refs=current_refs,
-            bound_model_versions=current_model_versions,
-        )
         selected_model_names: tuple[str, ...] = resolve_selected_model_names(
             graph=graph,
             select=select,
@@ -205,7 +191,7 @@ def run_virtual_rollback(
         stale_after: tuple[str, ...] = stale_after_rollback(
             graph=graph,
             final_version_hashes=final_version_hashes,
-            expected_version_hashes=current_semantics.expected_version_hashes,
+            expected_version_hashes=checkpoint_ref_map,
         )
         is_partial_scope: bool = bool(select or exclude)
         if is_partial_scope:
@@ -220,7 +206,7 @@ def run_virtual_rollback(
             stale_after = stale_after_rollback(
                 graph=graph,
                 final_version_hashes=final_version_hashes,
-                expected_version_hashes=current_semantics.expected_version_hashes,
+                expected_version_hashes=checkpoint_ref_map,
             )
             if stale_after and not allow_partial_rollback:
                 raise PlannerInputError(

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/tests/e2e/src/sqlbuild/cli/commands/main/build/test_virtual_build.py

@@ -1262,7 +1262,7 @@ def test_given_partial_rollback_when_allowed_then_it_marks_vde_working(
                 "stg_orders",
             ),
             expected_allowed_stdout_fragments=(
-                "status               active",
+                "status               finalized",
                 "rolled back models   2",
             ),
             expected_query_results=(("SELECT id FROM dev__dev.fact_orders ORDER BY id", ((1,),)),),
@@ -1309,6 +1309,127 @@ def test_given_partial_rollback_missing_stale_upstreams_when_including_them_then
         )
 
 
+@pytest.mark.parametrize(
+    "test_case",
+    [
+        VirtualPartialRollbackE2ETestCase(
+            description="partial rollback matches checkpoint when workspace changed",
+            blocked_command=(),
+            allowed_command=(
+                "--no-color",
+                "rollback",
+                "--select",
+                "fact_orders",
+                "--include-stale-upstreams",
+            ),
+            expected_blocked_stderr_fragments=(),
+            expected_allowed_stdout_fragments=(
+                "status               finalized",
+                "rolled back models   2",
+            ),
+            expected_query_results=(("SELECT id FROM dev__dev.fact_orders ORDER BY id", ((1,),)),),
+        )
+    ],
+    ids=["partial rollback matches checkpoint when workspace changed"],
+)
+def test_given_partial_rollback_matches_checkpoint_when_workspace_changed_then_no_override_needed(
+    test_case: VirtualPartialRollbackE2ETestCase,
+    tmp_path: Path,
+) -> None:
+    project_dir: Path = prepare_inline_project(
+        tmp_path=tmp_path,
+        project_name="virtual_rollback_stale_workspace",
+        repo_files=build_virtual_plan_repo_files(stg_orders_sql="SELECT 1 AS id"),
+    )
+    assert run_sqb(command=("state", "init"), project_dir=project_dir).returncode == 0
+    assert run_sqb(command=("--no-color", "build"), project_dir=project_dir).returncode == 0
+    (project_dir / "models" / "stg_orders.sql").write_text(
+        "MODEL ();\n\nSELECT 2 AS id\n",
+        encoding="utf-8",
+    )
+    assert run_sqb(command=("--no-color", "build"), project_dir=project_dir).returncode == 0
+    (project_dir / "models" / "stg_orders.sql").write_text(
+        "MODEL ();\n\nSELECT 3 AS id\n",
+        encoding="utf-8",
+    )
+
+    rollback_result: subprocess.CompletedProcess[str] = run_sqb(
+        command=test_case.allowed_command,
+        project_dir=project_dir,
+    )
+
+    assert rollback_result.returncode == 0, rollback_result.stdout + rollback_result.stderr
+    fragment: str
+    for fragment in test_case.expected_allowed_stdout_fragments:
+        assert fragment in rollback_result.stdout
+    for query_sql, expected_rows in test_case.expected_query_results:
+        assert query_duckdb(db_path=project_dir / "warehouse.duckdb", sql=query_sql) == list(
+            expected_rows
+        )
+
+
+@pytest.mark.parametrize(
+    "test_case",
+    [
+        VirtualPromoteE2ETestCase(
+            description="finalized source promotes after workspace changes again",
+            promote_command=("--no-color", "promote", "--from", "pr", "--to", "dev"),
+            expected_promote_fragments=(
+                "Virtual promotion complete",
+                "target status          finalized",
+            ),
+            expected_query_results=(("SELECT id FROM dev__dev.stg_orders ORDER BY id", ((2,),)),),
+        )
+    ],
+    ids=["finalized source promotes after workspace changes again"],
+)
+def test_given_finalized_source_vde_when_workspace_changes_again_then_whole_promotion_succeeds(
+    test_case: VirtualPromoteE2ETestCase,
+    tmp_path: Path,
+) -> None:
+    project_dir: Path = prepare_inline_project(
+        tmp_path=tmp_path,
+        project_name="virtual_promote_finalized_stale_workspace",
+        repo_files=build_virtual_plan_repo_files(stg_orders_sql="SELECT 1 AS id"),
+    )
+    init_result: subprocess.CompletedProcess[str] = run_sqb(
+        command=("state", "init"),
+        project_dir=project_dir,
+    )
+    assert init_result.returncode == 0, init_result.stderr
+    default_build_result: subprocess.CompletedProcess[str] = run_sqb(
+        command=("--no-color", "build"),
+        project_dir=project_dir,
+    )
+    assert default_build_result.returncode == 0, default_build_result.stderr
+    (project_dir / "models" / "stg_orders.sql").write_text(
+        "MODEL ();\n\nSELECT 2 AS id\n",
+        encoding="utf-8",
+    )
+    branch_build_result: subprocess.CompletedProcess[str] = run_sqb(
+        command=("--no-color", "build", "--virtual-env", "pr"),
+        project_dir=project_dir,
+    )
+    assert branch_build_result.returncode == 0, branch_build_result.stderr
+    (project_dir / "models" / "stg_orders.sql").write_text(
+        "MODEL ();\n\nSELECT 3 AS id\n",
+        encoding="utf-8",
+    )
+
+    promote_result: subprocess.CompletedProcess[str] = run_sqb(
+        command=test_case.promote_command,
+        project_dir=project_dir,
+    )
+
+    assert promote_result.returncode == 0, promote_result.stdout + promote_result.stderr
+    for fragment in test_case.expected_promote_fragments:
+        assert fragment in promote_result.stdout
+    for query_sql, expected_rows in test_case.expected_query_results:
+        assert query_duckdb(db_path=project_dir / "warehouse.duckdb", sql=query_sql) == list(
+            expected_rows
+        )
+
+
 @pytest.mark.parametrize(
     "test_case",
     [

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/tests/e2e/src/sqlbuild/cli/commands/main/diff/test_diff.py

@@ -317,12 +317,16 @@ DIFF_KEY_FAILURE_E2E_TEST_CASES: list[DiffKeyFailureE2ETestCase] = [
 
 VIRTUAL_DIFF_E2E_TEST_CASES: list[VirtualDiffE2ETestCase] = [
     VirtualDiffE2ETestCase(
-        description="whole VDE diff blocks when one side is working",
+        description="whole VDE diff allows finalized VDE stale against workspace",
         command=("--no-color", "diff", "dev:pr", "--schema-only"),
-        expected_exit_code=1,
-        expected_stderr_fragments=(
-            "whole-VDE virtual diff requires finalized VDEs",
-            "--allow-partial-diff",
+        expected_exit_code=0,
+        expected_stdout_fragments=(
+            "Virtual diff",
+            "working VDEs            no",
+            "selected models         3",
+            "compared models         2",
+            "unchanged refs skipped  1",
+            "SQLBuild Diff Summary",
         ),
     ),
     VirtualDiffE2ETestCase(
@@ -337,7 +341,7 @@ VIRTUAL_DIFF_E2E_TEST_CASES: list[VirtualDiffE2ETestCase] = [
         expected_exit_code=0,
         expected_stdout_fragments=(
             "Virtual diff",
-            "working VDEs            yes (partial allowed)",
+            "working VDEs            no",
             "selected models         3",
             "compared models         2",
             "unchanged refs skipped  1",
@@ -474,3 +478,59 @@ def test_given_virtual_diff_with_working_vde_when_running_then_it_respects_parti
         assert fragment in result.stdout, result.stdout + result.stderr
     for fragment in test_case.expected_stderr_fragments:
         assert fragment in result.stderr, result.stdout + result.stderr
+
+
+@pytest.mark.parametrize(
+    "test_case",
+    [
+        VirtualDiffE2ETestCase(
+            description="active VDE requires partial diff flag",
+            command=("--no-color", "diff", "dev:pr", "--schema-only"),
+            expected_exit_code=1,
+            expected_stderr_fragments=(
+                "whole-VDE virtual diff requires finalized VDEs",
+                "non-finalized VDEs: pr",
+                "--allow-partial-diff",
+            ),
+        )
+    ],
+    ids=["active VDE requires partial diff flag"],
+)
+def test_given_virtual_diff_with_active_vde_when_running_whole_diff_then_requires_partial_flag(
+    test_case: VirtualDiffE2ETestCase,
+    tmp_path: Path,
+) -> None:
+    project_dir: Path = prepare_inline_project(
+        tmp_path=tmp_path,
+        project_name="virtual_diff_active_guard",
+        repo_files=build_virtual_plan_repo_files(stg_orders_sql="SELECT 1 AS id"),
+    )
+    init_result: subprocess.CompletedProcess[str] = run_sqb(
+        command=("state", "init"),
+        project_dir=project_dir,
+    )
+    assert init_result.returncode == 0, init_result.stderr
+    default_build_result: subprocess.CompletedProcess[str] = run_sqb(
+        command=("--no-color", "build"),
+        project_dir=project_dir,
+    )
+    assert default_build_result.returncode == 0, default_build_result.stderr
+    (project_dir / "models" / "stg_orders.sql").write_text(
+        "MODEL ();\n\nSELECT 2 AS id\n",
+        encoding="utf-8",
+    )
+    branch_build_result: subprocess.CompletedProcess[str] = run_sqb(
+        command=("--no-color", "build", "--virtual-env", "pr", "--select", "stg_orders"),
+        project_dir=project_dir,
+    )
+    assert branch_build_result.returncode == 0, branch_build_result.stderr
+
+    result: subprocess.CompletedProcess[str] = run_sqb(
+        command=test_case.command,
+        project_dir=project_dir,
+    )
+
+    assert result.returncode == test_case.expected_exit_code, result.stdout + result.stderr
+    fragment: str
+    for fragment in test_case.expected_stderr_fragments:
+        assert fragment in result.stderr

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/tests/unit/src/sqlbuild/cli/commands/main/helpers/diff/_test_types.py

@@ -23,6 +23,8 @@ class RenderVirtualDiffHeaderTestCase:
     skipped_names: tuple[str, ...]
     from_stale: tuple[str, ...]
     to_stale: tuple[str, ...]
+    from_working: bool
+    to_working: bool
     allow_partial_diff: bool
     verbose: bool
     expected_fragments: tuple[str, ...]

{sqlbuild-0.22.0 → sqlbuild-0.22.1}/tests/unit/src/sqlbuild/cli/commands/main/helpers/diff/test_virtual_output.py

@@ -14,6 +14,8 @@ TEST_CASES: list[RenderVirtualDiffHeaderTestCase] = [
         skipped_names=("dim_customers",),
         from_stale=("fact_orders", "stg_orders"),
         to_stale=(),
+        from_working=True,
+        to_working=False,
         allow_partial_diff=True,
         verbose=False,
         expected_fragments=(
@@ -31,10 +33,13 @@ TEST_CASES: list[RenderVirtualDiffHeaderTestCase] = [
         skipped_names=(),
         from_stale=("fact_orders",),
         to_stale=("stg_orders",),
+        from_working=False,
+        to_working=False,
         allow_partial_diff=False,
         verbose=True,
         expected_fragments=(
             "compared models         3",
+            "working VDEs            no",
             "dev not current with workspace: fact_orders",
             "pr not current with workspace: stg_orders",
         ),
@@ -57,6 +62,8 @@ def test_given_virtual_diff_metadata_when_formatting_header_then_expected_detail
         skipped_names=test_case.skipped_names,
         from_stale=test_case.from_stale,
         to_stale=test_case.to_stale,
+        from_working=test_case.from_working,
+        to_working=test_case.to_working,
         allow_partial_diff=test_case.allow_partial_diff,
         verbose=test_case.verbose,
         use_color=False,

sqlbuild-0.22.0/.release-please-manifest.json

@@ -1,3 +0,0 @@
-{
-  ".": "0.22.0"
-}