michael@0: /* Any copyright is dedicated to the Public Domain.
michael@0:    http://creativecommons.org/publicdomain/zero/1.0/ */
michael@0: 
michael@0: const TEST_BASE = "chrome://mochitests/content/browser/browser/devtools/styleeditor/test/";
michael@0: const TEST_BASE_HTTP = "http://example.com/browser/browser/devtools/styleeditor/test/";
michael@0: const TEST_BASE_HTTPS = "https://example.com/browser/browser/devtools/styleeditor/test/";
michael@0: const TEST_HOST = 'mochi.test:8888';
michael@0: 
michael@0: let {devtools} = Cu.import("resource://gre/modules/devtools/Loader.jsm", {});
michael@0: let TargetFactory = devtools.TargetFactory;
michael@0: let {LoadContextInfo} = Cu.import("resource://gre/modules/LoadContextInfo.jsm", {});
michael@0: let {console} = Cu.import("resource://gre/modules/devtools/Console.jsm", {});
michael@0: let {Promise: promise} = Cu.import("resource://gre/modules/Promise.jsm", {});
michael@0: 
michael@0: let gPanelWindow;
michael@0: let cache = Cc["@mozilla.org/netwerk/cache-storage-service;1"]
michael@0:               .getService(Ci.nsICacheStorageService);
michael@0: 
michael@0: 
michael@0: // Import the GCLI test helper
michael@0: let testDir = gTestPath.substr(0, gTestPath.lastIndexOf("/"));
michael@0: Services.scriptloader.loadSubScript(testDir + "../../../commandline/test/helpers.js", this);
michael@0: 
michael@0: gDevTools.testing = true;
michael@0: SimpleTest.registerCleanupFunction(() => {
michael@0:   gDevTools.testing = false;
michael@0: });
michael@0: 
michael@0: /**
michael@0:  * Define an async test based on a generator function
michael@0:  */
michael@0: function asyncTest(generator) {
michael@0:   return () => Task.spawn(generator).then(null, ok.bind(null, false)).then(finish);
michael@0: }
michael@0: 
michael@0: function* cleanup()
michael@0: {
michael@0:   gPanelWindow = null;
michael@0:   while (gBrowser.tabs.length > 1) {
michael@0:     let target = TargetFactory.forTab(gBrowser.selectedTab);
michael@0:     yield gDevTools.closeToolbox(target);
michael@0: 
michael@0:     gBrowser.removeCurrentTab();
michael@0:   }
michael@0: }
michael@0: 
michael@0: function addTabAndOpenStyleEditors(count, callback, uri) {
michael@0:   let deferred = promise.defer();
michael@0:   let currentCount = 0;
michael@0:   let panel;
michael@0:   addTabAndCheckOnStyleEditorAdded(p => panel = p, function () {
michael@0:     currentCount++;
michael@0:     info(currentCount + " of " + count + " editors opened");
michael@0:     if (currentCount == count) {
michael@0:       if (callback) {
michael@0:         callback(panel);
michael@0:       }
michael@0:       deferred.resolve(panel);
michael@0:     }
michael@0:   });
michael@0: 
michael@0:   if (uri) {
michael@0:     content.location = uri;
michael@0:   }
michael@0:   return deferred.promise;
michael@0: }
michael@0: 
michael@0: function addTabAndCheckOnStyleEditorAdded(callbackOnce, callbackOnAdded) {
michael@0:   gBrowser.selectedTab = gBrowser.addTab();
michael@0:   gBrowser.selectedBrowser.addEventListener("load", function onLoad() {
michael@0:     gBrowser.selectedBrowser.removeEventListener("load", onLoad, true);
michael@0:     openStyleEditorInWindow(window, function (panel) {
michael@0:       // Execute the individual callback with the panel argument.
michael@0:       callbackOnce(panel);
michael@0:       // Report editors that already opened while loading.
michael@0:       for (let editor of panel.UI.editors) {
michael@0:         callbackOnAdded(editor);
michael@0:       }
michael@0:       // Report new editors added afterwards.
michael@0:       panel.UI.on("editor-added", (event, editor) => callbackOnAdded(editor));
michael@0:     });
michael@0:   }, true);
michael@0: }
michael@0: 
michael@0: function openStyleEditorInWindow(win, callback) {
michael@0:   let target = TargetFactory.forTab(win.gBrowser.selectedTab);
michael@0:   win.gDevTools.showToolbox(target, "styleeditor").then(function(toolbox) {
michael@0:     let panel = toolbox.getCurrentPanel();
michael@0:     gPanelWindow = panel._panelWin;
michael@0: 
michael@0:     panel.UI._alwaysDisableAnimations = true;
michael@0:     callback(panel);
michael@0:   });
michael@0: }
michael@0: 
michael@0: function checkDiskCacheFor(host, done)
michael@0: {
michael@0:   let foundPrivateData = false;
michael@0: 
michael@0:   Visitor.prototype = {
michael@0:     onCacheStorageInfo: function(num, consumption)
michael@0:     {
michael@0:       info("disk storage contains " + num + " entries");
michael@0:     },
michael@0:     onCacheEntryInfo: function(entry)
michael@0:     {
michael@0:       info(entry.key);
michael@0:       foundPrivateData |= entry.key.contains(host);
michael@0:     },
michael@0:     onCacheEntryVisitCompleted: function()
michael@0:     {
michael@0:       is(foundPrivateData, false, "web content present in disk cache");
michael@0:       done();
michael@0:     }
michael@0:   };
michael@0:   function Visitor() {}
michael@0: 
michael@0:   var storage = cache.diskCacheStorage(LoadContextInfo.default, false);
michael@0:   storage.asyncVisitStorage(new Visitor(), true /* Do walk entries */);
michael@0: }
michael@0: 
michael@0: registerCleanupFunction(cleanup);