1 files changed, 90 insertions, 0 deletions
diff --git a/testing/raptor/browsertime/browsertime_benchmark.js b/testing/raptor/browsertime/browsertime_benchmark.js
new file mode 100644
index 0000000000..35bc2c1844
--- /dev/null
+++ b/testing/raptor/browsertime/browsertime_benchmark.js
@@ -0,0 +1,90 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+/* eslint-env node */
+
+module.exports = async function (context, commands) {
+  context.log.info("Starting a browsertime benchamrk");
+  let url = context.options.browsertime.url;
+  let page_cycles = context.options.browsertime.page_cycles;
+  let page_cycle_delay = context.options.browsertime.page_cycle_delay;
+  let post_startup_delay = context.options.browsertime.post_startup_delay;
+  let page_timeout = context.options.timeouts.pageLoad;
+  let ret = false;
+  let expose_profiler = context.options.browsertime.expose_profiler;
+
+  context.log.info(
+    "Waiting for %d ms (post_startup_delay)",
+    post_startup_delay
+  );
+  await commands.wait.byTime(post_startup_delay);
+
+  for (let count = 0; count < page_cycles; count++) {
+    context.log.info("Navigating to about:blank");
+    await commands.navigate("about:blank");
+
+    context.log.info("Cycle %d, waiting for %d ms", count, page_cycle_delay);
+    await commands.wait.byTime(page_cycle_delay);
+
+    context.log.info("Cycle %d, starting the measure", count);
+    if (expose_profiler === "true") {
+      context.log.info("Custom profiler start!");
+      await commands.profiler.start();
+    }
+    await commands.measure.start(url);
+
+    context.log.info("Benchmark custom metric collection");
+
+    let data = null;
+    let starttime = await commands.js.run(`return performance.now();`);
+    while (
+      data == null &&
+      (await commands.js.run(`return performance.now();`)) - starttime <
+        page_timeout
+    ) {
+      let wait_time = 3000;
+      context.log.info("Waiting %d ms for data from benchmark...", wait_time);
+      await commands.wait.byTime(wait_time);
+      data = await commands.js.run(
+        "return window.sessionStorage.getItem('benchmark_results');"
+      );
+    }
+    if (expose_profiler === "true") {
+      context.log.info("Custom profiler stop!");
+      await commands.profiler.stop();
+    }
+    if (
+      data == null &&
+      (await commands.js.run(`return performance.now();`)) - starttime >=
+        page_timeout
+    ) {
+      ret = false;
+      context.log.error("Benchmark timed out. Aborting...");
+    } else if (data) {
+      // Reset benchmark results
+      await commands.js.run(
+        "return window.sessionStorage.removeItem('benchmark_results');"
+      );
+
+      context.log.info("Value of benchmark data: ", data);
+      data = JSON.parse(data);
+
+      if (!Array.isArray(data)) {
+        commands.measure.addObject({ browsertime_benchmark: data });
+      } else {
+        commands.measure.addObject({
+          browsertime_benchmark: {
+            [data[1]]: data.slice(2),
+          },
+        });
+      }
+      ret = true;
+    } else {
+      context.log.error("No data collected from benchmark.");
+      ret = false;
+    }
+  }
+  context.log.info("Browsertime benchmark ended.");
+  return ret;
+};