Fix engine compatibility

insani7y · insani7y · commit b7f0d9f2a1cf · 2024-06-25T20:36:15.000+03:00
diff --git a/psqlpy_piccolo/engine.py b/psqlpy_piccolo/engine.py
@@ -531,6 +531,8 @@ async def start_connection_pool(self: Self, **kwargs: Dict[str, Any]) -> None:
             config = dict(self.config)
             config.update(**kwargs)
             self.pool = ConnectionPool(
+                db_name=config.pop("database", None),
+                username=config.pop("user", None),
                 max_db_pool_size=config.pop("max_size"),
                 **config,
             )
@@ -545,11 +547,13 @@ async def close_connection_pool(self) -> None:
 
     async def get_new_connection(self) -> Connection:
         """Returns a new connection - doesn't retrieve it from the pool."""
+        config = dict(self.config)
         return await (
             ConnectionPool(
-                db_name=self.config.pop("database", None),
-                username=self.config.pop("user", None),
-                **self.config,
+                db_name=config.pop("database", None),
+                username=config.pop("user", None),
+                max_db_pool_size=config.pop("max_size"),
+                **config,
             )
         ).connection()
 
