gecko-dev/testing/mochitest/start_b2g.js

/* 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 outOfProcess = __webDriverArguments[0]
var mochitestUrl = __webDriverArguments[1]
var onDevice = __webDriverArguments[2]
var wifiSettings = __webDriverArguments[3]
var chrome = __webDriverArguments[4]
var prefs = Components.classes["@mozilla.org/preferences-service;1"].
                            getService(Components.interfaces.nsIPrefBranch)
var settings = window.navigator.mozSettings;
var cm = Components.classes["@mozilla.org/categorymanager;1"].
                    getService(Components.interfaces.nsICategoryManager);

if (wifiSettings)
  wifiSettings = JSON.parse(wifiSettings);

const CHILD_SCRIPT = "chrome://specialpowers/content/specialpowers.js";
const CHILD_SCRIPT_API = "chrome://specialpowers/content/specialpowersAPI.js";
const CHILD_LOGGER_SCRIPT = "chrome://specialpowers/content/MozillaLogger.js";

var homescreen = document.getElementById('systemapp');
var container = homescreen.contentWindow.document.getElementById('test-container');

// Disable udpate timers which cause failure in b2g permisson prompt tests.
if (cm) {
  cm.deleteCategoryEntry("update-timer", "WebappsUpdateTimer", false);
  cm.deleteCategoryEntry("update-timer", "nsUpdateService", false);
}

var SECURITY_PREF = "security.turn_off_all_security_so_that_viruses_can_take_over_this_computer";
Components.utils.import("resource://gre/modules/Services.jsm");
Services.prefs.setBoolPref(SECURITY_PREF, true);
// RIL DOM events and mozChromeEvents will be pending until
// this observer message is sent.
Services.obs.notifyObservers(null, 'system-message-listener-ready', null);

function openWindow(aEvent) {
  var popupIframe = aEvent.detail.frameElement;
  popupIframe.id = 'popupiframe';

  // This is to size the iframe to what is requested in the window.open call,
  // e.g. window.open("", "", "width=600,height=600");
  if (aEvent.detail.features.indexOf('width') != -1) {
    let width = aEvent.detail.features.substr(aEvent.detail.features.indexOf('width')+6);
    width = width.substr(0,width.indexOf(',') == -1 ? width.length : width.indexOf(','));
    popupIframe.setAttribute('width', width);
  }
  if (aEvent.detail.features.indexOf('height') != -1) {
    let height = aEvent.detail.features.substr(aEvent.detail.features.indexOf('height')+7);
    height = height.substr(0, height.indexOf(',') == -1 ? height.length : height.indexOf(','));
    popupIframe.setAttribute('height', height);
  }

  popupIframe.addEventListener('mozbrowserclose', function(e) {
    container.parentNode.removeChild(popupIframe);
    container.focus();
  });

  // yes, the popup can call window.open too!
  popupIframe.addEventListener('mozbrowseropenwindow', openWindow);

  popupIframe.addEventListener('mozbrowserloadstart', function(e) {
    popupIframe.focus();
    let mm = popupIframe.QueryInterface(Ci.nsIFrameLoaderOwner).frameLoader.messageManager;
    mm.loadFrameScript(CHILD_LOGGER_SCRIPT, true);
    mm.loadFrameScript(CHILD_SCRIPT_API, true);
    mm.loadFrameScript(CHILD_SCRIPT, true);
    mm.loadFrameScript('data:,attachSpecialPowersToWindow(content);', true);
  });

  container.parentNode.appendChild(popupIframe);
}
container.addEventListener('mozbrowseropenwindow', openWindow);
container.addEventListener('mozbrowsershowmodalprompt', function (e) {
  if (e.detail.message == 'setVisible::false') {
    container.setVisible(false);
  }
  else if (e.detail.message == 'setVisible::true') {
    container.setVisible(true);
  }
});

function StartTest() {
  if (outOfProcess) {
    let mm = container.QueryInterface(Ci.nsIFrameLoaderOwner).frameLoader.messageManager;

    //Workaround for bug 848411, once that bug is fixed, the following line can be removed
    function contentScript() {
      addEventListener("DOMWindowCreated", function listener(e) {
        removeEventListener("DOMWindowCreated", listener, false);
        var window = e.target.defaultView;
        window.wrappedJSObject.SpecialPowers.addPermission("allowXULXBL", true, window.document);
      });
    }
    mm.loadFrameScript("data:,(" + encodeURI(contentScript.toSource()) + ")();", true);
  }

  if (chrome) {
    let loader = Components.classes["@mozilla.org/moz/jssubscript-loader;1"].getService(Ci.mozIJSSubScriptLoader);
    if (typeof(SpecialPowers) == 'undefined') {
      loader.loadSubScript("chrome://specialpowers/content/specialpowersAPI.js");
      loader.loadSubScript("chrome://specialpowers/content/SpecialPowersObserverAPI.js");
      loader.loadSubScript("chrome://specialpowers/content/ChromePowers.js");
    }
    loader.loadSubScript("chrome://mochikit/content/browser-test.js");
    b2gStart();
  }

  if (onDevice) {
    var cpuLock = Components.classes["@mozilla.org/power/powermanagerservice;1"]
      .getService(Ci.nsIPowerManagerService)
      .newWakeLock("cpu");

    let manager = navigator.mozWifiManager;
    let con = manager.connection;
    manager.setPowerSavingMode(false);

    manager.onenabled = function () {
      if(wifiSettings) {
        var req = manager.getKnownNetworks();
        req.onsuccess = function () {
          var networks = req.result;
          for (var i = 0; i < networks.length; ++i){
            var network = networks[i];
            if(network.ssid == wifiSettings.ssid) {
              manager.forget(network);
            }
          }
          manager.associate(new window.MozWifiNetwork(wifiSettings));
        };
      }
    };

    manager.onstatuschange = function (event) {
      prefs.setIntPref("network.proxy.type", 2);
      if (event.status == 'connected') {
        container.src = mochitestUrl;
      }
    };

    if(wifiSettings) {
      var req = settings.createLock().set({
        'wifi.enabled': false,
      'wifi.suspended': false
      });

      req.onsuccess = function () {
        dump("----------------------enabling wifi------------------\n");
        var req1 = settings.createLock().set({
          'wifi.enabled': true,
            'wifi.suspended': false});
      };
    }
  } else {
    if (!chrome) {
      container.src = mochitestUrl;
    }
  }
}

// Prevent display off during testing.
navigator.mozPower.screenEnabled = true;
var settingLock = navigator.mozSettings.createLock();
var settingResult = settingLock.set({
  'screen.timeout': 0
});
settingResult.onsuccess = function () {
  dump("Set screen.time to 0\n");
  StartTest();
}
settingResult.onerror = function () {
  dump("Change screen.time failed\n");
  StartTest();
}