From 48dd60b355bc32d6591fb62d9b054bafca72e2b7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tim=20Swe=C3=B1a=20=28Swast=2C=20formerly=29?=
 <swast@google.com>
Date: Wed, 6 Mar 2024 14:43:01 -0600
Subject: [PATCH 1/2] chore: increase timeout on load tests

---
 noxfile.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/noxfile.py b/noxfile.py
index bcd39f961f..db503c43fd 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -290,6 +290,7 @@ def run_system(
     install_test_extra=True,
     print_duration=False,
     extra_pytest_options=(),
+    timeout_seconds=900,
 ):
     """Run the system test suite."""
     constraints_path = str(
@@ -311,7 +312,7 @@ def run_system(
         "--quiet",
         "-n=20",
         # Any individual test taking longer than 15 mins will be terminated.
-        "--timeout=900",
+        f"--timeout={timeout_seconds}",
         # Log 20 slowest tests
         "--durations=20",
         f"--junitxml={prefix_name}_{session.python}_sponge_log.xml",
@@ -395,6 +396,7 @@ def load(session: nox.sessions.Session):
         prefix_name="load",
         test_folder=os.path.join("tests", "system", "load"),
         print_duration=True,
+        timeout_seconds=60 * 60,
     )
 
 

From b4242b7758816ff046355bb100eb8fc38e7347fe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tim=20Swe=C3=B1a=20=28Swast=2C=20formerly=29?=
 <swast@google.com>
Date: Wed, 6 Mar 2024 14:48:04 -0600
Subject: [PATCH 2/2] increase vm timeout too

---
 .kokoro/load/common.cfg | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.kokoro/load/common.cfg b/.kokoro/load/common.cfg
index 97e0651aa9..7f6fa7e0d9 100644
--- a/.kokoro/load/common.cfg
+++ b/.kokoro/load/common.cfg
@@ -8,3 +8,4 @@ action {
 }
 
 build_file: "python-bigquery-dataframes/.kokoro/build.sh"
+timeout_mins: 360