From 992c8bcf75b9477e1652cf2a458418e375903cc0 Mon Sep 17 00:00:00 2001
From: toasted-nutbread <toasted-nutbread@users.noreply.github.com>
Date: Sat, 31 Jul 2021 14:36:14 -0400
Subject: [PATCH] test-workers (#1863)

* Add getImportedScripts utility function to test-sw.js

* Move functionality to testServiceWorker

* Rename file

* Add loadEslint utility function

* Add filterScriptPaths utility
---
 test/{test-sw.js => test-workers.js} | 75 +++++++++++++++++-----------
 1 file changed, 45 insertions(+), 30 deletions(-)
 rename test/{test-sw.js => test-workers.js} (52%)

diff --git a/test/test-sw.js b/test/test-workers.js
similarity index 52%
rename from test/test-sw.js
rename to test/test-workers.js
index 498a0a32..9f9c4d13 100644
--- a/test/test-sw.js
+++ b/test/test-workers.js
@@ -22,6 +22,15 @@ const {VM} = require('../dev/vm');
 const assert = require('assert');
 
 
+function loadEslint() {
+    return JSON.parse(fs.readFileSync(path.join(__dirname, '..', '.eslintrc.json'), {encoding: 'utf8'}));
+}
+
+function filterScriptPaths(scriptPaths) {
+    const extDirName = 'ext';
+    return scriptPaths.filter((src) => !src.startsWith('/lib/')).map((src) => `${extDirName}${src}`);
+}
+
 function getAllHtmlScriptPaths(fileName) {
     const domSource = fs.readFileSync(fileName, {encoding: 'utf8'});
     const dom = new JSDOM(domSource);
@@ -43,38 +52,44 @@ function convertBackgroundScriptsToServiceWorkerScripts(scripts) {
     scripts[index] = '/js/dom/simple-dom-parser.js';
 }
 
+function getImportedScripts(scriptPath, fields) {
+    const importedScripts = [];
+
+    const importScripts = (...scripts) => {
+        importedScripts.push(...scripts);
+    };
+
+    const vm = new VM(Object.assign({importScripts}, fields));
+    vm.context.self = vm.context;
+    vm.execute([scriptPath]);
+
+    return importedScripts;
+}
+
+function testServiceWorker() {
+    // Verify that sw.js scripts match background.html scripts
+    const extDir = path.join(__dirname, '..', 'ext');
+    const scripts = getAllHtmlScriptPaths(path.join(extDir, 'background.html'));
+    convertBackgroundScriptsToServiceWorkerScripts(scripts);
+    const importedScripts = getImportedScripts('sw.js', {});
+    assert.deepStrictEqual(scripts, importedScripts);
+
+    // Verify that eslint config lists files correctly
+    const expectedSwRulesFiles = filterScriptPaths(scripts);
+    const eslintConfig = loadEslint();
+    const swRules = eslintConfig.overrides.find((item) => (
+        typeof item.env === 'object' &&
+        item.env !== null &&
+        item.env.serviceworker === true
+    ));
+    assert.ok(typeof swRules !== 'undefined');
+    assert.ok(Array.isArray(swRules.files));
+    assert.deepStrictEqual(swRules.files, expectedSwRulesFiles);
+}
+
 function main() {
     try {
-        // Verify that sw.js scripts match background.html scripts
-        const rootDir = path.join(__dirname, '..');
-        const extDirName = 'ext';
-        const extDir = path.join(rootDir, extDirName);
-
-        const scripts = getAllHtmlScriptPaths(path.join(extDir, 'background.html'));
-        convertBackgroundScriptsToServiceWorkerScripts(scripts);
-        const importedScripts = [];
-
-        const importScripts = (...scripts2) => {
-            importedScripts.push(...scripts2);
-        };
-
-        const vm = new VM({importScripts});
-        vm.context.self = vm.context;
-        vm.execute(['sw.js']);
-
-        vm.assert.deepStrictEqual(scripts, importedScripts);
-
-        // Verify that eslint config lists files correctly
-        const expectedSwRulesFiles = scripts.filter((src) => !src.startsWith('/lib/')).map((src) => `${extDirName}${src}`);
-        const eslintConfig = JSON.parse(fs.readFileSync(path.join(rootDir, '.eslintrc.json'), {encoding: 'utf8'}));
-        const swRules = eslintConfig.overrides.find((item) => (
-            typeof item.env === 'object' &&
-            item.env !== null &&
-            item.env.serviceworker === true
-        ));
-        assert.ok(typeof swRules !== 'undefined');
-        assert.ok(Array.isArray(swRules.files));
-        assert.deepStrictEqual(swRules.files, expectedSwRulesFiles);
+        testServiceWorker();
     } catch (e) {
         console.error(e);
         process.exit(-1);