Added tests for custom decoder and encoder

Signed-off-by: chandr-andr (Kiselev Aleksandr) <[email protected]>

Author: chandr-andr

python/psqlpy/_internal/__init__.pyi

@@ -13,7 +13,7 @@ class QueryResult:
 
     def result(
         self: Self,
-        custom_decoders: dict[str, Callable[[list[int]], Any]] | None = None,
+        custom_decoders: dict[str, Callable[[bytes], Any]] | None = None,
     ) -> list[dict[Any, Any]]:
         """Return result from database as a list of dicts."""
     def as_class(

python/tests/test_value_converter.py

@@ -8,6 +8,7 @@
 from tests.conftest import DefaultPydanticModel, DefaultPythonModelClass
 
 from psqlpy import ConnectionPool
+from psqlpy._internal.extra_types import PyCustomType
 from psqlpy.extra_types import (
     BigInt,
     Integer,
@@ -425,3 +426,52 @@ class TopLevelModel(BaseModel):
     )
 
     assert isinstance(model_result[0], TopLevelModel)
+
+
+async def test_custom_type_as_parameter(
+    psql_pool: ConnectionPool,
+) -> None:
+    """Tests that we can use `PyCustomType`."""
+    await psql_pool.execute("DROP TABLE IF EXISTS for_test")
+    await psql_pool.execute(
+        "CREATE TABLE for_test (nickname VARCHAR)",
+    )
+
+    await psql_pool.execute(
+        querystring="INSERT INTO for_test VALUES ($1)",
+        parameters=[PyCustomType(b"Some Real Nickname")],
+    )
+
+    qs_result = await psql_pool.execute(
+        "SELECT * FROM for_test",
+    )
+
+    result = qs_result.result()
+    assert result[0]["nickname"] == "Some Real Nickname"
+
+
+async def test_custom_decoder(
+    psql_pool: ConnectionPool,
+) -> None:
+    await psql_pool.execute("DROP TABLE IF EXISTS for_test")
+    await psql_pool.execute(
+        "CREATE TABLE for_test (geo_point POINT)",
+    )
+
+    await psql_pool.execute(
+        "INSERT INTO for_test VALUES ('(1, 1)')",
+    )
+
+    def point_encoder(point_bytes: bytes) -> str:
+        return "Just An Example"
+
+    qs_result = await psql_pool.execute(
+        "SELECT * FROM for_test",
+    )
+    result = qs_result.result(
+        custom_decoders={
+            "geo_point": point_encoder,
+        },
+    )
+
+    assert result[0]["geo_point"] == "Just An Example"