1 files changed, 107 insertions, 0 deletions
diff --git a/tools/addon-sdk-1.4/packages/api-utils/tests/test-xpcom.js b/tools/addon-sdk-1.4/packages/api-utils/tests/test-xpcom.js
new file mode 100644
index 0000000..fdcb6e8
--- /dev/null
+++ b/tools/addon-sdk-1.4/packages/api-utils/tests/test-xpcom.js
@@ -0,0 +1,107 @@
+var traceback = require("traceback");
+var xpcom = require("xpcom");
+var {Cc,Ci,Cm,Cr} = require("chrome");
+var { Loader } = require("./helpers");
+
+exports.testRegister = function(test, text) {
+  if (!text)
+    text = "hai2u";
+
+  function Component() {}
+
+  Component.prototype = {
+    newChannel : function(aURI) {
+      var ios = Cc["@mozilla.org/network/io-service;1"].
+                getService(Ci.nsIIOService);
+
+      var channel = ios.newChannel(
+        "data:text/plain," + text,
+        null,
+        null
+      );
+
+      channel.originalURI = aURI;
+      return channel;
+    },
+    getURIFlags: function(aURI) {
+        return Ci.nsIAboutModule.ALLOW_SCRIPT;
+    },
+    QueryInterface: xpcom.utils.generateQI([Ci.nsIAboutModule])
+  };
+
+  var contractID = "@mozilla.org/network/protocol/about;1?what=boop";
+
+  var factory = xpcom.register({name: "test about:boop page",
+                                contractID: contractID,
+                                create: Component});
+
+  var manager = Cm.QueryInterface(Ci.nsIComponentRegistrar);
+  test.assertEqual(manager.isCIDRegistered(factory.uuid), true);
+
+  // We don't want to use Cc[contractID] here because it's immutable,
+  // so it can't accept updated versions of a contractID during the
+  // same application session.
+  var aboutFactory = xpcom.getClass(contractID, Ci.nsIFactory);
+
+  test.assertNotEqual(aboutFactory.wrappedJSObject,
+                      undefined,
+                      "Factory wrappedJSObject should exist.");
+
+  var about = aboutFactory.createInstance(null, Ci.nsIAboutModule);
+  var ios = Cc["@mozilla.org/network/io-service;1"].
+            getService(Ci.nsIIOService);
+  test.assertEqual(
+    about.getURIFlags(ios.newURI("http://foo.com", null, null)),
+    Ci.nsIAboutModule.ALLOW_SCRIPT
+  );
+
+  var aboutURI = ios.newURI("about:boop", null, null);
+  var channel = ios.newChannelFromURI(aboutURI);
+  var iStream = channel.open();
+  var siStream = Cc['@mozilla.org/scriptableinputstream;1']
+                 .createInstance(Ci.nsIScriptableInputStream);
+  siStream.init(iStream);
+  var data = new String();
+  data += siStream.read(-1);
+  siStream.close();
+  iStream.close();
+  test.assertEqual(data, text);
+
+  factory.unregister();
+  test.assertEqual(manager.isCIDRegistered(factory.uuid), false);
+};
+
+exports.testReRegister = function(test) {
+  exports.testRegister(test, "hai2u again");
+};
+
+exports.testMakeUuid = function(test) {
+  var first = xpcom.makeUuid().toString();
+  var second = xpcom.makeUuid().toString();
+  test.assertMatches(first, /{[0-9a-f\-]+}/);
+  test.assertMatches(second, /{[0-9a-f\-]+}/);
+  test.assertNotEqual(first, second);
+};
+
+exports.testUnload = function(test) {
+  var loader = Loader(module);
+  var sbxpcom = loader.require("xpcom");
+
+  function Component() {}
+
+  Component.prototype = {
+    QueryInterface: sbxpcom.utils.generateQI([Ci.nsISupports])
+  };
+
+  var contractID = "@mozilla.org/blargle;1";
+  var factory = sbxpcom.register({name: "test component",
+                                  contractID: contractID,
+                                  create: Component});
+
+  var manager = Cm.QueryInterface(Ci.nsIComponentRegistrar);
+  test.assertEqual(manager.isCIDRegistered(factory.uuid), true);
+
+  loader.unload();
+
+  test.assertEqual(manager.isCIDRegistered(factory.uuid), false);
+};