Merge pull request #32 from coltdorsey/kirk/cypress_change_to_proxy_a…

…uth_header

Use Proxy-Authentication header

package.json

@@ -1,7 +1,7 @@
 {
   "name": "cypress-iap",
-  "version": "0.0.0-development",
-  "description": "Cypress command to authenticate with IAP",
+  "version": "2.0.0",
+  "description": "Cypress command to authenticate to Google Identity Aware Proxy(IAP)",
   "main": "index.js",
   "types": "index.d.ts",
   "author": "Colt Dorsey <coltdorsey@gmail.com>",

utils.js

@@ -1,12 +1,12 @@
 module.exports = (on, config) => {
-    on('task', {
-        async getIAPToken({ url, cid }) {
-            const { GoogleAuth } = require('google-auth-library');
-            const auth = new GoogleAuth();
-            const client = await auth.getIdTokenClient(cid);
-            const res = await client.request({ url });
-            return res.config.headers.Authorization.split(" ")[1]
-        }
-    })
-    return config
-}
+  on("task", {
+    async getIAPToken({ url, cid }) {
+      const { GoogleAuth } = require("google-auth-library");
+      const auth = new GoogleAuth();
+      const client = await auth.getIdTokenClient(cid);
+      const res = await client.request({ url });
+      return res.config.headers.Authorization;
+    },
+  });
+  return config;
+};

visit.js

@@ -1,19 +1,29 @@
 /// <reference types="cypress">
-Cypress.Commands.overwrite('visit', (originalFn, subject, ...args) => {
-    if (Cypress.config().baseUrl.indexOf('localhost') != -1) {
-        return originalFn(subject, ...args);
-    } else {
-        cy.exec('curl -I ' + Cypress.config().baseUrl + " | awk '/^location/ {split($NF, a, /[=&]/); print a[2]}'").then((client_id) => {
-            return client_id.stdout
-        }).then(client_id => {
-            cy.task('getIAPToken', { url: Cypress.config().baseUrl, cid: client_id }).then((token) => {
-                console.log(token)
-                cy.setCookie('GCP_IAAP_AUTH_TOKEN', token)
-                    .then( () => {
-                        return originalFn(subject, ...args)
-                    })
-            })
-        })
-
-    }
-})
+Cypress.Commands.overwrite("visit", (originalFn, subject, ...args) => {
+  if (Cypress.config().baseUrl.indexOf("localhost") != -1) {
+    return originalFn(subject, ...args);
+  } else {
+    cy.exec(
+      `curl -I ${
+        Cypress.config().baseUrl
+      } | awk '/^location/ {split($NF, a, /[=&]/); print a[2]}'`
+    )
+      .then((client_id) => {
+        return client_id.stdout;
+      })
+      .then((client_id) => {
+        cy.task("getIAPToken", {
+          url: Cypress.config().baseUrl,
+          cid: client_id,
+        }).then((iapToken) => {
+          cy.intercept(`${Cypress.config().baseUrl}/**`, (req) => {
+            req.headers["Proxy-Authorization"] = `${iapToken}`;
+          })
+            .as("Cypress-IAP")
+            .then(() => {
+              return originalFn(subject, ...args);
+            });
+        });
+      });
+  }
+});