test: test Cursor.results() method

diff --git a/tests/test_cursor_common.py b/tests/test_cursor_common.py
index 2d104daf78a9d1dad4cf9908f2c411ed1c17957b..c0732c1b08b424940cae069c3a6f422b7858fed0 100644 (file)
--- a/tests/test_cursor_common.py
+++ b/tests/test_cursor_common.py
@@ -910,3 +910,18 @@ def test_row_maker_returns_none(conn):
     assert list(cur) == recs
     stream = cur.stream(query)
     assert list(stream) == recs
+
+
+@pytest.mark.parametrize("count", [1, 2])
+def test_results_after_execute(conn, count):
+    with conn.cursor() as cur:
+        cur.execute(";".join(["select 1"] * count))
+        assert list(cur.results()) == [cur] * count
+
+
+@pytest.mark.parametrize("count", [0, 1, 2])
+@pytest.mark.parametrize("returning", [False, True])
+def test_results_after_executemany(conn, count, returning):
+    with conn.cursor() as cur:
+        cur.executemany("select 1", [()] * count, returning=returning)
+        assert list(cur.results()) == [cur] * returning * count

diff --git a/tests/test_cursor_common_async.py b/tests/test_cursor_common_async.py
index e219846e65d30927a4690ab2eed9f0cc9bb017a7..8681175084535f6533305de8f53c79a05ed9aeda 100644 (file)
--- a/tests/test_cursor_common_async.py
+++ b/tests/test_cursor_common_async.py
@@ -918,3 +918,18 @@ async def test_row_maker_returns_none(aconn):
     assert await alist(cur) == recs
     stream = cur.stream(query)
     assert await alist(stream) == recs
+
+
+@pytest.mark.parametrize("count", [1, 2])
+async def test_results_after_execute(aconn, count):
+    async with aconn.cursor() as cur:
+        await cur.execute(";".join(["select 1"] * count))
+        assert await alist(cur.results()) == [cur] * count
+
+
+@pytest.mark.parametrize("count", [0, 1, 2])
+@pytest.mark.parametrize("returning", [False, True])
+async def test_results_after_executemany(aconn, count, returning):
+    async with aconn.cursor() as cur:
+        await cur.executemany("select 1", [()] * count, returning=returning)
+        assert await alist(cur.results()) == [cur] * returning * count

diff --git a/tests/test_cursor_server.py b/tests/test_cursor_server.py
index 7bcc305290ad69d51d372c02d2a24a7f9abd5092..a22571cceb81a0b92e6da92484d7cd939de1d7a0 100644 (file)
--- a/tests/test_cursor_server.py
+++ b/tests/test_cursor_server.py
@@ -581,3 +581,10 @@ def test_row_maker_returns_none(conn):
         assert list(cur) == recs
         stream = cur.stream(query)
         assert list(stream) == recs
+
+
+@pytest.mark.parametrize("count", [1])
+def test_results_after_execute(conn, count):
+    with conn.cursor() as cur:
+        cur.execute(";".join(["select 1"] * count))
+        assert list(cur.results()) == [cur] * count

diff --git a/tests/test_cursor_server_async.py b/tests/test_cursor_server_async.py
index f3c57d56b7bb584b79431616e40c09ed903fdfd2..8bfb5ac445444ec6dd0ee7e8cf0e474d1ef21a53 100644 (file)
--- a/tests/test_cursor_server_async.py
+++ b/tests/test_cursor_server_async.py
@@ -587,3 +587,10 @@ async def test_row_maker_returns_none(aconn):
         assert await alist(cur) == recs
         stream = cur.stream(query)
         assert await alist(stream) == recs
+
+
+@pytest.mark.parametrize("count", [1])
+async def test_results_after_execute(aconn, count):
+    async with aconn.cursor() as cur:
+        await cur.execute(";".join(["select 1"] * count))
+        assert await alist(cur.results()) == [cur] * count