feat: complete production workflow migration

2026-08-06 14:29:57 +08:00
parent 7f215e79c4
commit 8df5266abb
448 changed files with 56937 additions and 14619 deletions
@@ -0,0 +1,42 @@
+import unittest
+
+from db import build_pg_config
+
+
+class DatabaseConfigTests(unittest.TestCase):
+    def test_complete_config_is_parsed(self):
+        config = build_pg_config(
+            {
+                "PG_HOST": "db.example.test",
+                "PG_PORT": "5544",
+                "PG_DB": "warehouse",
+                "PG_USER": "collector",
+                "PG_PASSWORD": "secret",
+            }
+        )
+
+        self.assertEqual(
+            config,
+            {
+                "host": "db.example.test",
+                "port": 5544,
+                "dbname": "warehouse",
+                "user": "collector",
+                "password": "secret",
+            },
+        )
+
+    def test_missing_final_config_is_rejected(self):
+        with self.assertRaisesRegex(RuntimeError, "PG_HOST"):
+            build_pg_config(
+                {
+                    "PG_PORT": "5432",
+                    "PG_DB": "warehouse",
+                    "PG_USER": "collector",
+                    "PG_PASSWORD": "secret",
+                }
+            )
+
+
+if __name__ == "__main__":
+    unittest.main()