test: fix results() test

Test a server-side cursor for real, drop unneeded parametrization.

diff --git a/tests/test_cursor_server.py b/tests/test_cursor_server.py
index a22571cceb81a0b92e6da92484d7cd939de1d7a0..cd680b29acf7c00ca6d88ba462ac71119a8398e3 100644 (file)
--- a/tests/test_cursor_server.py
+++ b/tests/test_cursor_server.py
@@ -583,8 +583,7 @@ def test_row_maker_returns_none(conn):
         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
+def test_results_after_execute(conn):
+    with conn.cursor("test") as cur:
+        cur.execute("select 1")
+        assert list(cur.results()) == [cur]

diff --git a/tests/test_cursor_server_async.py b/tests/test_cursor_server_async.py
index 8bfb5ac445444ec6dd0ee7e8cf0e474d1ef21a53..5a56b2c13e33fde05c9046db8378c69b32860aef 100644 (file)
--- a/tests/test_cursor_server_async.py
+++ b/tests/test_cursor_server_async.py
@@ -589,8 +589,7 @@ async def test_row_maker_returns_none(aconn):
         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
+async def test_results_after_execute(aconn):
+    async with aconn.cursor("test") as cur:
+        await cur.execute("select 1")
+        assert await alist(cur.results()) == [cur]