Bug 1130017 unload the temp loader in the unsafe-content-script test …

…add-on
mozilla · Feb 5, 2015 · 56d0736 · 56d0736
1 parent ad36ca4
commit 56d0736
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/test/addons/unsafe-content-script/main.js b/test/addons/unsafe-content-script/main.js
@@ -6,10 +6,10 @@
 const { create: makeFrame } = require("sdk/frame/utils");
 const { window } = require("sdk/addon/window");
 const { Loader } = require('sdk/test/loader');
-const loader = Loader(module);
-const Worker = loader.require("sdk/content/worker").Worker;
 
 exports.testMembranelessMode = function(assert, done) {
+  const loader = Loader(module);
+  const Worker = loader.require("sdk/content/worker").Worker;
 
   let url = "data:text/html;charset=utf-8," + encodeURIComponent(
     '<script>' +
@@ -48,13 +48,20 @@ exports.testMembranelessMode = function(assert, done) {
           done();
         }
     });
-    worker.port.on("done", function () {
+
+    worker.port.on("done", () => {
+      // cleanup
       element.parentNode.removeChild(element);
+      worker.destroy();
+      loader.unload();
+
       done();
     });
+
     worker.port.on("assert", function (data) {
       assert.ok(data.assertion, data.msg);
     });
+
   }
 };