fix test timeout

Author: ptiurin

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

@@ -1,13 +1,21 @@
+from typing import Callable
+
 from pytest import raises
 
 from firebolt.async_db import Connection
 from firebolt.utils.exception import QueryTimeoutError
+from tests.integration.dbapi.conftest import LONG_SELECT_DEFAULT_V2
 
 # Cannot have absolute path here since interpreter treats word async as keyword and not as a package name
 from .test_queries_async import LONG_SELECT
 
 
-async def test_query_timeout(connection: Connection):
+async def test_query_timeout(
+    connection: Connection, long_test_value: Callable[[int], int]
+) -> None:
     async with connection.cursor() as cursor:
         with raises(QueryTimeoutError):
-            await cursor.execute(LONG_SELECT, timeout_seconds=1)
+            await cursor.execute(
+                LONG_SELECT.format(long_value=long_test_value(LONG_SELECT_DEFAULT_V2)),
+                timeout_seconds=1,
+            )

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

@@ -1,12 +1,20 @@
+from typing import Callable
+
 from pytest import raises
 
 from firebolt.db import Connection
 from firebolt.utils.exception import QueryTimeoutError
+from tests.integration.dbapi.conftest import LONG_SELECT_DEFAULT_V2
 
 from .test_queries import LONG_SELECT
 
 
-def test_query_timeout(connection: Connection):
+def test_query_timeout(
+    connection: Connection, long_test_value: Callable[[int], int]
+) -> None:
     with connection.cursor() as cursor:
         with raises(QueryTimeoutError):
-            cursor.execute(LONG_SELECT, timeout_seconds=1)
+            cursor.execute(
+                LONG_SELECT.format(long_value=long_test_value(LONG_SELECT_DEFAULT_V2)),
+                timeout_seconds=1,
+            )