1 files changed, 167 insertions, 0 deletions
diff --git a/tools/addon-sdk-1.12/test/test-unload.js b/tools/addon-sdk-1.12/test/test-unload.js
new file mode 100644
index 0000000..1bbb64e
--- /dev/null
+++ b/tools/addon-sdk-1.12/test/test-unload.js
@@ -0,0 +1,167 @@
+/* -*- Mode: Java; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim:set ts=2 sw=2 sts=2 et: */
+/* 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/. */
+
+var unload = require("sdk/system/unload");
+var { Loader } = require("sdk/test/loader");
+
+exports.testUnloading = function(test) {
+  var loader = Loader(module, {
+    console: Object.create(console, {
+      exception: { value: function(error) {
+        exceptions.push(error);
+      }}
+    })
+  });
+  var exceptions = [];
+  var ul = loader.require("sdk/system/unload");
+  var unloadCalled = 0;
+  function unload() {
+    unloadCalled++;
+    throw new Error("error");
+  }
+  ul.when(unload);
+
+  // This should be ignored, as we already registered it
+  ul.when(unload);
+
+  function unload2() { unloadCalled++; }
+  ul.when(unload2);
+  loader.unload();
+  test.assertEqual(unloadCalled, 2,
+                   "Unloader functions are called on unload.");
+  test.assertEqual(exceptions.length, 1,
+                   "One unload handler threw exception");
+};
+
+exports.testEnsure = function(test) {
+  test.assertRaises(function() { unload.ensure({}); },
+                    "object has no 'unload' property",
+                    "passing obj with no unload prop should fail");
+  test.assertRaises(function() { unload.ensure({}, "destroy"); },
+                    "object has no 'destroy' property",
+                    "passing obj with no custom unload prop should fail");
+
+  var called = 0;
+  var obj = {unload: function() { called++; }};
+
+  unload.ensure(obj);
+  obj.unload();
+  test.assertEqual(called, 1,
+                   "unload() should be called");
+  obj.unload();
+  test.assertEqual(called, 1,
+                   "unload() should be called only once");
+};
+
+/**
+ * Check that destructors are called only once with Traits.
+ * - check that public API is calling the destructor and unregister it,
+ * - check that composed traits with multiple ensure calls, leads to only
+ * one destructor call.
+ */
+exports.testEnsureWithTraits = function(test) {
+
+  let { Trait } = require("sdk/deprecated/traits");
+  let loader = Loader(module);
+  let ul = loader.require("sdk/system/unload");
+
+  let called = 0;
+  let composedCalled = 0;
+  let composedTrait = Trait.compose({
+      constructor: function () {
+        // We have to give "public interface" of this trait, as we want to
+        // call public `unload` method and ensure that we call it only once,
+        // either when we call this public function manually or on add-on unload
+        ul.ensure(this._public);
+      },
+      unload: function unload() {
+        composedCalled++;
+      }
+    });
+  let obj = Trait.compose(
+    composedTrait.resolve({
+      constructor: "_constructor",
+      unload : "_unload" 
+    }), {
+      constructor: function constructor() {
+        // Same thing applies here, we need to pass public interface
+        ul.ensure(this._public);
+        this._constructor();
+      },
+      unload: function unload() {
+        called++;
+        this._unload();
+      }
+    })();
+
+  obj.unload();
+  test.assertEqual(called, 1,
+                   "unload() should be called");
+
+  test.assertEqual(composedCalled, 1,
+                   "composed object unload() should be called");
+
+  obj.unload();
+  test.assertEqual(called, 1,
+                   "unload() should be called only once");
+  test.assertEqual(composedCalled, 1,
+                   "composed object unload() should be called only once");
+
+  loader.unload();
+  test.assertEqual(called, 1,
+                   "unload() should be called only once, after addon unload");
+  test.assertEqual(composedCalled, 1,
+                   "composed object unload() should be called only once, " +
+                   "after addon unload");
+};
+
+exports.testEnsureWithTraitsPrivate = function(test) {
+
+  let { Trait } = require("sdk/deprecated/traits");
+  let loader = Loader(module);
+  let ul = loader.require("sdk/system/unload");
+
+  let called = 0;
+  let privateObj = null;
+  let obj = Trait.compose({
+      constructor: function constructor() {
+        // This time wa don't have to give public interface,
+        // as we want to call a private method:
+        ul.ensure(this, "_unload");
+        privateObj = this;
+      },
+      _unload: function unload() {
+        called++;
+        this._unload();
+      }
+    })();
+
+  loader.unload();
+  test.assertEqual(called, 1,
+                   "unload() should be called");
+
+  privateObj._unload();
+  test.assertEqual(called, 1,
+                   "_unload() should be called only once, after addon unload");
+};
+
+exports.testReason = function (test) {
+  var reason = "Reason doesn't actually have to be anything in particular.";
+  var loader = Loader(module);
+  var ul = loader.require("sdk/system/unload");
+  ul.when(function (rsn) {
+    test.assertEqual(rsn, reason,
+                     "when() reason should be reason given to loader");
+  });
+  var obj = {
+    unload: function (rsn) {
+      test.assertEqual(rsn, reason,
+                       "ensure() reason should be reason given to loader");
+    }
+  };
+  ul.ensure(obj);
+  loader.unload(reason);
+};