Skip to content

Commit 58947c2

Browse files
fix(postgres): format test_integration_postgres.py with ruff
Signed-off-by: sakshipatil-hue <sakshipatil8674788@gmail.com>
1 parent e6ac0e5 commit 58947c2

1 file changed

Lines changed: 11 additions & 5 deletions

File tree

tests/core/engine_adapter/integration/test_integration_postgres.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import typing as t
2+
import uuid
23
from contextlib import contextmanager
34
import pytest
45
from pytest import FixtureRequest
@@ -53,7 +54,8 @@ def create_users(
5354
_cleanup_user(engine_adapter, user_name)
5455

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

384-
model_def = """
386+
reader_username = roles["reader"]["username"]
387+
388+
model_def = f"""
385389
MODEL (
386390
name test_schema.physical_grants_model,
387391
kind FULL,
388392
grants (
389-
'select' = ['test_reader']
393+
'select' = ['{reader_username}']
390394
),
391395
grants_target_layer 'physical'
392396
);
@@ -421,12 +425,14 @@ def test_grants_plan_target_layer_virtual_only(
421425
with create_users(engine_adapter, "viewer") as roles:
422426
(tmp_path / "models").mkdir(exist_ok=True)
423427

424-
model_def = """
428+
viewer_username = roles["viewer"]["username"]
429+
430+
model_def = f"""
425431
MODEL (
426432
name test_schema.virtual_grants_model,
427433
kind FULL,
428434
grants (
429-
'select' = ['test_viewer']
435+
'select' = ['{viewer_username}']
430436
),
431437
grants_target_layer 'virtual'
432438
);

0 commit comments

Comments
 (0)