Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 11 additions & 5 deletions tests/core/engine_adapter/integration/test_integration_postgres.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import typing as t
import uuid
from contextlib import contextmanager
import pytest
from pytest import FixtureRequest
Expand Down Expand Up @@ -53,7 +54,8 @@ def create_users(
_cleanup_user(engine_adapter, user_name)

for role_name in role_names:
user_name = f"test_{role_name}"
random_suffix = uuid.uuid4().hex[:6]
user_name = f"test_{role_name}_{random_suffix}"
password = random_id()
engine_adapter.execute(f"CREATE USER \"{user_name}\" WITH PASSWORD '{password}'")
engine_adapter.execute(f'GRANT USAGE ON SCHEMA public TO "{user_name}"')
Expand Down Expand Up @@ -381,12 +383,14 @@ def test_grants_plan_target_layer_physical_only(
with create_users(engine_adapter, "reader") as roles:
(tmp_path / "models").mkdir(exist_ok=True)

model_def = """
reader_username = roles["reader"]["username"]

model_def = f"""
MODEL (
name test_schema.physical_grants_model,
kind FULL,
grants (
'select' = ['test_reader']
'select' = ['{reader_username}']
),
grants_target_layer 'physical'
);
Expand Down Expand Up @@ -421,12 +425,14 @@ def test_grants_plan_target_layer_virtual_only(
with create_users(engine_adapter, "viewer") as roles:
(tmp_path / "models").mkdir(exist_ok=True)

model_def = """
viewer_username = roles["viewer"]["username"]

model_def = f"""
MODEL (
name test_schema.virtual_grants_model,
kind FULL,
grants (
'select' = ['test_viewer']
'select' = ['{viewer_username}']
),
grants_target_layer 'virtual'
);
Expand Down
Loading