Add benchmarks argument to Driver

JetStreamDriver.js

@@ -216,11 +216,13 @@ const fileLoader = (function() {
 })();
 
 class Driver {
-    constructor() {
+    constructor(benchmarks) {
         this.isReady = false;
         this.isDone = false;
         this.errors = [];
-        this.benchmarks = new Set();
+        // Make benchmark list unique and sort it.
+        this.benchmarks = Array.from(new Set(benchmarks));
+        this.benchmarks.sort((a, b) => a.plan.name.toLowerCase() < b.plan.name.toLowerCase() ? 1 : -1);
         // TODO: Cleanup / remove / merge `blobDataCache` and `loadCache` vs.
         // the global `fileLoader` cache.
         this.blobDataCache = { };
@@ -231,36 +233,6 @@ class Driver {
         this.counter.failedPreloadResources = 0;
     }
 
-    enableBenchmark(benchmark) {
-        // TODO: Remove, make `this.benchmarks` immutable and set it once in the
-        // ctor instead of this and the global `addBenchmarksBy*` functions.
-        this.benchmarks.add(benchmark);
-    }
-
-    enableBenchmarksByName(name) {
-        const benchmark = benchmarksByName.get(name.toLowerCase());
-
-        if (!benchmark)
-            throw new Error(`Couldn't find benchmark named "${name}"`);
-
-        this.enableBenchmark(benchmark);
-    }
-
-    enableBenchmarksByTag(tag, excludeTags) {
-        const benchmarks = benchmarksByTag.get(tag.toLowerCase());
-
-        if (!benchmarks) {
-            const validTags = Array.from(benchmarksByTag.keys()).join(", ");
-            throw new Error(`Couldn't find tag named: ${tag}.\n Choices are ${validTags}`);
-        }
-
-        for (const benchmark of benchmarks) {
-            if (excludeTags && benchmark.hasAnyTag(...excludeTags))
-                continue
-            this.enableBenchmark(benchmark);
-        }
-    }
-
     async start() {
         let statusElement = false;
         let summaryElement = false;
@@ -466,15 +438,9 @@ class Driver {
         });
     }
 
-    initializeBenchmarks() {
-        this.benchmarks = Array.from(this.benchmarks);
-        this.benchmarks.sort((a, b) => a.plan.name.toLowerCase() < b.plan.name.toLowerCase() ? 1 : -1);
-    }
-
     async initialize() {
         if (isInBrowser)
             window.addEventListener("error", (e) => this.pushError("driver startup", e.error));
-        this.initializeBenchmarks();
         await this.prefetchResources();
         this.prepareToRun();
         this.isReady = true;
@@ -2344,12 +2310,11 @@ for (const benchmark of BENCHMARKS) {
     }
 }
 
-this.JetStream = new Driver();
-
 
 function processTestList(testList)
 {
     let benchmarkNames = [];
+    let benchmarks = [];
 
     if (testList instanceof Array)
         benchmarkNames = testList;
@@ -2359,21 +2324,51 @@ function processTestList(testList)
     for (let name of benchmarkNames) {
         name = name.toLowerCase();
         if (benchmarksByTag.has(name))
-            globalThis.JetStream.enableBenchmarksByTag(name);
+            benchmarks.push(...enableBenchmarksByTag(name));
         else
-            globalThis.JetStream.enableBenchmarksByName(name);
+            benchmarks.push(enableBenchmarkByName(name));
     }
+    return benchmarks;
+}
+
+
+function enableBenchmarkByName(name) {
+    const benchmark = benchmarksByName.get(name.toLowerCase());
+
+    if (!benchmark)
+        throw new Error(`Couldn't find benchmark named "${name}"`);
+
+    return benchmark;
 }
 
+
+function enableBenchmarksByTag(tag, excludeTags) {
+    let benchmarks = benchmarksByTag.get(tag.toLowerCase());
+    if (!benchmarks) {
+        const validTags = Array.from(benchmarksByTag.keys()).join(", ");
+        throw new Error(`Couldn't find tag named: ${tag}.\n Choices are ${validTags}`);
+    }
+    if (excludeTags) {
+        benchmarks = benchmarks.filter(benchmark => {
+            return !benchmark.hasAnyTag(...excludeTags);
+        });
+    }
+    return benchmarks;
+}
+
+
+let benchmarks = [];
 const defaultDisabledTags = [];
 // FIXME: add better support to run Worker tests in shells.
 if (!isInBrowser)
     defaultDisabledTags.push("WorkerTests");
 
 if (typeof testList !== "undefined") {
-    processTestList(testList);
+    benchmarks = processTestList(testList);
 } else if (customTestList.length) {
-    processTestList(customTestList);
+    benchmarks = processTestList(customTestList);
 } else {
-    globalThis.JetStream.enableBenchmarksByTag("Default", defaultDisabledTags)
+    benchmarks = enableBenchmarksByTag("Default", defaultDisabledTags)
 }
+
+this.JetStream = new Driver(benchmarks);

tests/unit-tests.js

@@ -38,10 +38,8 @@ function assertEquals(actual, expected, message) {
 
 
 (function testDriverBenchmarksOrder() {
-  const driver = new Driver();
-  driver.enableBenchmarksByTag("all");
-  assertEquals(driver.benchmarks.size, BENCHMARKS.length);
-  driver.initializeBenchmarks();
+  const benchmarks = enableBenchmarksByTag("all");
+  const driver = new Driver(benchmarks);
   assertEquals(driver.benchmarks.length, BENCHMARKS.length);
   const names = driver.benchmarks.map(b => b.name.toLowerCase()).sort().reverse();
   for (let i = 0; i < names.length; i++) {
@@ -51,16 +49,22 @@ function assertEquals(actual, expected, message) {
 
 
 (function testEnableByTag() {
-  const driverA = new Driver();
-  const driverB = new Driver();
-  driverA.enableBenchmarksByTag("Default");
-  driverB.enableBenchmarksByTag("default");
-  assertTrue(driverA.benchmarks.size > 0);
-  assertEquals(driverA.benchmarks.size, driverB.benchmarks.size);
+  const driverA = new Driver(enableBenchmarksByTag("Default"));
+  const driverB = new Driver(enableBenchmarksByTag("default"));
+  assertTrue(driverA.benchmarks.length > 0);
+  assertEquals(driverA.benchmarks.length, driverB.benchmarks.length);
   const enabledBenchmarkNames = new Set(
       Array.from(driverA.benchmarks).map(b => b.name));
   for (const benchmark of BENCHMARKS) {
     if (benchmark.tags.has("default"))
       assertTrue(enabledBenchmarkNames.has(benchmark.name));
   }
 })();
+
+(function testDriverEnableDuplicateAndSort() {
+    const benchmarks = [...enableBenchmarksByTag("wasm"), ...enableBenchmarksByTag("wasm")];
+    const uniqueBenchmarks = new Set(benchmarks);
+    assertFalse(uniqueBenchmarks.size, benchmarks.length);
+    const driver = new Driver(benchmarks);
+    assertEquals(driver.benchmarks.length, uniqueBenchmarks.size);
+})