2012-05-21 04:12:37 -07:00
|
|
|
/* 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/. */
|
2007-05-23 16:52:50 -07:00
|
|
|
|
|
|
|
// This file tests the download manager backend
|
|
|
|
|
2012-08-14 07:06:12 -07:00
|
|
|
const Cc = Components.classes;
|
|
|
|
const Ci = Components.interfaces;
|
|
|
|
const Cu = Components.utils;
|
|
|
|
const Cr = Components.results;
|
|
|
|
|
2009-05-14 04:37:40 -07:00
|
|
|
do_get_profile();
|
2007-05-23 16:52:50 -07:00
|
|
|
|
2012-08-14 07:06:12 -07:00
|
|
|
Cu.import("resource://gre/modules/XPCOMUtils.jsm");
|
|
|
|
Cu.import("resource://gre/modules/Services.jsm");
|
|
|
|
Cu.import("resource://testing-common/httpd.js");
|
2010-11-22 19:36:42 -08:00
|
|
|
|
|
|
|
var downloadUtils = { };
|
|
|
|
XPCOMUtils.defineLazyServiceGetter(downloadUtils,
|
|
|
|
"downloadManager",
|
|
|
|
"@mozilla.org/download-manager;1",
|
|
|
|
Ci.nsIDownloadManager);
|
|
|
|
|
2007-05-23 16:52:50 -07:00
|
|
|
function createURI(aObj)
|
|
|
|
{
|
|
|
|
var ios = Cc["@mozilla.org/network/io-service;1"].
|
|
|
|
getService(Ci.nsIIOService);
|
|
|
|
return (aObj instanceof Ci.nsIFile) ? ios.newFileURI(aObj) :
|
|
|
|
ios.newURI(aObj, null, null);
|
|
|
|
}
|
|
|
|
|
|
|
|
var dirSvc = Cc["@mozilla.org/file/directory_service;1"].
|
|
|
|
getService(Ci.nsIProperties);
|
2009-08-05 12:36:20 -07:00
|
|
|
|
2009-05-14 04:37:40 -07:00
|
|
|
var provider = {
|
|
|
|
getFile: function(prop, persistent) {
|
|
|
|
persistent.value = true;
|
|
|
|
if (prop == "DLoads") {
|
|
|
|
var file = dirSvc.get("ProfD", Ci.nsILocalFile);
|
|
|
|
file.append("downloads.rdf");
|
|
|
|
return file;
|
|
|
|
}
|
|
|
|
print("*** Throwing trying to get " + prop);
|
|
|
|
throw Cr.NS_ERROR_FAILURE;
|
|
|
|
},
|
|
|
|
QueryInterface: function(iid) {
|
2010-04-01 04:29:13 -07:00
|
|
|
if (iid.equals(Ci.nsIDirectoryServiceProvider) ||
|
2009-05-14 04:37:40 -07:00
|
|
|
iid.equals(Ci.nsISupports)) {
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
throw Cr.NS_ERROR_NO_INTERFACE;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
dirSvc.QueryInterface(Ci.nsIDirectoryService).registerProvider(provider);
|
2007-12-26 17:42:05 -08:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Imports a download test file to use. Works with rdf and sqlite files.
|
|
|
|
*
|
|
|
|
* @param aFName
|
|
|
|
* The name of the file to import. This file should be located in the
|
|
|
|
* same directory as this file.
|
|
|
|
*/
|
2007-05-25 10:25:51 -07:00
|
|
|
function importDownloadsFile(aFName)
|
|
|
|
{
|
2009-03-21 08:20:00 -07:00
|
|
|
var file = do_get_file(aFName);
|
2007-05-25 10:25:51 -07:00
|
|
|
var newFile = dirSvc.get("ProfD", Ci.nsIFile);
|
2007-12-26 17:42:05 -08:00
|
|
|
if (/\.rdf$/i.test(aFName))
|
|
|
|
file.copyTo(newFile, "downloads.rdf");
|
|
|
|
else if (/\.sqlite$/i.test(aFName))
|
|
|
|
file.copyTo(newFile, "downloads.sqlite");
|
|
|
|
else
|
|
|
|
do_throw("Unexpected filename!");
|
2007-05-25 10:25:51 -07:00
|
|
|
}
|
|
|
|
|
2007-06-04 17:03:35 -07:00
|
|
|
var gDownloadCount = 0;
|
2007-10-30 11:31:50 -07:00
|
|
|
/**
|
|
|
|
* Adds a download to the DM, and starts it.
|
2008-11-03 15:22:21 -08:00
|
|
|
* @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
|
2012-10-03 10:10:20 -07:00
|
|
|
* isPrivate: whether the download is private or not
|
2007-10-30 11:31:50 -07:00
|
|
|
*/
|
2008-11-03 15:22:21 -08:00
|
|
|
function addDownload(aParams)
|
2007-06-04 17:03:35 -07:00
|
|
|
{
|
2008-11-03 15:22:21 -08:00
|
|
|
if (!aParams)
|
|
|
|
aParams = {};
|
|
|
|
if (!("resultFileName" in aParams))
|
|
|
|
aParams.resultFileName = "download.result";
|
|
|
|
if (!("targetFile" in aParams)) {
|
|
|
|
aParams.targetFile = dirSvc.get("ProfD", Ci.nsIFile);
|
|
|
|
aParams.targetFile.append(aParams.resultFileName);
|
|
|
|
}
|
|
|
|
if (!("sourceURI" in aParams))
|
2009-05-14 04:37:40 -07:00
|
|
|
aParams.sourceURI = "http://localhost:4444/head_download_manager.js";
|
2008-11-03 15:22:21 -08:00
|
|
|
if (!("downloadName" in aParams))
|
|
|
|
aParams.downloadName = null;
|
|
|
|
if (!("runBeforeStart" in aParams))
|
|
|
|
aParams.runBeforeStart = function () {};
|
|
|
|
|
2007-06-04 17:03:35 -07:00
|
|
|
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;
|
|
|
|
|
2007-08-15 10:56:50 -07:00
|
|
|
// it is part of the active downloads the moment addDownload is called
|
|
|
|
gDownloadCount++;
|
|
|
|
|
2012-10-04 12:08:28 -07:00
|
|
|
let dm = downloadUtils.downloadManager;
|
2008-11-03 15:22:21 -08:00
|
|
|
var dl = dm.addDownload(Ci.nsIDownloadManager.DOWNLOAD_TYPE_DOWNLOAD,
|
|
|
|
createURI(aParams.sourceURI),
|
|
|
|
createURI(aParams.targetFile), aParams.downloadName, null,
|
2012-10-03 10:10:20 -07:00
|
|
|
Math.round(Date.now() * 1000), null, persist, aParams.isPrivate);
|
2007-06-04 17:03:35 -07:00
|
|
|
|
|
|
|
// This will throw if it isn't found, and that would mean test failure, so no
|
|
|
|
// try catch block
|
|
|
|
var test = dm.getDownload(dl.id);
|
|
|
|
|
2008-11-03 15:22:21 -08:00
|
|
|
aParams.runBeforeStart.call(undefined, dl);
|
|
|
|
|
2007-06-04 17:03:35 -07:00
|
|
|
persist.progressListener = dl.QueryInterface(Ci.nsIWebProgressListener);
|
2012-10-04 12:07:51 -07:00
|
|
|
persist.savePrivacyAwareURI(dl.source, null, null, null, null, dl.targetFile,
|
|
|
|
aParams.isPrivate);
|
2007-06-04 17:03:35 -07:00
|
|
|
|
|
|
|
return dl;
|
|
|
|
}
|
|
|
|
|
|
|
|
function getDownloadListener()
|
|
|
|
{
|
|
|
|
return {
|
|
|
|
onDownloadStateChange: function(aState, aDownload)
|
|
|
|
{
|
2008-06-12 06:56:40 -07:00
|
|
|
if (aDownload.state == Ci.nsIDownloadManager.DOWNLOAD_QUEUED)
|
|
|
|
do_test_pending();
|
2007-06-04 17:03:35 -07:00
|
|
|
|
2008-06-12 06:56:40 -07:00
|
|
|
if (aDownload.state == Ci.nsIDownloadManager.DOWNLOAD_FINISHED ||
|
|
|
|
aDownload.state == Ci.nsIDownloadManager.DOWNLOAD_CANCELED ||
|
2007-06-04 17:03:35 -07:00
|
|
|
aDownload.state == Ci.nsIDownloadManager.DOWNLOAD_FAILED) {
|
|
|
|
gDownloadCount--;
|
2008-06-12 06:56:40 -07:00
|
|
|
do_test_finished();
|
2007-06-04 17:03:35 -07:00
|
|
|
}
|
2008-07-22 13:13:35 -07:00
|
|
|
|
2009-04-15 13:19:35 -07:00
|
|
|
if (gDownloadCount == 0 && typeof httpserv != "undefined" && httpserv)
|
|
|
|
{
|
|
|
|
do_test_pending();
|
|
|
|
httpserv.stop(do_test_finished);
|
|
|
|
}
|
2007-06-04 17:03:35 -07:00
|
|
|
},
|
|
|
|
onStateChange: function(a, b, c, d, e) { },
|
|
|
|
onProgressChange: function(a, b, c, d, e, f, g) { },
|
|
|
|
onSecurityChange: function(a, b, c, d) { }
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2010-11-19 05:23:35 -08:00
|
|
|
XPCOMUtils.defineLazyGetter(this, "Services", function() {
|
|
|
|
Cu.import("resource://gre/modules/Services.jsm");
|
|
|
|
return Services;
|
|
|
|
});
|
|
|
|
|
2008-07-22 13:13:35 -07:00
|
|
|
// Disable alert service notifications
|
2010-11-19 05:23:35 -08:00
|
|
|
Services.prefs.setBoolPref("browser.download.manager.showAlertOnComplete", false);
|
2011-12-28 08:33:36 -08:00
|
|
|
|
|
|
|
do_register_cleanup(function() {
|
|
|
|
Services.obs.notifyObservers(null, "quit-application", null);
|
|
|
|
});
|