Added test to check rowcount works after execmany returning tuples

diff --git a/tests/test_cursor.py b/tests/test_cursor.py
index 45750143e69061db454caadd806ee469dfa56e8f..ba121f9e6cdc6e5ee93649b32ebc22bd4a536f5b 100644 (file)
--- a/tests/test_cursor.py
+++ b/tests/test_cursor.py
@@ -180,6 +180,15 @@ def test_executemany_rowcount(conn, execmany):
     assert cur.rowcount == 2
 
 
+def test_executemany_returning_rowcount(conn, execmany):
+    cur = conn.cursor()
+    cur.executemany(
+        "insert into execmany(num, data) values (%s, %s) returning num",
+        [(10, "hello"), (20, "world")],
+    )
+    assert cur.rowcount == 2
+
+
 @pytest.mark.parametrize(
     "query",
     [

diff --git a/tests/test_cursor_async.py b/tests/test_cursor_async.py
index 7c3750b61104d0486b094200d74dcb35243b9042..68052270f702a36991e45d30f83ad2950e62de59 100644 (file)
--- a/tests/test_cursor_async.py
+++ b/tests/test_cursor_async.py
@@ -179,6 +179,15 @@ async def test_executemany_rowcount(aconn, execmany):
     assert cur.rowcount == 2
 
 
+async def test_executemany_returning_rowcount(aconn, execmany):
+    cur = await aconn.cursor()
+    await cur.executemany(
+        "insert into execmany(num, data) values (%s, %s) returning num",
+        [(10, "hello"), (20, "world")],
+    )
+    assert cur.rowcount == 2
+
+
 @pytest.mark.parametrize(
     "query",
     [