Implemented execute_many in transaction

Author: insani7y

python/tests/helpers.py

@@ -0,0 +1,13 @@
+import typing
+
+from psqlpy import Transaction
+
+
+async def count_rows_in_test_table(
+    table_name: str,
+    transaction: Transaction,
+) -> int:
+    query_result: typing.Final = await transaction.execute(
+        f"SELECT COUNT(*) FROM {table_name}",
+    )
+    return query_result.result()[0]["count"]

python/tests/test_connection.py

@@ -2,8 +2,9 @@
 
 from psqlpy import PSQLPool, QueryResult, Transaction
 
+pytestmark = pytest.mark.anyio
+
 
-@pytest.mark.anyio()
 async def test_connection_execute(
     psql_pool: PSQLPool,
     table_name: str,
@@ -19,7 +20,6 @@ async def test_connection_execute(
     assert len(conn_result.result()) == number_database_records
 
 
-@pytest.mark.anyio()
 async def test_connection_transaction(
     psql_pool: PSQLPool,
 ) -> None:

python/tests/test_connection_pool.py

@@ -2,8 +2,9 @@
 
 from psqlpy import Connection, ConnRecyclingMethod, PSQLPool, QueryResult
 
+pytestmark = pytest.mark.anyio
+
 
-@pytest.mark.anyio()
 async def test_pool_dsn_startup() -> None:
     """Test that connection pool can startup with dsn."""
     pg_pool = PSQLPool(
@@ -14,7 +15,6 @@ async def test_pool_dsn_startup() -> None:
     await pg_pool.execute("SELECT 1")
 
 
-@pytest.mark.anyio()
 async def test_pool_execute(
     psql_pool: PSQLPool,
     table_name: str,
@@ -32,7 +32,6 @@ async def test_pool_execute(
     assert len(inner_result) == number_database_records
 
 
-@pytest.mark.anyio()
 async def test_pool_connection(
     psql_pool: PSQLPool,
 ) -> None:
@@ -41,7 +40,6 @@ async def test_pool_connection(
     assert isinstance(connection, Connection)
 
 
-@pytest.mark.anyio()
 @pytest.mark.parametrize(
     "conn_recycling_method",
     [

python/tests/test_cursor.py

@@ -2,8 +2,9 @@
 
 from psqlpy import Cursor
 
+pytestmark = pytest.mark.anyio
+
 
-@pytest.mark.anyio()
 async def test_cursor_fetch(
     number_database_records: int,
     test_cursor: Cursor,
@@ -13,7 +14,6 @@ async def test_cursor_fetch(
     assert len(result.result()) == number_database_records // 2
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_next(
     test_cursor: Cursor,
 ) -> None:
@@ -22,7 +22,6 @@ async def test_cursor_fetch_next(
     assert len(result.result()) == 1
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_prior(
     test_cursor: Cursor,
 ) -> None:
@@ -35,7 +34,6 @@ async def test_cursor_fetch_prior(
     assert len(result.result()) == 1
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_first(
     test_cursor: Cursor,
 ) -> None:
@@ -49,7 +47,6 @@ async def test_cursor_fetch_first(
     assert fetch_first.result() == first.result()
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_last(
     test_cursor: Cursor,
     number_database_records: int,
@@ -64,7 +61,6 @@ async def test_cursor_fetch_last(
     assert all_res.result()[-1] == last_res.result()[0]
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_absolute(
     test_cursor: Cursor,
     number_database_records: int,
@@ -85,7 +81,6 @@ async def test_cursor_fetch_absolute(
     assert all_res.result()[-1] == last_record.result()[0]
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_relative(
     test_cursor: Cursor,
     number_database_records: int,
@@ -107,7 +102,6 @@ async def test_cursor_fetch_relative(
     assert not (records.result())
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_forward_all(
     test_cursor: Cursor,
     number_database_records: int,
@@ -124,7 +118,6 @@ async def test_cursor_fetch_forward_all(
     )
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_backward(
     test_cursor: Cursor,
 ) -> None:
@@ -142,7 +135,6 @@ async def test_cursor_fetch_backward(
     assert len(must_not_be_empty.result()) == expected_number_of_results
 
 
-@pytest.mark.anyio()
 async def test_cursor_fetch_backward_all(
     test_cursor: Cursor,
 ) -> None:

python/tests/test_transaction.py

@@ -1,12 +1,16 @@
 from __future__ import annotations
 
+import typing
+
 import pytest
+from tests.helpers import count_rows_in_test_table
 
 from psqlpy import Cursor, IsolationLevel, PSQLPool, ReadVariant
 from psqlpy.exceptions import DBTransactionError, RustPSQLDriverPyBaseError
 
+pytestmark = pytest.mark.anyio
+
 
-@pytest.mark.anyio()
 @pytest.mark.parametrize(
     ("isolation_level", "deferrable", "read_variant"),
     [
@@ -42,7 +46,6 @@ async def test_transaction_init_parameters(
             assert read_variant is not ReadVariant.ReadOnly
 
 
-@pytest.mark.anyio()
 async def test_transaction_begin(
     psql_pool: PSQLPool,
     table_name: str,
@@ -66,7 +69,6 @@ async def test_transaction_begin(
     assert len(result.result()) == number_database_records
 
 
-@pytest.mark.anyio()
 async def test_transaction_commit(
     psql_pool: PSQLPool,
     table_name: str,
@@ -100,7 +102,6 @@ async def test_transaction_commit(
     assert len(result.result())
 
 
-@pytest.mark.anyio()
 async def test_transaction_savepoint(
     psql_pool: PSQLPool,
     table_name: str,
@@ -133,7 +134,6 @@ async def test_transaction_savepoint(
     await transaction.commit()
 
 
-@pytest.mark.anyio()
 async def test_transaction_rollback(
     psql_pool: PSQLPool,
     table_name: str,
@@ -171,7 +171,6 @@ async def test_transaction_rollback(
     assert not (result_from_conn.result())
 
 
-@pytest.mark.anyio()
 async def test_transaction_release_savepoint(
     psql_pool: PSQLPool,
 ) -> None:
@@ -194,7 +193,6 @@ async def test_transaction_release_savepoint(
     await transaction.savepoint(sp_name_1)
 
 
-@pytest.mark.anyio()
 async def test_transaction_cursor(
     psql_pool: PSQLPool,
     table_name: str,
@@ -205,3 +203,29 @@ async def test_transaction_cursor(
         cursor = await transaction.cursor(f"SELECT * FROM {table_name}")
 
         assert isinstance(cursor, Cursor)
+
+
+@pytest.mark.parametrize(
+    ("insert_values"),
+    [
+        [[1, "name1"], [2, "name2"]],
+        [[10, "name1"], [20, "name2"], [30, "name3"]],
+        [[1, "name1"]],
+    ],
+)
+async def test_transaction_execute_many(
+    psql_pool: PSQLPool,
+    table_name: str,
+    number_database_records: int,
+    insert_values: list[list[typing.Any]],
+) -> None:
+    connection = await psql_pool.connection()
+    async with connection.transaction() as transaction:
+        await transaction.execute_many(
+            f"INSERT INTO {table_name} VALUES ($1, $2)",
+            insert_values,
+        )
+        assert await count_rows_in_test_table(
+            table_name,
+            transaction,
+        ) - number_database_records == len(insert_values)