The Tor Browser: comparison services/common/tests/unit/head_helpers.js

comparison: services/common/tests/unit/head_helpers.js

services/common/tests/unit/head_helpers.js

branch: TOR_BUG_3246
changeset 5: 4ab42b5ab56c

equal deleted inserted replaced

--1:000000000000
+:baab70428848
+/* 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/. */
+Cu.import("resource://gre/modules/Log.jsm");
+Cu.import("resource://services-common/utils.js");
+Cu.import("resource://testing-common/httpd.js");
+Cu.import("resource://testing-common/services-common/logging.js");
+let btoa = Cu.import("resource://gre/modules/Log.jsm").btoa;
+let atob = Cu.import("resource://gre/modules/Log.jsm").atob;
+function do_check_empty(obj) {
+do_check_attribute_count(obj, 0);
+}
+function do_check_attribute_count(obj, c) {
+do_check_eq(c, Object.keys(obj).length);
+}
+function do_check_throws(aFunc, aResult, aStack) {
+if (!aStack) {
+try {
+// We might not have a 'Components' object.
+aStack = Components.stack.caller;
+} catch (e) {}
+}
+try {
+aFunc();
+} catch (e) {
+do_check_eq(e.result, aResult, aStack);
+return;
+}
+do_throw("Expected result " + aResult + ", none thrown.", aStack);
+}
+/**
+* Test whether specified function throws exception with expected
+* result.
+*
+* @param func
+*        Function to be tested.
+* @param message
+*        Message of expected exception. <code>null</code> for no throws.
+*/
+function do_check_throws_message(aFunc, aResult) {
+try {
+aFunc();
+} catch (e) {
+do_check_eq(e.message, aResult);
+return;
+}
+do_throw("Expected an error, none thrown.");
+}
+/**
+* Print some debug message to the console. All arguments will be printed,
+* separated by spaces.
+*
+* @param [arg0, arg1, arg2, ...]
+*        Any number of arguments to print out
+* @usage _("Hello World") -> prints "Hello World"
+* @usage _(1, 2, 3) -> prints "1 2 3"
+*/
+let _ = function(some, debug, text, to) print(Array.slice(arguments).join(" "));
+function httpd_setup (handlers, port=-1) {
+let server = new HttpServer();
+for (let path in handlers) {
+server.registerPathHandler(path, handlers[path]);
+}
+try {
+server.start(port);
+} catch (ex) {
+_("==========================================");
+_("Got exception starting HTTP server on port " + port);
+_("Error: " + CommonUtils.exceptionStr(ex));
+_("Is there a process already listening on port " + port + "?");
+_("==========================================");
+do_throw(ex);
+}
+// Set the base URI for convenience.
+let i = server.identity;
+server.baseURI = i.primaryScheme + "://" + i.primaryHost + ":" + i.primaryPort;
+return server;
+}
+function httpd_handler(statusCode, status, body) {
+return function handler(request, response) {
+_("Processing request");
+// Allow test functions to inspect the request.
+request.body = readBytesFromInputStream(request.bodyInputStream);
+handler.request = request;
+response.setStatusLine(request.httpVersion, statusCode, status);
+if (body) {
+response.bodyOutputStream.write(body, body.length);
+}
+};
+}
+/*
+* Read bytes string from an nsIInputStream.  If 'count' is omitted,
+* all available input is read.
+*/
+function readBytesFromInputStream(inputStream, count) {
+return CommonUtils.readBytesFromInputStream(inputStream, count);
+}
+/*
+* Ensure exceptions from inside callbacks leads to test failures.
+*/
+function ensureThrows(func) {
+return function() {
+try {
+func.apply(this, arguments);
+} catch (ex) {
+do_throw(ex);
+}
+};
+}
+/**
+* Proxy auth helpers.
+*/
+/**
+* Fake a PAC to prompt a channel replacement.
+*/
+let PACSystemSettings = {
+CID: Components.ID("{5645d2c1-d6d8-4091-b117-fe7ee4027db7}"),
+contractID: "@mozilla.org/system-proxy-settings;1",
+QueryInterface: XPCOMUtils.generateQI([Ci.nsIFactory,
+Ci.nsISystemProxySettings]),
+createInstance: function createInstance(outer, iid) {
+if (outer) {
+throw Cr.NS_ERROR_NO_AGGREGATION;
+}
+return this.QueryInterface(iid);
+},
+lockFactory: function lockFactory(lock) {
+throw Cr.NS_ERROR_NOT_IMPLEMENTED;
+},
+// Replace this URI for each test to avoid caching. We want to ensure that
+// each test gets a completely fresh setup.
+mainThreadOnly: true,
+PACURI: null,
+getProxyForURI: function getProxyForURI(aURI) {
+throw Cr.NS_ERROR_NOT_IMPLEMENTED;
+}
+};
+function installFakePAC() {
+_("Installing fake PAC.");
+Cm.nsIComponentRegistrar
+.registerFactory(PACSystemSettings.CID,
+"Fake system proxy-settings",
+PACSystemSettings.contractID,
+PACSystemSettings);
+}
+function uninstallFakePAC() {
+_("Uninstalling fake PAC.");
+let CID = PACSystemSettings.CID;
+Cm.nsIComponentRegistrar.unregisterFactory(CID, PACSystemSettings);
+}
+// Many tests do service.startOver() and don't expect the provider type to
+// change (whereas by default, a startOver will do exactly that so FxA is
+// subsequently used). The tests that know how to deal with
+// the Firefox Accounts identity hack things to ensure that still works.
+function ensureStartOverKeepsIdentity() {
+Cu.import("resource://gre/modules/Services.jsm");
+Services.prefs.setBoolPref("services.sync-testing.startOverKeepIdentity", true);
+do_register_cleanup(function() {
+Services.prefs.clearUserPref("services.sync-testing.startOverKeepIdentity");
+});
+}
+ensureStartOverKeepsIdentity();