Skip to content

Commit 9da034d

Browse files
newtonapplenewtonapple
committed
Change default grants_target_layer to VIRTUAL
1 parent 66350aa commit 9da034d

5 files changed

Lines changed: 37 additions & 21 deletions

File tree

‎sqlmesh/core/model/meta.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ class GrantsTargetLayer(str, Enum):
6262

6363
@classproperty
6464
def default(cls) -> "GrantsTargetLayer":
65-
return GrantsTargetLayer.ALL
65+
return GrantsTargetLayer.VIRTUAL
6666

6767
@property
6868
def is_all(self) -> bool:

‎tests/core/engine_adapter/integration/test_integration_postgres.py‎

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -475,7 +475,8 @@ def test_grants_plan(engine_adapter: PostgresEngineAdapter, ctx: TestContext, tm
475475
kind FULL,
476476
grants (
477477
'select' = ['test_analyst']
478-
)
478+
),
479+
grants_target_layer 'all'
479480
);
480481
SELECT 1 as id, CURRENT_DATE as created_date
481482
"""
@@ -510,7 +511,8 @@ def test_grants_plan(engine_adapter: PostgresEngineAdapter, ctx: TestContext, tm
510511
grants (
511512
'select' = ['test_analyst', 'test_etl_user'],
512513
'insert' = ['test_etl_user']
513-
)
514+
),
515+
grants_target_layer 'all'
514516
);
515517
SELECT 1 as id, CURRENT_DATE as created_date, 'v2' as version
516518
"""
@@ -636,7 +638,8 @@ def test_grants_plan_full_refresh_model_via_replace(
636638
kind FULL,
637639
grants (
638640
'SELECT' = ['{roles["reader"]["username"]}']
639-
)
641+
),
642+
grants_target_layer 'all'
640643
);
641644
SELECT 1 as id, 'test_data' as status
642645
"""
@@ -683,7 +686,8 @@ def test_grants_plan_incremental_model_first_insert(
683686
),
684687
grants (
685688
'SELECT' = ['{roles["reader"]["username"]}']
686-
)
689+
),
690+
grants_target_layer 'all'
687691
);
688692
689693
SELECT 1 as id, @start_ds::timestamp as ts, 'data' as value
@@ -727,7 +731,8 @@ def test_grants_plan_clone_environment(
727731
kind FULL,
728732
grants (
729733
'SELECT' = ['{roles["reader"]["username"]}']
730-
)
734+
),
735+
grants_target_layer 'all'
731736
);
732737
733738
SELECT 1 as id, 'data' as value
@@ -783,7 +788,8 @@ def test_grants_metadata_only_changes(
783788
kind FULL,
784789
grants (
785790
'select' = ['{roles["reader"]["username"]}']
786-
)
791+
),
792+
grants_target_layer 'all'
787793
);
788794
SELECT 1 as id, 'unchanged_query' as data
789795
"""
@@ -817,7 +823,8 @@ def test_grants_metadata_only_changes(
817823
grants (
818824
'select' = ['{roles["writer"]["username"]}', '{roles["admin"]["username"]}'],
819825
'insert' = ['{roles["admin"]["username"]}']
820-
)
826+
),
827+
grants_target_layer 'all'
821828
);
822829
SELECT 1 as id, 'unchanged_query' as data
823830
"""
@@ -851,7 +858,8 @@ def test_grants_metadata_only_changes(
851858
kind FULL,
852859
grants (
853860
'select' = ['{roles["reader"]["username"]}']
854-
)
861+
),
862+
grants_target_layer 'all'
855863
);
856864
SELECT 1 as id, 'unchanged_query' as data
857865
"""

‎tests/core/test_model.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1927,7 +1927,7 @@ def test_render_definition_with_defaults():
19271927
materialized FALSE
19281928
),
19291929
virtual_environment_mode 'full',
1930-
grants_target_layer 'all'
1930+
grants_target_layer 'virtual'
19311931
);
19321932
19331933
{query}
@@ -2021,7 +2021,7 @@ def test_render_definition_with_grants():
20212021
"""
20222022
)
20232023
default_model = load_sql_based_model(default_expressions)
2024-
assert default_model.grants_target_layer == GrantsTargetLayer.ALL # default value
2024+
assert default_model.grants_target_layer == GrantsTargetLayer.VIRTUAL # default value
20252025

20262026

20272027
def test_render_definition_partitioned_by():

‎tests/core/test_snapshot.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def test_json(snapshot: Snapshot):
168168
"enabled": True,
169169
"extract_dependencies_from_query": True,
170170
"virtual_environment_mode": "full",
171-
"grants_target_layer": "all",
171+
"grants_target_layer": "virtual",
172172
},
173173
"name": '"name"',
174174
"parents": [{"name": '"parent"."tbl"', "identifier": snapshot.parents[0].identifier}],
@@ -985,7 +985,7 @@ def test_fingerprint(model: Model, parent_model: Model):
985985

986986
original_fingerprint = SnapshotFingerprint(
987987
data_hash="2406542604",
988-
metadata_hash="185287368",
988+
metadata_hash="1056339358",
989989
)
990990

991991
assert fingerprint == original_fingerprint
@@ -1046,7 +1046,7 @@ def test_fingerprint_seed_model():
10461046

10471047
expected_fingerprint = SnapshotFingerprint(
10481048
data_hash="1586624913",
1049-
metadata_hash="1817881990",
1049+
metadata_hash="2674364560",
10501050
)
10511051

10521052
model = load_sql_based_model(expressions, path=Path("./examples/sushi/models/test_model.sql"))
@@ -1085,7 +1085,7 @@ def test_fingerprint_jinja_macros(model: Model):
10851085
)
10861086
original_fingerprint = SnapshotFingerprint(
10871087
data_hash="93332825",
1088-
metadata_hash="185287368",
1088+
metadata_hash="1056339358",
10891089
)
10901090

10911091
fingerprint = fingerprint_from_node(model, nodes={})

‎tests/core/test_snapshot_evaluator.py‎

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4777,7 +4777,9 @@ def test_grants_create_model_kind(
47774777
sync_grants_mock = mocker.patch.object(adapter_mock, "sync_grants_config")
47784778

47794779
grants = {"select": ["user1"]}
4780-
model = _create_grants_test_model(grants=grants, kind=model_kind_name)
4780+
model = _create_grants_test_model(
4781+
grants=grants, kind=model_kind_name, grants_target_layer=GrantsTargetLayer.ALL
4782+
)
47814783
snapshot = make_snapshot(model)
47824784

47834785
evaluator = SnapshotEvaluator(adapter_mock)
@@ -4846,6 +4848,7 @@ def test_grants_update(
48464848
parse_one("SELECT 1 as id"),
48474849
kind="FULL",
48484850
grants={"select": ["user1"]},
4851+
grants_target_layer=GrantsTargetLayer.ALL,
48494852
)
48504853

48514854
snapshot = make_snapshot(model)
@@ -4903,7 +4906,8 @@ def test_grants_create_and_evaluate(
49034906
grants (
49044907
'select' = ['reader1', 'reader2'],
49054908
'insert' = ['writer']
4906-
)
4909+
),
4910+
grants_target_layer 'all'
49074911
);
49084912
SELECT ds::DATE, value::INT FROM source WHERE ds BETWEEN @start_ds AND @end_ds;
49094913
"""
@@ -4952,7 +4956,7 @@ def test_grants_materializable_strategy_migrate(
49524956
sync_grants_mock = mocker.patch.object(adapter_mock, "sync_grants_config")
49534957
strategy = strategy_class(adapter_mock)
49544958
grants = {"select": ["user1"]}
4955-
model = _create_grants_test_model(grants=grants)
4959+
model = _create_grants_test_model(grants=grants, grants_target_layer=GrantsTargetLayer.ALL)
49564960
snapshot = make_snapshot(model)
49574961

49584962
strategy.migrate(
@@ -4977,7 +4981,7 @@ def test_grants_clone_snapshot_in_dev(
49774981

49784982
evaluator = SnapshotEvaluator(adapter_mock)
49794983
grants = {"select": ["user1", "user2"]}
4980-
model = _create_grants_test_model(grants=grants)
4984+
model = _create_grants_test_model(grants=grants, grants_target_layer=GrantsTargetLayer.ALL)
49814985
snapshot = make_snapshot(model)
49824986
snapshot.categorize_as(SnapshotChangeCategory.BREAKING)
49834987

@@ -5013,7 +5017,9 @@ def test_grants_evaluator_insert_without_replace_query_for_model(
50135017
evaluator = SnapshotEvaluator(adapter_mock)
50145018

50155019
grants = {"select": ["reader1", "reader2"]}
5016-
model = _create_grants_test_model(grants=grants, kind=model_kind_name)
5020+
model = _create_grants_test_model(
5021+
grants=grants, kind=model_kind_name, grants_target_layer=GrantsTargetLayer.ALL
5022+
)
50175023
snapshot = make_snapshot(model)
50185024
snapshot.categorize_as(SnapshotChangeCategory.BREAKING)
50195025

@@ -5070,7 +5076,9 @@ def test_grants_evaluator_insert_with_replace_query_for_model(
50705076
evaluator = SnapshotEvaluator(adapter_mock)
50715077

50725078
grants = {"select": ["user1"]}
5073-
model = _create_grants_test_model(grants=grants, kind=model_kind_name)
5079+
model = _create_grants_test_model(
5080+
grants=grants, kind=model_kind_name, grants_target_layer=GrantsTargetLayer.ALL
5081+
)
50745082
snapshot = make_snapshot(model)
50755083
snapshot.categorize_as(SnapshotChangeCategory.BREAKING)
50765084

0 commit comments

Comments
 (0)