Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

/* 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 MockFilePicker = SpecialPowers.MockFilePicker;
MockFilePicker.init(window.browsingContext);
/**
* When saving an inner frame as file only, the POST data of the outer page is
* sent to the address of the inner page.
*
* Web page generated by POST is retried as GET when Save Frame As used, and the
* page is no longer in the cache.
*/
function test() {
waitForExplicitFinish();
BrowserTestUtils.startLoadingURIString(
gBrowser,
);
gBrowser.addEventListener("pageshow", function pageShown(event) {
if (event.target.location == "about:blank") {
return;
}
gBrowser.removeEventListener("pageshow", pageShown);
// Submit the form in the outer page, then wait for both the outer
// document and the inner frame to be loaded again.
gBrowser.addEventListener("DOMContentLoaded", handleOuterSubmit);
gBrowser.contentDocument.getElementById("postForm").submit();
});
var framesLoaded = 0;
var innerFrame;
function handleOuterSubmit() {
if (++framesLoaded < 2) {
return;
}
gBrowser.removeEventListener("DOMContentLoaded", handleOuterSubmit);
innerFrame = gBrowser.contentDocument.getElementById("innerFrame");
// Submit the form in the inner page.
gBrowser.addEventListener("DOMContentLoaded", handleInnerSubmit);
innerFrame.contentDocument.getElementById("postForm").submit();
}
function handleInnerSubmit() {
gBrowser.removeEventListener("DOMContentLoaded", handleInnerSubmit);
// Create the folder the page will be saved into.
var destDir = createTemporarySaveDirectory();
var file = destDir.clone();
file.append("no_default_file_name");
MockFilePicker.setFiles([file]);
MockFilePicker.showCallback = function () {
MockFilePicker.filterIndex = 1; // kSaveAsType_URL
};
mockTransferCallback = onTransferComplete;
mockTransferRegisterer.register();
registerCleanupFunction(function () {
mockTransferRegisterer.unregister();
MockFilePicker.cleanup();
destDir.remove(true);
});
var docToSave = innerFrame.contentDocument;
// We call internalSave instead of saveDocument to bypass the history
// cache.
internalSave(
docToSave.location.href,
null,
docToSave,
null,
null,
docToSave.contentType,
false,
null,
null,
docToSave.referrer ? makeURI(docToSave.referrer) : null,
docToSave,
false,
null
);
}
function onTransferComplete(downloadSuccess) {
ok(
downloadSuccess,
"The inner frame should have been downloaded successfully"
);
// Read the entire saved file.
var file = MockFilePicker.getNsIFile();
var fileContents = readShortFile(file);
// Check if outer POST data is found (bug 471962).
is(
fileContents.indexOf("inputfield=outer"),
-1,
"The saved inner frame does not contain outer POST data"
);
// Check if inner POST data is found (bug 485196).
isnot(
fileContents.indexOf("inputfield=inner"),
-1,
"The saved inner frame was generated using the correct POST data"
);
finish();
}
}
Services.scriptloader.loadSubScript(
this
);
function createTemporarySaveDirectory() {
var saveDir = Services.dirsvc.get("TmpD", Ci.nsIFile);
saveDir.append("testsavedir");
if (!saveDir.exists()) {
saveDir.create(Ci.nsIFile.DIRECTORY_TYPE, 0o755);
}
return saveDir;
}
/**
* Reads the contents of the provided short file (up to 1 MiB).
*
* @param aFile
* nsIFile object pointing to the file to be read.
*
* @return
* String containing the raw octets read from the file.
*/
function readShortFile(aFile) {
var inputStream = Cc[
"@mozilla.org/network/file-input-stream;1"
].createInstance(Ci.nsIFileInputStream);
inputStream.init(aFile, -1, 0, 0);
try {
var scrInputStream = Cc[
"@mozilla.org/scriptableinputstream;1"
].createInstance(Ci.nsIScriptableInputStream);
scrInputStream.init(inputStream);
try {
// Assume that the file is much shorter than 1 MiB.
return scrInputStream.read(1048576);
} finally {
// Close the scriptable stream after reading, even if the operation
// failed.
scrInputStream.close();
}
} finally {
// Close the stream after reading, if it is still open, even if the read
// operation failed.
inputStream.close();
}
}