mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-21 17:25:36 +00:00
Bug 814701 - toolkit/components/downloads/test/unit/test_privatebrowsing.js fails on Birch; r=ehsan
--HG-- rename : toolkit/components/downloads/test/unit/test_privatebrowsing.js => toolkit/components/downloads/test/browser/browser_privatebrowsing.js rename : toolkit/components/downloads/test/unit/head_download_manager.js => toolkit/components/downloads/test/browser/head.js
This commit is contained in:
parent
f601413333
commit
21ed21c6b3
@ -15,4 +15,11 @@ MOCHITEST_BROWSER_FILES = \
|
||||
browser_nsIDownloadManagerUI.js \
|
||||
$(NULL)
|
||||
|
||||
ifdef MOZ_PER_WINDOW_PRIVATE_BROWSING
|
||||
MOCHITEST_BROWSER_FILES += \
|
||||
browser_privatebrowsing.js \
|
||||
head.js \
|
||||
$(NULL)
|
||||
endif
|
||||
|
||||
include $(topsrcdir)/config/rules.mk
|
||||
|
@ -0,0 +1,219 @@
|
||||
/* 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/. */
|
||||
|
||||
/**
|
||||
Test download manager's interaction with the private browsing service.
|
||||
|
||||
An overview of what this test does follows:
|
||||
|
||||
* Open a normal window.
|
||||
* Create a download (Download-A) with specific details.
|
||||
* Check that Download-A is retrievable.
|
||||
* Open a private window.
|
||||
* Check that Download-A is accessible.
|
||||
* Create another download (Download-B) with specific and different details.
|
||||
* Check that Download-B is not retrievable.
|
||||
* Open a normal window.
|
||||
* Check that Download-A is retrievable.
|
||||
* Check that Download-B is not accessible.
|
||||
**/
|
||||
|
||||
function test() {
|
||||
waitForExplicitFinish();
|
||||
|
||||
let windowsToClose = [];
|
||||
let tmpDir = Services.dirsvc.get("TmpD", Ci.nsIFile);
|
||||
|
||||
// properties of Download-A
|
||||
let downloadA;
|
||||
const downloadASource = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAADAAAAAwCAYAAABXAvmHAAAAAXNSR0IArs4c6QAAAAZiS0dEAP8A/wD/oL2nkwAAAAlwSFlzAAALEwAACxMBAJqcGAAAAAd0SU1FB9gKHhQaLFEVkXsAAAAIdEVYdENvbW1lbnQA9syWvwAABXZJREFUaN7VmnuIVUUcxz/3uma5ZJJmrZGVuZWupGSZsVNwTRYJYk8vRzd6o0kglgpm9q/ZEhlBUEssUpTtqMixl6LlURtDwyzCWmLxkZZL6qZRi/nc/tjf2Ybjufd6797HOrDM7NzfmfP9zfzm9zxwkbdEIRYxyhsCTAYmAWOBkcAwYBBwFugEDgN7gd3AdmCTtn5HWRkwynsamA7U5bnEBqBFW395SRkwylsIzAWqnOnvgTVAG3AIOA4cAwYAlwFDgZuAUcB4YApQIc+2A29p6zcWlQGjvEeBJUC1TO0BmoAPtfXbc1yrH/AwMB+YKNNtwGJt/VUFZ8Ao713gOfn3O2CBtv7mAt2hUcAi4BmZatLWn10QBozyRgArgFoRixe09d8vhkYxypsKfAwMBrYBDdr6B/JmwChvNLBWRCYA6rX1/y6mWjTKqwQ+BVIiUvXa+q3p6JNZdj4E3wJMKTZ4AG39TuA+oFnevVaw5MaAiE01sEJbf4a2/rlSGSdt/S5gJrAqxJATA3Jha4GdwFPlsLDChBZbUSuYst8BUZUr5cKOyVU9FuFODAZagWuAaVEVG3cCS6SfWW7wchLHgcci2OIZEAtbDWzR1l/dVxw2bf1N4X0QjD2tIkI7V/oF7qQyqa40a58Rd6EVWA+8Z3VwNI4wwxqIs/e7OHnNVgdbY2gWAQ8JxsbzTkAcsyog0NbfeYGbUwFcBdwLvAq0KpNK5bHJlcDNwBPAFmVS7yiTSkZOYQ+wGqgSrOeJ0HTpmzO9yeogEf6JozZOrCfisK1VJjUihzWSwNXiRhwTktnA8zGPNkewdjMg/nwdcBr45EK3zerglNXBj1YHDSKjAJdHRTDLGl1WB4etDpYDs5yfZsWQfwUcAeoEc88JTA4JemFtX3fG+cYH651xdcxlPgd84WIOGZgk/Te9UBa7nfF1ea7hXvR/09BsdzGHDIyV/ucya8ypzvjrNDS7XMyhGh0p/S+9ePlYZ3zwQh9SJpUAhgD3A8tk+i/g5TSP7HcxhwwMk/5ILxiY74w3ZgGdziYclQiv0epgXxqaDhG1YS4DlY5hIofd6w/cAiwUxwvgH+CNPDdhKHAnMAHYl8YqnzXKOxFirsj1DVksagcw3epgfzY7EFmzUkRwLjADWKVM6k2rg3lplhgQNWSd0g/KkZ8zAtoCrwCjrQ6+zHVTrA46rQ52iD35SKZfVCZVH+OdDgT6hZjDEzgs4G9Md3Tpdq8IbZnjfc6RqNBtwx3MPSewV/pRfcD5dFX5HTG/17iYkxEjNIG+1S6NmRvvYk5GrFtdHwBd44x/i/l9sos5ZGCT9DcY5Y0pMwOuPVkXucBXSqzegzkpurVDgmeAhlIjViY1UJnUXcqkWkSNIq710qgZEA20Icxsu3agRURojlHeEm39E0UE3JWF5FfgEauDQ87uJ5yIseW8gEZS3O2iTp8s8SGcpDujvU4CmRqrg2hU+IBY/XY3HZ+ICepfk8VGauuf7AuqyCivQtRrNfCSm4aPxp2Nko8cLoz0lTZPwLdFawhxeaHFYYbCKK+2D+z+bU4+aHHW1KJkvppEvNYY5VWVOSv3mSibprjCRyLDw1Z07i5gkrb+6RKDvwTYDNwNbNPWV3F0mbLTDXIfbges5O1LBf4K4FsB35bJNiUzpPMOAPWywETgJ6O860sA/lpxE8bxf4EjbZUm1xLTn8CD2vpbiwA8IdpmKdCfQpSYIi9wi3yfA89q6/9RIPC3Ah9IOAmFLPJFXuSWWbskenpbW39HnsZpGvC4k04pXpk1xmK7he6DdKckNwI/AAejJSkJBWvorn/dI35XaQvdMYxk+tTgEHBKsgeDRa6jrTyfGsQwUraPPS769h+G3Ox+KOb9iAAAAABJRU5ErkJggg==";
|
||||
const downloadADest = "download-file-A";
|
||||
const downloadAName = "download-A";
|
||||
|
||||
// properties of Download-B
|
||||
let downloadB;
|
||||
const downloadBSource = "data:application/octet-stream;base64,iVBORw0KGgoAAAANSUhEUgAAADAAAAAwCAYAAABXAvmHAAAAAXNSR0IArs4c6QAAAAZiS0dEAP8A/wD/oL2nkwAAAAlwSFlzAAALEwAACxMBAJqcGAAAAAd0SU1FB9gKHhQaLFEVkXsAAAAIdEVYdENvbW1lbnQA9syWvwAABXZJREFUaN7VmnuIVUUcxz/3uma5ZJJmrZGVuZWupGSZsVNwTRYJYk8vRzd6o0kglgpm9q/ZEhlBUEssUpTtqMixl6LlURtDwyzCWmLxkZZL6qZRi/nc/tjf2Ybjufd6797HOrDM7NzfmfP9zfzm9zxwkbdEIRYxyhsCTAYmAWOBkcAwYBBwFugEDgN7gd3AdmCTtn5HWRkwynsamA7U5bnEBqBFW395SRkwylsIzAWqnOnvgTVAG3AIOA4cAwYAlwFDgZuAUcB4YApQIc+2A29p6zcWlQGjvEeBJUC1TO0BmoAPtfXbc1yrH/AwMB+YKNNtwGJt/VUFZ8Ao713gOfn3O2CBtv7mAt2hUcAi4BmZatLWn10QBozyRgArgFoRixe09d8vhkYxypsKfAwMBrYBDdr6B/JmwChvNLBWRCYA6rX1/y6mWjTKqwQ+BVIiUvXa+q3p6JNZdj4E3wJMKTZ4AG39TuA+oFnevVaw5MaAiE01sEJbf4a2/rlSGSdt/S5gJrAqxJATA3Jha4GdwFPlsLDChBZbUSuYst8BUZUr5cKOyVU9FuFODAZagWuAaVEVG3cCS6SfWW7wchLHgcci2OIZEAtbDWzR1l/dVxw2bf1N4X0QjD2tIkI7V/oF7qQyqa40a58Rd6EVWA+8Z3VwNI4wwxqIs/e7OHnNVgdbY2gWAQ8JxsbzTkAcsyog0NbfeYGbUwFcBdwLvAq0KpNK5bHJlcDNwBPAFmVS7yiTSkZOYQ+wGqgSrOeJ0HTpmzO9yeogEf6JozZOrCfisK1VJjUihzWSwNXiRhwTktnA8zGPNkewdjMg/nwdcBr45EK3zerglNXBj1YHDSKjAJdHRTDLGl1WB4etDpYDs5yfZsWQfwUcAeoEc88JTA4JemFtX3fG+cYH651xdcxlPgd84WIOGZgk/Te9UBa7nfF1ea7hXvR/09BsdzGHDIyV/ucya8ypzvjrNDS7XMyhGh0p/S+9ePlYZ3zwQh9SJpUAhgD3A8tk+i/g5TSP7HcxhwwMk/5ILxiY74w3ZgGdziYclQiv0epgXxqaDhG1YS4DlY5hIofd6w/cAiwUxwvgH+CNPDdhKHAnMAHYl8YqnzXKOxFirsj1DVksagcw3epgfzY7EFmzUkRwLjADWKVM6k2rg3lplhgQNWSd0g/KkZ8zAtoCrwCjrQ6+zHVTrA46rQ52iD35SKZfVCZVH+OdDgT6hZjDEzgs4G9Md3Tpdq8IbZnjfc6RqNBtwx3MPSewV/pRfcD5dFX5HTG/17iYkxEjNIG+1S6NmRvvYk5GrFtdHwBd44x/i/l9sos5ZGCT9DcY5Y0pMwOuPVkXucBXSqzegzkpurVDgmeAhlIjViY1UJnUXcqkWkSNIq710qgZEA20Icxsu3agRURojlHeEm39E0UE3JWF5FfgEauDQ87uJ5yIseW8gEZS3O2iTp8s8SGcpDujvU4CmRqrg2hU+IBY/XY3HZ+ICepfk8VGauuf7AuqyCivQtRrNfCSm4aPxp2Nko8cLoz0lTZPwLdFawhxeaHFYYbCKK+2D+z+bU4+aHHW1KJkvppEvNYY5VWVOSv3mSibprjCRyLDw1Z07i5gkrb+6RKDvwTYDNwNbNPWV3F0mbLTDXIfbges5O1LBf4K4FsB35bJNiUzpPMOAPWywETgJ6O860sA/lpxE8bxf4EjbZUm1xLTn8CD2vpbiwA8IdpmKdCfQpSYIi9wi3yfA89q6/9RIPC3Ah9IOAmFLPJFXuSWWbskenpbW39HnsZpGvC4k04pXpk1xmK7he6DdKckNwI/AAejJSkJBWvorn/dI35XaQvdMYxk+tTgEHBKsgeDRa6jrTyfGsQwUraPPS769h+G3Ox+KOb9iAAAAABJRU5ErkJggg==";
|
||||
const downloadBDest = "download-file-B";
|
||||
const downloadBName = "download-B";
|
||||
|
||||
// Create all target files
|
||||
let fileA = createFile(downloadADest);
|
||||
let fileB = createFile(downloadBDest);
|
||||
|
||||
let listener = {
|
||||
onDownloadStateChange: function(aState, aDownload) {
|
||||
switch (aDownload.state) {
|
||||
case Services.downloads.DOWNLOAD_FAILED:
|
||||
case Services.downloads.DOWNLOAD_CANCELED:
|
||||
case Services.downloads.DOWNLOAD_DIRTY:
|
||||
case Services.downloads.DOWNLOAD_BLOCKED_POLICY:
|
||||
// Fail!
|
||||
if (aDownload.targetFile.exists())
|
||||
aDownload.targetFile.remove(false);
|
||||
Services.downloads.removeListener(this);
|
||||
throw ("Download failed (name: " + aDownload.displayName + ", state: " + aDownload.state + ")");
|
||||
finish();
|
||||
break;
|
||||
case Services.downloads.DOWNLOAD_FINISHED:
|
||||
ok(aDownload.targetFile.exists(), "Check that targetFile exists");
|
||||
aDownload.targetFile.remove(false);
|
||||
this.onDownloadFinished();
|
||||
break;
|
||||
}
|
||||
},
|
||||
onStateChange: function(a, b, c, d, e) { },
|
||||
onProgressChange: function(a, b, c, d, e, f, g) { },
|
||||
onSecurityChange: function(a, b, c, d) { },
|
||||
onDownloadFinished: function() {
|
||||
//onDownloadFinished is only called one time (on Download-A finished)
|
||||
//because the listener isn't called on private windows (on Download-B finished)
|
||||
is(Services.downloads.activeDownloadCount, 0,
|
||||
"Check that activeDownloadCount is zero");
|
||||
|
||||
//Open private window
|
||||
testOnWindow(true, function(win) {
|
||||
// Create Download-B
|
||||
downloadB = addDownload({
|
||||
isPrivate: true,
|
||||
targetFile: fileB,
|
||||
sourceURI: downloadBSource,
|
||||
downloadName: downloadBName,
|
||||
runBeforeStart: function (aDownload) {
|
||||
checkDownloads(aDownload);
|
||||
}
|
||||
});
|
||||
// wait for Download-B to finish
|
||||
isDownloadComplete (downloadB);
|
||||
});
|
||||
}
|
||||
};
|
||||
Services.downloads.addListener(listener);
|
||||
|
||||
registerCleanupFunction(function() {
|
||||
Services.downloads.removeListener(listener);
|
||||
windowsToClose.forEach(function(win) {
|
||||
win.close();
|
||||
});
|
||||
});
|
||||
|
||||
function cleanUp() {
|
||||
Services.downloads.removeDownload(downloadA.id);
|
||||
if (downloadA.targetFile.exists()) {
|
||||
downloadA.targetFile.remove(false);
|
||||
}
|
||||
if (downloadB.targetFile.exists()) {
|
||||
downloadB.targetFile.remove(false);
|
||||
}
|
||||
Services.downloads.cleanUp();
|
||||
is(Services.downloads.activeDownloadCount, 0,
|
||||
"Make sure download DB is empty");
|
||||
finish();
|
||||
}
|
||||
|
||||
function testOnWindow(aIsPrivate, aCallback) {
|
||||
whenNewWindowLoaded(aIsPrivate, function(win) {
|
||||
windowsToClose.push(win);
|
||||
aCallback(win);
|
||||
});
|
||||
}
|
||||
|
||||
function createFile(aFileName) {
|
||||
let file = tmpDir.clone();
|
||||
file.append(aFileName);
|
||||
file.createUnique(Ci.nsIFile.NORMAL_FILE_TYPE, 0666);
|
||||
return file;
|
||||
}
|
||||
|
||||
function checkDownloads(aDownloadB) {
|
||||
// Check that Download-A is accessible
|
||||
ok(is_download_available(downloadA.id, downloadASource,
|
||||
fileA, downloadAName),
|
||||
"Check that Download-A is accessible");
|
||||
|
||||
// Check that Download-B is not accessible
|
||||
is(Services.downloads.activeDownloadCount, 0,
|
||||
"Check that Download-B is not retrievable");
|
||||
|
||||
// Try to access the Download-B but it will fail since it's private
|
||||
try {
|
||||
let downloadID = aDownloadB.id;
|
||||
ok(false,
|
||||
"Check that Download-B is not available");
|
||||
} catch (e) {
|
||||
ok(true,
|
||||
"Check that Download-B is not available");
|
||||
}
|
||||
}
|
||||
|
||||
function isDownloadComplete(aDownload) {
|
||||
executeSoon(function() {
|
||||
if (!aDownload.cancelable) {
|
||||
testOnWindow(false, function () {
|
||||
checkDownloads(downloadB);
|
||||
cleanUp();
|
||||
});
|
||||
} else {
|
||||
isDownloadComplete(aDownload);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// Make sure we're starting with an empty DB
|
||||
is(Services.downloads.activeDownloadCount, 0,
|
||||
"Make sure we're starting with an empty DB");
|
||||
|
||||
testOnWindow(false, function(win) {
|
||||
// Create Download-A
|
||||
downloadA = addDownload({
|
||||
isPrivate: false,
|
||||
targetFile: fileA,
|
||||
sourceURI: downloadASource,
|
||||
downloadName: downloadAName,
|
||||
runBeforeStart: function (aDownload) {
|
||||
// Check that Download-A is retrievable
|
||||
is(Services.downloads.activeDownloadCount, 1,
|
||||
"Check that Download-A is retrievable");
|
||||
ok(is_active_download_available(aDownload.id, downloadASource, fileA,
|
||||
downloadAName, false), "Check that active download (Download-A) is available");
|
||||
ok(is_download_available(aDownload.id, downloadASource, fileA,
|
||||
downloadAName), "Check that download (Download-A) is available");
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Try to see if an active download is available using the |activeDownloads|
|
||||
* property.
|
||||
*/
|
||||
function is_active_download_available(aID, aSrc, aDst, aName, aPrivate) {
|
||||
let enumerator = Services.downloads.activeDownloads;
|
||||
while (enumerator.hasMoreElements()) {
|
||||
let download = enumerator.getNext();
|
||||
if (download.id == aID &&
|
||||
download.source.spec == aSrc &&
|
||||
download.targetFile.path == aDst.path &&
|
||||
download.displayName == aName &&
|
||||
download.isPrivate == aPrivate)
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Try to see if a download is available using the |getDownload| method. The
|
||||
* download can both be active or inactive.
|
||||
*/
|
||||
function is_download_available(aID, aSrc, aDst, aName) {
|
||||
let download;
|
||||
try {
|
||||
download = Services.downloads.getDownload(aID);
|
||||
} catch (ex) {
|
||||
return false;
|
||||
}
|
||||
return (download.id == aID &&
|
||||
download.source.spec == aSrc &&
|
||||
download.targetFile.path == aDst.path &&
|
||||
download.displayName == aName);
|
||||
}
|
91
toolkit/components/downloads/test/browser/head.js
Normal file
91
toolkit/components/downloads/test/browser/head.js
Normal file
@ -0,0 +1,91 @@
|
||||
/* 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/. */
|
||||
|
||||
let HttpServer =
|
||||
Cu.import("resource://testing-common/httpd.js", {}).HttpServer;
|
||||
|
||||
function createURI(aObj) {
|
||||
return (aObj instanceof Ci.nsIFile) ?
|
||||
Services.io.newFileURI(aObj) : Services.io.newURI(aObj, null, null);
|
||||
}
|
||||
|
||||
var gDownloadCount = 0;
|
||||
/**
|
||||
* Adds a download to the DM, and starts it.
|
||||
* @param aParams (optional): an optional object which contains the function
|
||||
* parameters:
|
||||
* resultFileName: leaf node for the target file
|
||||
* targetFile: nsIFile for the target (overrides resultFileName)
|
||||
* sourceURI: the download source URI
|
||||
* downloadName: the display name of the download
|
||||
* runBeforeStart: a function to run before starting the download
|
||||
* isPrivate: whether the download is private or not
|
||||
*/
|
||||
function addDownload(aParams) {
|
||||
if (!aParams)
|
||||
aParams = {};
|
||||
if (!("resultFileName" in aParams))
|
||||
aParams.resultFileName = "download.result";
|
||||
if (!("targetFile" in aParams)) {
|
||||
aParams.targetFile = Services.dirsvc.get("ProfD", Ci.nsIFile);
|
||||
aParams.targetFile.append(aParams.resultFileName);
|
||||
}
|
||||
if (!("sourceURI" in aParams))
|
||||
aParams.sourceURI = "http://localhost:4444/head.js";
|
||||
if (!("downloadName" in aParams))
|
||||
aParams.downloadName = null;
|
||||
if (!("runBeforeStart" in aParams))
|
||||
aParams.runBeforeStart = function () {};
|
||||
|
||||
const nsIWBP = Ci.nsIWebBrowserPersist;
|
||||
var persist = Cc["@mozilla.org/embedding/browser/nsWebBrowserPersist;1"].
|
||||
createInstance(Ci.nsIWebBrowserPersist);
|
||||
persist.persistFlags = nsIWBP.PERSIST_FLAGS_REPLACE_EXISTING_FILES |
|
||||
nsIWBP.PERSIST_FLAGS_BYPASS_CACHE |
|
||||
nsIWBP.PERSIST_FLAGS_AUTODETECT_APPLY_CONVERSION;
|
||||
|
||||
// it is part of the active downloads the moment addDownload is called
|
||||
gDownloadCount++;
|
||||
|
||||
var dl =
|
||||
Services.downloads.addDownload(
|
||||
Ci.nsIDownloadManager.DOWNLOAD_TYPE_DOWNLOAD,
|
||||
createURI(aParams.sourceURI), createURI(aParams.targetFile),
|
||||
aParams.downloadName, null, Math.round(Date.now() * 1000), null,
|
||||
persist, aParams.isPrivate);
|
||||
|
||||
// This will throw if it isn't found, and that would mean test failure, so no
|
||||
// try catch block
|
||||
if (!aParams.isPrivate)
|
||||
Services.downloads.getDownload(dl.id);
|
||||
|
||||
aParams.runBeforeStart.call(undefined, dl);
|
||||
|
||||
persist.progressListener = dl.QueryInterface(Ci.nsIWebProgressListener);
|
||||
persist.savePrivacyAwareURI(dl.source, null, null, null, null, dl.targetFile,
|
||||
aParams.isPrivate);
|
||||
|
||||
return dl;
|
||||
}
|
||||
|
||||
function whenNewWindowLoaded(aIsPrivate, aCallback) {
|
||||
let win = OpenBrowserWindow({private: aIsPrivate});
|
||||
win.addEventListener("load", function onLoad() {
|
||||
win.removeEventListener("load", onLoad, false);
|
||||
executeSoon(function() aCallback(win));
|
||||
}, false);
|
||||
}
|
||||
|
||||
function startServer() {
|
||||
let httpServer = new HttpServer();
|
||||
let currentWorkDir = Services.dirsvc.get("CurWorkD", Ci.nsILocalFile);
|
||||
httpServer.registerDirectory("/file/", currentWorkDir);
|
||||
httpServer.registerPathHandler("/noresume", function (meta, response) {
|
||||
response.setHeader("Content-Type", "text/html", false);
|
||||
response.setHeader("Accept-Ranges", "none", false);
|
||||
response.write("foo");
|
||||
});
|
||||
httpServer.start(4444);
|
||||
return httpServer;
|
||||
}
|
@ -23,6 +23,7 @@ skip-if = os == "android"
|
||||
[test_old_download_files_removed.js]
|
||||
[test_private_resume.js]
|
||||
[test_privatebrowsing.js]
|
||||
skip-if = perwindowprivatebrowsing
|
||||
[test_privatebrowsing_cancel.js]
|
||||
[test_removeDownloadsByTimeframe.js]
|
||||
[test_resume.js]
|
||||
|
Loading…
Reference in New Issue
Block a user