1 files changed, 46 insertions, 0 deletions
diff --git a/testing/talos/talos/tests/devtools/addon/api.js b/testing/talos/talos/tests/devtools/addon/api.js
new file mode 100644
index 0000000000..9d9719bb68
--- /dev/null
+++ b/testing/talos/talos/tests/devtools/addon/api.js
@@ -0,0 +1,46 @@
+"use strict";
+
+/* globals ExtensionAPI, Services, XPCOMUtils */
+
+this.damp = class extends ExtensionAPI {
+  getAPI(context) {
+    return {
+      damp: {
+        startTest() {
+          // Some notes about using a DevTools loader for DAMP.
+          //
+          // The DAMP loader needs to be same loader as the one used by the
+          // toolbox later on. Otherwise, we will not retrieve the proper
+          // instance of some modules.
+          // The main devtools loader is the exported `loader` from Loader.sys.mjs,
+          // we have to use that.
+
+          dump("[damp-api] Retrieve the main DevTools loader\n");
+          const { loader, require } = ChromeUtils.importESModule(
+            "resource://devtools/shared/loader/Loader.sys.mjs"
+          );
+
+          const { rootURI } = context.extension;
+          const dampRootDir = rootURI.QueryInterface(Ci.nsIFileURL);
+
+          const protocolHandler = Services.io
+            .getProtocolHandler("resource")
+            .QueryInterface(Ci.nsIResProtocolHandler);
+
+          // Serve testing/talos/talos/tests/devtools/addon/ via "resource://damp-test"
+          // Loader.sys.mjs will map `require("damp-test/...")` to `resource://damp-test/content/...`
+          // Thus allowing to load damp files from the content folder via the DevTools loader.
+          protocolHandler.setSubstitution("damp-test", dampRootDir);
+
+          // Expose the window to modules loaded for DAMP.
+          loader.loader.globals.dampWindow = context.appWindow;
+          loader.loader.globals.fetch = context.appWindow.fetch;
+
+          dump("[damp-api] Retrieve the DAMP runner and start the test\n");
+          const { damp } = require("damp-test/damp");
+          return damp.startTest();
+        },
+      },
+    };
+  }
+};