Spaces:
Running
Running
File size: 2,143 Bytes
698fc89 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | import pytest
from psycopg2.extensions import parse_dsn
from core.postgres_db import normalize_postgres_dsn
def test_normalize_postgres_dsn_standard():
dsn = "postgresql://user:secret123@db.supabase.co:5432/postgres"
norm = normalize_postgres_dsn(dsn)
parsed = parse_dsn(norm)
assert parsed["user"] == "user"
assert parsed["password"] == "secret123"
assert parsed["host"] == "db.supabase.co"
assert parsed["port"] == "5432"
assert parsed["dbname"] == "postgres"
def test_normalize_postgres_dsn_with_at_symbol_in_password():
"""
Prevents the libpq error:
invalid integer value "<host>:<port>" for connection option "port"
when password contains an unencoded '@' character.
"""
dsn = "postgresql://postgres.myproject:my@secret@atlas123asdqwat-1.pooler.supabase.com:5432/postgres"
norm = normalize_postgres_dsn(dsn)
parsed = parse_dsn(norm)
assert parsed["user"] == "postgres.myproject"
assert parsed["password"] == "my@secret"
assert parsed["host"] == "atlas123asdqwat-1.pooler.supabase.com"
assert parsed["port"] == "5432"
assert parsed["dbname"] == "postgres"
def test_normalize_postgres_dsn_with_special_characters_in_password():
dsn = "postgresql://postgres.ref:p@ss:w0rd!#%@aws-0-eu-central-1.pooler.supabase.com:6543/postgres?sslmode=require"
norm = normalize_postgres_dsn(dsn)
parsed = parse_dsn(norm)
assert parsed["user"] == "postgres.ref"
assert parsed["password"] == "p@ss:w0rd!#%"
assert parsed["host"] == "aws-0-eu-central-1.pooler.supabase.com"
assert parsed["port"] == "6543"
assert parsed["dbname"] == "postgres"
assert parsed["sslmode"] == "require"
def test_normalize_postgres_dsn_handles_postgres_scheme_and_quotes():
dsn = " 'postgres://postgres:pass123@localhost:5432/test_db' "
norm = normalize_postgres_dsn(dsn)
assert norm.startswith("postgresql://")
parsed = parse_dsn(norm)
assert parsed["user"] == "postgres"
assert parsed["password"] == "pass123"
assert parsed["host"] == "localhost"
assert parsed["port"] == "5432"
assert parsed["dbname"] == "test_db"
|