Add tests on executemany to check the same problem of bug #112

diff --git a/tests/test_cursor.py b/tests/test_cursor.py
index 1c3fdd24619ece02107b333831dc9cf48b3dd741..eab60206facf4eed9dc54c53b192c4307d466563 100644 (file)
--- a/tests/test_cursor.py
+++ b/tests/test_cursor.py
@@ -104,6 +104,15 @@ def test_execute_type_change(conn):
     assert cur.fetchall() == [(1,), (100_000,)]
 
 
+def test_executemany_type_change(conn):
+    conn.execute("create table bug_112 (num integer)")
+    sql = "insert into bug_112 (num) values (%s)"
+    cur = conn.cursor()
+    cur.executemany(sql, [(1,), (100_000,)])
+    cur.execute("select num from bug_112 order by num")
+    assert cur.fetchall() == [(1,), (100_000,)]
+
+
 @pytest.mark.parametrize(
     "query", ["copy testcopy from stdin", "copy testcopy to stdout"]
 )

diff --git a/tests/test_cursor_async.py b/tests/test_cursor_async.py
index 4daab53bef56480050c8982a3875089dd20081d6..3aaa4356049b5f2c17e441d28d3552819b292aaf 100644 (file)
--- a/tests/test_cursor_async.py
+++ b/tests/test_cursor_async.py
@@ -107,6 +107,15 @@ async def test_execute_type_change(aconn):
     assert (await cur.fetchall()) == [(1,), (100_000,)]
 
 
+async def test_executemany_type_change(aconn):
+    await aconn.execute("create table bug_112 (num integer)")
+    sql = "insert into bug_112 (num) values (%s)"
+    cur = aconn.cursor()
+    await cur.executemany(sql, [(1,), (100_000,)])
+    await cur.execute("select num from bug_112 order by num")
+    assert (await cur.fetchall()) == [(1,), (100_000,)]
+
+
 @pytest.mark.parametrize(
     "query", ["copy testcopy from stdin", "copy testcopy to stdout"]
 )