Added fetch method, changed parameters annotations. Added tests

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

Author: chandr-andr

python/tests/test_connection.py

@@ -26,6 +26,21 @@ async def test_connection_execute(
     assert len(conn_result.result()) == number_database_records
 
 
+async def test_connection_fetch(
+    psql_pool: ConnectionPool,
+    table_name: str,
+    number_database_records: int,
+) -> None:
+    """Test that single connection can fetch queries."""
+    connection = await psql_pool.connection()
+
+    conn_result = await connection.fetch(
+        querystring=f"SELECT * FROM {table_name}",
+    )
+    assert isinstance(conn_result, QueryResult)
+    assert len(conn_result.result()) == number_database_records
+
+
 async def test_connection_connection(
     psql_pool: ConnectionPool,
 ) -> None:

python/tests/test_connection_pool.py

@@ -49,6 +49,23 @@ async def test_pool_execute(
     assert len(inner_result) == number_database_records
 
 
+async def test_pool_fetch(
+    psql_pool: ConnectionPool,
+    table_name: str,
+    number_database_records: int,
+) -> None:
+    """Test that ConnectionPool can fetch queries."""
+    select_result = await psql_pool.fetch(
+        f"SELECT * FROM {table_name}",
+    )
+
+    assert type(select_result) == QueryResult
+
+    inner_result = select_result.result()
+    assert isinstance(inner_result, list)
+    assert len(inner_result) == number_database_records
+
+
 async def test_pool_connection(
     psql_pool: ConnectionPool,
 ) -> None:

python/tests/test_transaction.py

@@ -205,6 +205,21 @@ async def test_transaction_cursor(
         assert isinstance(cursor, Cursor)
 
 
+async def test_transaction_fetch(
+    psql_pool: ConnectionPool,
+    table_name: str,
+    number_database_records: int,
+) -> None:
+    """Test that single connection can fetch queries."""
+    connection = await psql_pool.connection()
+
+    async with connection.transaction() as transaction:
+        conn_result = await transaction.fetch(
+            querystring=f"SELECT * FROM {table_name}",
+        )
+    assert len(conn_result.result()) == number_database_records
+
+
 @pytest.mark.parametrize(
     ("insert_values"),
     [