From 137e50c2eae6af372e99fe9e32d77e401d0f2777 Mon Sep 17 00:00:00 2001
From: Sakthivel Subramanian <sakthipvmj@gmail.com>
Date: Tue, 26 Nov 2024 12:03:57 +0530
Subject: [PATCH] ci(spanner): use airlock for presubmit jobs

---
 .kokoro/test.sh | 9 +++++++++
 package.json    | 3 ++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/.kokoro/test.sh b/.kokoro/test.sh
index 862d478d3..ad22f5a96 100755
--- a/.kokoro/test.sh
+++ b/.kokoro/test.sh
@@ -20,6 +20,15 @@ export NPM_CONFIG_PREFIX=${HOME}/.npm-global
 
 cd $(dirname $0)/..
 
+# Enable airlock only for Kokoro presubmit jobs
+if [[ ! -z "${KOKORO_JOB_TYPE}" && ${KOKORO_JOB_TYPE} =~ ^.*presubmit.*$ ]]; then
+  cat > .npmrc <<EOL
+registry=https://us-npm.pkg.dev/artifact-foundry-prod/npm-3p-trusted/
+//us-npm.pkg.dev/artifact-foundry-prod/npm-3p-trusted/:always-auth=true
+EOL
+  npm_config_registry=https://registry.npmjs.org npx google-artifactregistry-auth
+fi
+
 npm install
 # If tests are running against main branch, configure flakybot
 # to open issues on failures:
diff --git a/package.json b/package.json
index 3f880bd82..6cf311d28 100644
--- a/package.json
+++ b/package.json
@@ -51,7 +51,8 @@
     "prelint": "cd samples; npm link ../; npm install",
     "precompile": "gts clean",
     "compile-protos": "compileProtos src",
-    "coverage": "c8 mocha build/test build/test/common && c8 report --check-coverage"
+    "coverage": "c8 mocha build/test build/test/common && c8 report --check-coverage",
+    "artifactregistry-login": "npm_config_registry=https://registry.npmjs.org npx google-artifactregistry-auth"
   },
   "dependencies": {
     "@google-cloud/common": "^5.0.0",