From ec2cabe6136ae5115736d3aa89db608a6554204f Mon Sep 17 00:00:00 2001
From: Kendall Werts <kmwerts@gmail.com>
Date: Wed, 26 Feb 2020 10:56:41 -0600
Subject: [PATCH] working on the never-ask test

---
 test/common.js                   |  7 +++++--
 test/features/assignment.test.js | 10 +++++++---
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/test/common.js b/test/common.js
index f9f0f57..fc1d38c 100644
--- a/test/common.js
+++ b/test/common.js
@@ -72,9 +72,12 @@ const buildPopupDom = popup => {
 
 const buildConfirmPage = async (url) => {
   console.log(url)
-  const webExtension = await webExtensionsJSDOM.fromFile(url);
+  const webExtension = await webExtensionsJSDOM
+    .fromFile(path.join(__dirname, "../src/confirm-page.html"), {
+      apiFake: true,
+      jsdom: {url}
+    });
 
-  webExtension.browser.runtime.onMessage = window.browser.runtime.sendMessage;
   return webExtension;
 };
 
diff --git a/test/features/assignment.test.js b/test/features/assignment.test.js
index 526342f..e1aa680 100644
--- a/test/features/assignment.test.js
+++ b/test/features/assignment.test.js
@@ -89,9 +89,13 @@ describe("Assignment Comfirm Page Feature", function () {
     describe("Set assignment to 'never ask' ", function () {
       beforeEach(async function () {
         // click confirm page to always open in container
-        const confirmPage = await buildConfirmPage("src/confirm-page.html?" +
-               `url=${encodeURIComponent(url)}` +
-               `&cookieStoreId=${this.webExt.tab.cookieStoreId}`);
+        const confirmPage = await buildConfirmPage("moz-extension://fake/confirm-page.html?" + 
+          `url=${encodeURIComponent(url)}` +
+          `&cookieStoreId=${this.webExt.tab.cookieStoreId}`);
+
+        confirmPage.browser.runtime.sendMessage.callsFake((...args) => {
+          this.webExt.browser.runtime.onMessage.addListener.yield(...args);
+        });
         await confirmPage.document.getElementById("never-ask").click();
         await confirmPage.document.getElementById("confirm").click();
       });