chore: import upstream snapshot with attribution

2026-07-13 13:12:00 +08:00
commit 3de48288cb
2986 changed files with 1131193 additions and 0 deletions
@@ -0,0 +1,37 @@
+import pytest
+
+from omnigent.db.utils import normalize_database_url
+
+
+@pytest.mark.parametrize(
+    "input_url, expected",
+    [
+        # PaaS providers emit postgres:// (shorthand form)
+        (
+            "postgres://user:pw@host:5432/db",
+            "postgresql+psycopg://user:pw@host:5432/db",
+        ),
+        # Standard form without driver specifier
+        (
+            "postgresql://user:pw@host:5432/db",
+            "postgresql+psycopg://user:pw@host:5432/db",
+        ),
+        # Already correct — pass through unchanged
+        (
+            "postgresql+psycopg://user:pw@host:5432/db",
+            "postgresql+psycopg://user:pw@host:5432/db",
+        ),
+        # SQLite (local dev) — pass through unchanged
+        (
+            "sqlite:///./omnigent.db",
+            "sqlite:///./omnigent.db",
+        ),
+        # Credentials with special characters survive the prefix rewrite
+        (
+            "postgres://user:p%40ssword@host/db",
+            "postgresql+psycopg://user:p%40ssword@host/db",
+        ),
+    ],
+)
+def test_normalize_database_url(input_url: str, expected: str) -> None:
+    assert normalize_database_url(input_url) == expected