test: reduce query size for v1 long test

tests/integration/conftest.py

@@ -1,5 +1,7 @@
 from logging import getLogger
 from os import environ
+from time import time
+from typing import Optional
 
 from pytest import fixture, mark
 
@@ -142,3 +144,20 @@ def engine_url() -> str:
 @fixture(scope="session")
 def stopped_engine_url() -> str:
     return must_env(STOPPED_ENGINE_URL_ENV)
+
+
+@fixture(scope="function")
+def minimal_time():
+    limit: Optional[float] = None
+
+    def setter(value):
+        nonlocal limit
+        limit = value
+
+    start = time()
+    yield setter
+    end = time()
+    if limit is not None:
+        assert (
+            end - start >= limit
+        ), f"Test took {end - start} seconds, less than {limit} seconds"

tests/integration/dbapi/async/V1/test_queries_async.py

@@ -1,6 +1,6 @@
 from datetime import date, datetime
 from decimal import Decimal
-from typing import Any, List
+from typing import Any, Callable, List
 
 from pytest import fixture, mark, raises
 
@@ -170,14 +170,19 @@ async def test_select(
 
 
 @mark.slow
-@mark.timeout(timeout=600)
+@mark.timeout(timeout=1000)
 async def test_long_query(
     connection: Connection,
+    minimal_time: Callable[[float], None],
 ) -> None:
     """AWS ALB TCP timeout set to 350; make sure we handle the keepalive correctly."""
+
+    # Fail test if it takes less than 350 seconds
+    minimal_time(350)
+
     with connection.cursor() as c:
         await c.execute(
-            "SELECT checksum(*) FROM GENERATE_SERIES(1, 200000000000)",  # approx 6m runtime
+            "SELECT checksum(*) FROM GENERATE_SERIES(1, 250000000000)",  # approx 6m runtime
         )
         data = await c.fetchall()
         assert len(data) == 1, "Invalid data size returned by fetchall"

tests/integration/dbapi/async/V2/test_queries_async.py

@@ -1,7 +1,7 @@
 from datetime import date, datetime
 from decimal import Decimal
 from os import environ
-from typing import List
+from typing import Callable, List
 
 from pytest import fixture, mark, raises
 
@@ -104,8 +104,12 @@ async def test_select(
 @mark.timeout(timeout=550)
 async def test_long_query(
     connection: Connection,
+    minimal_time: Callable[[float], None],
 ) -> None:
     """AWS ALB TCP timeout set to 350; make sure we handle the keepalive correctly."""
+
+    minimal_time(350)
+
     with connection.cursor() as c:
         await c.execute(
             "SELECT checksum(*) FROM GENERATE_SERIES(1, 200000000000)",  # approx 6m runtime

tests/integration/dbapi/sync/V1/test_queries.py

@@ -1,7 +1,7 @@
 from datetime import date, datetime
 from decimal import Decimal
 from threading import Thread
-from typing import Any, List
+from typing import Any, Callable, List
 
 from pytest import fixture, mark, raises
 
@@ -121,14 +121,18 @@ def test_select(
 
 
 @mark.slow
-@mark.timeout(timeout=600)
+@mark.timeout(timeout=1000)
 def test_long_query(
-    connection: Connection,
+    connection: Connection, minimal_time: Callable[[float], None]
 ) -> None:
     """AWS ALB TCP timeout set to 350, make sure we handle the keepalive correctly."""
+
+    # Fail test if it takes less than 350 seconds
+    minimal_time(350)
+
     with connection.cursor() as c:
         c.execute(
-            "SELECT checksum(*) FROM GENERATE_SERIES(1, 200000000000)",  # approx 6m runtime
+            "SELECT checksum(*) FROM GENERATE_SERIES(1, 250000000000)",  # approx 6m runtime
         )
         data = c.fetchall()
         assert len(data) == 1, "Invalid data size returned by fetchall"

tests/integration/dbapi/sync/V2/test_queries.py

@@ -2,7 +2,7 @@
 from decimal import Decimal
 from os import environ
 from threading import Thread
-from typing import Any, List
+from typing import Any, Callable, List
 
 from pytest import fixture, mark, raises
 
@@ -110,8 +110,12 @@ def test_select(
 @mark.timeout(timeout=550)
 def test_long_query(
     connection: Connection,
+    minimal_time: Callable[[float], None],
 ) -> None:
-    """AWS ALB TCP timeout set to 350, make sure we handle the keepalive correctly."""
+    """AWS ALB TCP timeout set to 350; make sure we handle the keepalive correctly."""
+
+    minimal_time(350)
+
     with connection.cursor() as c:
         c.execute(
             "SELECT checksum(*) FROM GENERATE_SERIES(1, 200000000000)",  # approx 6m runtime