1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
|
/* Any copyright is dedicated to the Public Domain.
http://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";
const RELATIVE_DIR = "toolkit/components/pdfjs/test/";
const TESTROOT = "http://example.com/browser/" + RELATIVE_DIR;
var MockFilePicker = SpecialPowers.MockFilePicker;
MockFilePicker.init(window);
/* import-globals-from ../../../content/tests/browser/common/mockTransfer.js */
Services.scriptloader.loadSubScript(
"chrome://mochitests/content/browser/toolkit/content/tests/browser/common/mockTransfer.js",
this
);
function createPromiseForTransferComplete(expectedFileName, destFile) {
return new Promise(resolve => {
MockFilePicker.showCallback = fp => {
info("Filepicker shown, checking filename");
is(fp.defaultString, expectedFileName, "Filename should be correct.");
let fileName = fp.defaultString;
destFile.append(fileName);
MockFilePicker.setFiles([destFile]);
MockFilePicker.filterIndex = 0; // kSaveAsType_Complete
MockFilePicker.showCallback = null;
mockTransferCallback = function(downloadSuccess) {
ok(downloadSuccess, "File should have been downloaded successfully");
mockTransferCallback = () => {};
resolve();
};
};
});
}
let tempDir = createTemporarySaveDirectory();
add_setup(async function() {
mockTransferRegisterer.register();
registerCleanupFunction(function() {
mockTransferRegisterer.unregister();
MockFilePicker.cleanup();
tempDir.remove(true);
});
});
/**
* Check triggering "Save Page As" on a non-forms PDF opens the "Save As" dialog
* and successfully saves the file.
*/
add_task(async function test_pdf_saveas() {
await BrowserTestUtils.withNewTab(
{ gBrowser, url: "about:blank" },
async function(browser) {
await waitForPdfJS(browser, TESTROOT + "file_pdfjs_test.pdf");
let destFile = tempDir.clone();
MockFilePicker.displayDirectory = tempDir;
let fileSavedPromise = createPromiseForTransferComplete(
"file_pdfjs_test.pdf",
destFile
);
saveBrowser(browser);
await fileSavedPromise;
}
);
});
/**
* Check triggering "Save Page As" on a PDF with forms that has been modified
* does the following:
* 1) opens the "Save As" dialog
* 2) successfully saves the file
* 3) the new file contains the new form data
*/
add_task(async function test_pdf_saveas_forms() {
await SpecialPowers.pushPrefEnv({
set: [["pdfjs.renderInteractiveForms", true]],
});
let destFile = tempDir.clone();
await BrowserTestUtils.withNewTab(
{ gBrowser, url: "about:blank" },
async function(browser) {
await waitForPdfJSAnnotationLayer(
browser,
TESTROOT + "file_pdfjs_form.pdf"
);
// Fill in the form input field.
await SpecialPowers.spawn(browser, [], async function() {
let formInput = content.document.querySelector(
"#viewerContainer input"
);
ok(formInput, "PDF contains text field.");
is(formInput.value, "", "Text field is empty to start.");
formInput.value = "test";
formInput.dispatchEvent(new content.window.Event("input"));
});
MockFilePicker.displayDirectory = tempDir;
let fileSavedPromise = createPromiseForTransferComplete(
"file_pdfjs_form.pdf",
destFile
);
saveBrowser(browser);
await fileSavedPromise;
}
);
// Now that the file has been modified and saved, load it to verify the form
// data persisted.
await BrowserTestUtils.withNewTab(
{ gBrowser, url: "about:blank" },
async function(browser) {
await waitForPdfJSAnnotationLayer(browser, NetUtil.newURI(destFile).spec);
await SpecialPowers.spawn(browser, [], async function() {
let formInput = content.document.querySelector(
"#viewerContainer input"
);
ok(formInput, "PDF contains text field.");
is(formInput.value, "test", "Text field is filled in.");
});
}
);
});
/**
* Check triggering "Save Page As" on a PDF which was loaded with a custom
* content disposition filename defaults to using the provided filename.
*/
add_task(async function test_pdf_saveas_customname() {
// Ensure improvements_to_download_panel is enabled so that we render the
// attachment directly with the original request in a new tab.
await SpecialPowers.pushPrefEnv({
set: [
["browser.helperApps.showOpenOptionForPdfJS", true],
["browser.helperApps.showOpenOptionForViewableInternally", true],
["browser.download.improvements_to_download_panel", true],
["browser.download.always_ask_before_handling_new_types", false],
["browser.download.open_pdf_attachments_inline", true],
],
});
await BrowserTestUtils.withNewTab(
{ gBrowser, url: TESTROOT + "file_pdf_download_link.html" },
async function(browser) {
// Click on the download link in the loaded file. This will create a new
// tab with the PDF loaded in it.
BrowserTestUtils.synthesizeMouseAtCenter("#custom_filename", {}, browser);
let tab = await BrowserTestUtils.waitForNewTab(gBrowser);
info("tab created");
// Wait for the PDF's metadata to be fully loaded before downloading, as
// otherwise it won't be aware of the content disposition filename yet.
await BrowserTestUtils.waitForContentEvent(
tab.linkedBrowser,
"metadataloaded",
false,
null,
true
);
info("metadata loaded");
let destFile = tempDir.clone();
MockFilePicker.displayDirectory = tempDir;
let fileSavedPromise = createPromiseForTransferComplete(
"custom_filename.pdf",
destFile
);
saveBrowser(tab.linkedBrowser);
await fileSavedPromise;
BrowserTestUtils.removeTab(tab);
}
);
await SpecialPowers.popPrefEnv();
});
|
