From 78d48f99f93b6738f4239c3f4bb718d3aa5cbb56 Mon Sep 17 00:00:00 2001
From: TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com>
Date: Thu, 2 Apr 2026 15:32:37 +0900
Subject: feat: add DB connection pooling with configurable pool_size,
 overflow, recycle

---
 shared/tests/test_db.py | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

(limited to 'shared/tests/test_db.py')

diff --git a/shared/tests/test_db.py b/shared/tests/test_db.py
index 239ee64..439a66e 100644
--- a/shared/tests/test_db.py
+++ b/shared/tests/test_db.py
@@ -100,6 +100,54 @@ class TestDatabaseConnect:
                     await db.connect()
                     mock_create.assert_called_once()
 
+    @pytest.mark.asyncio
+    async def test_connect_passes_pool_params_for_postgres(self):
+        from shared.db import Database
+
+        db = Database("postgresql+asyncpg://host/db")
+
+        mock_conn = AsyncMock()
+        mock_cm = AsyncMock()
+        mock_cm.__aenter__.return_value = mock_conn
+
+        mock_engine = MagicMock()
+        mock_engine.begin.return_value = mock_cm
+        mock_engine.dispose = AsyncMock()
+
+        with patch("shared.db.create_async_engine", return_value=mock_engine) as mock_create:
+            with patch("shared.db.async_sessionmaker"):
+                with patch("shared.db.Base") as mock_base:
+                    mock_base.metadata.create_all = MagicMock()
+                    await db.connect(pool_size=5, max_overflow=3, pool_recycle=1800)
+                    mock_create.assert_called_once_with(
+                        "postgresql+asyncpg://host/db",
+                        pool_pre_ping=True,
+                        pool_size=5,
+                        max_overflow=3,
+                        pool_recycle=1800,
+                    )
+
+    @pytest.mark.asyncio
+    async def test_connect_skips_pool_params_for_sqlite(self):
+        from shared.db import Database
+
+        db = Database("sqlite+aiosqlite:///test.db")
+
+        mock_conn = AsyncMock()
+        mock_cm = AsyncMock()
+        mock_cm.__aenter__.return_value = mock_conn
+
+        mock_engine = MagicMock()
+        mock_engine.begin.return_value = mock_cm
+        mock_engine.dispose = AsyncMock()
+
+        with patch("shared.db.create_async_engine", return_value=mock_engine) as mock_create:
+            with patch("shared.db.async_sessionmaker"):
+                with patch("shared.db.Base") as mock_base:
+                    mock_base.metadata.create_all = MagicMock()
+                    await db.connect()
+                    mock_create.assert_called_once_with("sqlite+aiosqlite:///test.db")
+
     @pytest.mark.asyncio
     async def test_init_tables_is_alias_for_connect(self):
         from shared.db import Database
-- 
cgit v1.2.3