feat: add new configuration for nightly retry-related tests. (#2319)

egreco12 · Evan Greco · gcf-owl-bot[bot] · web-flow · commit e78b84151d1e · 2023-11-14T15:14:40.000-08:00
---------

Co-authored-by: Evan Greco &lt;egreco@google.com&gt;
Co-authored-by: Owl Bot &lt;gcf-owl-bot[bot]@users.noreply.github.com&gt;
diff --git a/.kokoro/build.sh b/.kokoro/build.sh
@@ -115,6 +115,18 @@ clirr)
     mvn -B -Denforcer.skip=true clirr:check
     RETURN_CODE=$?
     ;;
+retry_quota)
+    mvn -B ${INTEGRATION_TEST_ARGS} \
+      -Dtest=ITBigQueryWriteQuotaRetryTest \
+      test
+    RETURN_CODE=$?
+    ;;
+retry_non_quota)
+    mvn -B ${INTEGRATION_TEST_ARGS} \
+      -Dtest=ITBigQueryWriteNonQuotaRetryTest \
+      test
+    RETURN_CODE=$?
+    ;;
 *)
     ;;
 esac
diff --git a/.kokoro/nightly/retry_non_quota.cfg b/.kokoro/nightly/retry_non_quota.cfg
@@ -0,0 +1,37 @@
+# Format: //devtools/kokoro/config/proto/build.proto
+
+env_vars: {
+  key: "TRAMPOLINE_IMAGE"
+  value: "gcr.io/cloud-devrel-kokoro-resources/java8"
+}
+
+# TODO: remove this after we've migrated all tests and scripts
+env_vars: {
+  key: "GCLOUD_PROJECT"
+  value: "bq-write-api-java-retry-test"
+}
+
+env_vars: {
+  key: "GOOGLE_CLOUD_PROJECT"
+  value: "bq-write-api-java-retry-test"
+}
+
+env_vars: {
+  key: "JOB_TYPE"
+  value: "retry_non_quota"
+}
+
+env_vars: {
+  key: "ENABLE_FLAKYBOT"
+  value: "true"
+}
+
+env_vars: {
+  key: "GOOGLE_APPLICATION_CREDENTIALS"
+  value: "secret_manager/java-it-service-account"
+}
+
+env_vars: {
+  key: "SECRET_MANAGER_KEYS"
+  value: "java-it-service-account"
+}
diff --git a/.kokoro/nightly/retry_quota.cfg b/.kokoro/nightly/retry_quota.cfg
@@ -0,0 +1,37 @@
+# Format: //devtools/kokoro/config/proto/build.proto
+
+env_vars: {
+  key: "TRAMPOLINE_IMAGE"
+  value: "gcr.io/cloud-devrel-kokoro-resources/java8"
+}
+
+# TODO: remove this after we've migrated all tests and scripts
+env_vars: {
+  key: "GCLOUD_PROJECT"
+  value: "bq-writeapi-java-quota-retry"
+}
+
+env_vars: {
+  key: "GOOGLE_CLOUD_PROJECT"
+  value: "bq-writeapi-java-quota-retry"
+}
+
+env_vars: {
+  key: "JOB_TYPE"
+  value: "retry_quota"
+}
+
+env_vars: {
+  key: "ENABLE_FLAKYBOT"
+  value: "true"
+}
+
+env_vars: {
+  key: "GOOGLE_APPLICATION_CREDENTIALS"
+  value: "secret_manager/java-it-service-account"
+}
+
+env_vars: {
+  key: "SECRET_MANAGER_KEYS"
+  value: "java-it-service-account"
+}
diff --git a/owlbot.py b/owlbot.py
@@ -25,6 +25,8 @@
     excludes=[
         "renovate.json",
         ".kokoro/build.sh",
+        ".kokoro/nightly/retry_non_quota.cfg",
+        ".kokoro/nightly/retry_quota.cfg",
         ".kokoro/nightly/samples.cfg",
         ".kokoro/presubmit/samples.cfg",
         ".kokoro/dependencies.sh",
