From c3bbf8963d6c344226c6e32d80e76d438c62aa17 Mon Sep 17 00:00:00 2001
From: Ross Wollman <rwoll@users.noreply.github.com>
Date: Mon, 20 Jun 2022 11:24:23 -0700
Subject: [PATCH] test: installation test for CDN failover (#14963)

---
 .../src/server/registry/index.ts              |  8 +++
 ...laywright-cdn-failover-should-work.spec.ts | 50 +++++++++++++++++++
 2 files changed, 58 insertions(+)
 create mode 100644 tests/installation/playwright-cdn-failover-should-work.spec.ts

diff --git a/packages/playwright-core/src/server/registry/index.ts b/packages/playwright-core/src/server/registry/index.ts
index 3d4912f238..8f07acd56d 100644
--- a/packages/playwright-core/src/server/registry/index.ts
+++ b/packages/playwright-core/src/server/registry/index.ts
@@ -42,6 +42,14 @@ const PLAYWRIGHT_CDN_MIRRORS = [
   'https://playwright-verizon.azureedge.net',
 ];
 
+if (process.env.PW_TEST_CDN_THAT_SHOULD_WORK) {
+  for (let i = 0; i < PLAYWRIGHT_CDN_MIRRORS.length; i++) {
+    const cdn = PLAYWRIGHT_CDN_MIRRORS[i];
+    if (cdn !== process.env.PW_TEST_CDN_THAT_SHOULD_WORK)
+      PLAYWRIGHT_CDN_MIRRORS[i] = cdn + '.does-not-resolve.playwright.dev';
+  }
+}
+
 const EXECUTABLE_PATHS = {
   'chromium': {
     'linux': ['chrome-linux', 'chrome'],
diff --git a/tests/installation/playwright-cdn-failover-should-work.spec.ts b/tests/installation/playwright-cdn-failover-should-work.spec.ts
new file mode 100644
index 0000000000..5e2ae07d6f
--- /dev/null
+++ b/tests/installation/playwright-cdn-failover-should-work.spec.ts
@@ -0,0 +1,50 @@
+/**
+ * Copyright (c) Microsoft Corporation.
+ *
+ * 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.
+ */
+import { test, expect } from './npmTest';
+
+const CDNS = [
+  'https://playwright.azureedge.net',
+  'https://playwright-akamai.azureedge.net',
+  'https://playwright-verizon.azureedge.net',
+];
+
+const DL_STAT_BLOCK = /^.*from url: (.*)$\n^.*to location: (.*)$\n^.*response status code: (.*)$\n^.*total bytes: (.*)$\n^.*SUCCESS downloading (\w+) .*$/gm;
+
+const parsedDownloads = (rawLogs: string) => {
+  const out: { url: string, status: number, name: string }[] = [];
+  for (const match of rawLogs.matchAll(DL_STAT_BLOCK)) {
+    const [, url, /* filepath */, status, /* size */, name] = match;
+    out.push({ url, status: Number.parseInt(status, 10), name: name.toLocaleLowerCase() });
+  }
+  return out;
+};
+
+
+for (const cdn of CDNS) {
+  test(`playwright cdn failover should work (${cdn})`, async ({ exec, nodeMajorVersion, installedSoftwareOnDisk }) => {
+    const result = await exec('npm i --foreground-scripts playwright', { env: { PW_TEST_CDN_THAT_SHOULD_WORK: cdn, DEBUG: 'pw:install' } });
+    expect(result).toHaveLoggedSoftwareDownload(['chromium', 'ffmpeg', 'firefox', 'webkit']);
+    expect(await installedSoftwareOnDisk()).toEqual(['chromium', 'ffmpeg', 'firefox', 'webkit']);
+    const dls = parsedDownloads(result);
+    for (const software of ['chromium', 'ffmpeg', 'firefox', 'webkit'])
+      expect(dls).toContainEqual({ status: 200, name: software, url: expect.stringContaining(cdn) });
+    await exec('node sanity.js playwright');
+    if (nodeMajorVersion >= 14)
+      await exec('node esm-playwright.mjs');
+    const stdio = await exec('npx playwright', 'test', '-c', '.', { expectToExitWithError: true });
+    expect(stdio).toContain(`Please install @playwright/test package to use Playwright Test.`);
+  });
+}