Fix engine compatibility

insani7y · insani7y · commit 493ad077cf10 · 2024-06-25T20:10:58.000+03:00
diff --git a/psqlpy_piccolo/engine.py b/psqlpy_piccolo/engine.py
@@ -485,7 +485,11 @@ async def prep_database(self: Self) -> None:
                     level=Level.medium,
                 )
 
-    async def start_connnection_pool(self: Self, **kwargs: Dict[str, Any]) -> None:
+    async def start_connnection_pool(
+        self: Self,
+        max_pool_size: int = 1,
+        **kwargs: Dict[str, Any],
+    ) -> None:
         """Start new connection pool.
 
         Create and start new connection pool.
@@ -527,7 +531,10 @@ async def start_connection_pool(self: Self, **kwargs: Dict[str, Any]) -> None:
         else:
             config = dict(self.config)
             config.update(**kwargs)
-            self.pool = ConnectionPool(**config)
+            self.pool = ConnectionPool(
+                max_db_pool_size=config.pop("max_size"),
+                **config,
+            )
 
     async def close_connection_pool(self) -> None:
         """Close connection pool."""
