From 37c9e3eb7a8e4b69bc5b8d050b00545b67b301f9 Mon Sep 17 00:00:00 2001
From: Christina Holland <chholland@google.com>
Date: Thu, 19 Dec 2024 12:07:05 -0800
Subject: [PATCH] Set e2e test workflow to poll npm to check the version is
 available

---
 .github/workflows/e2e-test.yml      |  6 +++
 scripts/release/poll-npm-publish.js | 68 +++++++++++++++++++++++++++++
 2 files changed, 74 insertions(+)
 create mode 100644 scripts/release/poll-npm-publish.js

diff --git a/.github/workflows/e2e-test.yml b/.github/workflows/e2e-test.yml
index f22f77e0fe5..0857860571f 100644
--- a/.github/workflows/e2e-test.yml
+++ b/.github/workflows/e2e-test.yml
@@ -50,6 +50,12 @@ jobs:
           TEST_ACCOUNT: ${{ secrets.TEST_ACCOUNT }}
         run: |
           echo "export const config = $PROJECT_CONFIG; export const testAccount = $TEST_ACCOUNT" > firebase-config.js
+      - name: Poll npm until version to test is available for install
+        run: |
+          echo "Polling npm for firebase@${{ github.event.client_payload.versionOrTag }}"
+          node ../scripts/release/poll-npm-publish.js
+        env:
+          VERSION: ${{ github.event.client_payload.versionOrTag }}
       - name: Yarn install
         run: |
           echo "Installing firebase@${{ github.event.client_payload.versionOrTag }}"
diff --git a/scripts/release/poll-npm-publish.js b/scripts/release/poll-npm-publish.js
new file mode 100644
index 00000000000..029262bbb12
--- /dev/null
+++ b/scripts/release/poll-npm-publish.js
@@ -0,0 +1,68 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+const { exec } = require('child_process');
+
+const MAX_ATTEMPTS = 15;
+const RETRY_DELAY_SECONDS = 60;
+
+async function pollNpmPublish() {
+  const version = process.env.VERSION;
+
+  if (!version) {
+    console.log(`Couldn't find env var VERSION.`);
+    return;
+  }
+
+  const getNpmPublishedVersion = () =>
+    new Promise((resolve, reject) => {
+      exec(`npm view firebase@${version} version`, (error, stdout) => {
+        if (error) {
+          reject(error);
+        }
+        const version = stdout.trim();
+        if (!version.match(/^\d+\.\d+\.\d+$/)) {
+          reject(
+            new Error(
+              `npm view did not return a valid semver version. Received: ${version}`
+            )
+          );
+        }
+        resolve(version);
+      });
+    });
+  for (let i = 0; i < MAX_ATTEMPTS; i++) {
+    const latestPublishedVersion = await getNpmPublishedVersion();
+    if (latestPublishedVersion === process.env.VERSION) {
+      console.log(`Found firebase@${version} in the npm registry.`);
+      return;
+    }
+    console.log(`Didn't find firebase@${version} in the npm registry.`);
+    if (i < MAX_ATTEMPTS - 1) {
+      console.log(`Trying again in ${RETRY_DELAY_SECONDS} seconds.`);
+      await new Promise(resolve =>
+        setTimeout(resolve, RETRY_DELAY_SECONDS * 1000)
+      );
+    }
+  }
+  console.log(
+    `Was not able to find firebase@${version} on npm. Ending process.`
+  );
+  process.exit(1);
+}
+
+pollNpmPublish();