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

---------

Co-authored-by: Evan Greco <egreco@google.com>
Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>

.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

.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"
+}

.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"
+}

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",