1 files changed, 94 insertions, 0 deletions
diff --git a/devtools/server/tests/xpcshell/test_register_actor.js b/devtools/server/tests/xpcshell/test_register_actor.js
new file mode 100644
index 0000000000..f38ab73572
--- /dev/null
+++ b/devtools/server/tests/xpcshell/test_register_actor.js
@@ -0,0 +1,94 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+function run_test() {
+  // Allow incoming connections.
+  DevToolsServer.keepAlive = true;
+  DevToolsServer.init();
+  DevToolsServer.registerAllActors();
+
+  add_test(test_lazy_api);
+  add_test(manual_remove);
+  add_test(cleanup);
+  run_next_test();
+}
+
+// Bug 988237: Test the new lazy actor actor-register
+function test_lazy_api() {
+  let isActorLoaded = false;
+  let isActorInstantiated = false;
+  function onActorEvent(subject, topic, data) {
+    if (data == "loaded") {
+      isActorLoaded = true;
+    } else if (data == "instantiated") {
+      isActorInstantiated = true;
+    }
+  }
+  Services.obs.addObserver(onActorEvent, "actor");
+  ActorRegistry.registerModule("xpcshell-test/registertestactors-lazy", {
+    prefix: "lazy",
+    constructor: "LazyActor",
+    type: { global: true, target: true },
+  });
+  // The actor is immediatly registered, but not loaded
+  Assert.ok(
+    ActorRegistry.targetScopedActorFactories.hasOwnProperty("lazyActor")
+  );
+  Assert.ok(ActorRegistry.globalActorFactories.hasOwnProperty("lazyActor"));
+  Assert.ok(!isActorLoaded);
+  Assert.ok(!isActorInstantiated);
+
+  const client = new DevToolsClient(DevToolsServer.connectPipe());
+  client.connect().then(function onConnect() {
+    client.mainRoot.rootForm.then(onRootForm);
+  });
+  function onRootForm(response) {
+    // On rootForm, the actor is still not loaded,
+    // but we can see its name in the list of available actors
+    Assert.ok(!isActorLoaded);
+    Assert.ok(!isActorInstantiated);
+    Assert.ok("lazyActor" in response);
+
+    const { LazyFront } = require("xpcshell-test/registertestactors-lazy");
+    const front = new LazyFront(client);
+    // As this Front isn't instantiated by protocol.js, we have to manually
+    // set its actor ID and manage it:
+    front.actorID = response.lazyActor;
+    client.addActorPool(front);
+    front.manage(front);
+
+    front.hello().then(onRequest);
+  }
+  function onRequest(response) {
+    Assert.equal(response, "world");
+
+    // Finally, the actor is loaded on the first request being made to it
+    Assert.ok(isActorLoaded);
+    Assert.ok(isActorInstantiated);
+
+    Services.obs.removeObserver(onActorEvent, "actor");
+    client.close().then(() => run_next_test());
+  }
+}
+
+function manual_remove() {
+  Assert.ok(ActorRegistry.globalActorFactories.hasOwnProperty("lazyActor"));
+  ActorRegistry.removeGlobalActor("lazyActor");
+  Assert.ok(!ActorRegistry.globalActorFactories.hasOwnProperty("lazyActor"));
+
+  run_next_test();
+}
+
+function cleanup() {
+  DevToolsServer.destroy();
+
+  // Check that all actors are unregistered on server destruction
+  Assert.ok(
+    !ActorRegistry.targetScopedActorFactories.hasOwnProperty("lazyActor")
+  );
+  Assert.ok(!ActorRegistry.globalActorFactories.hasOwnProperty("lazyActor"));
+
+  run_next_test();
+}