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
|
/* -*- indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set ts=2 et sw=2 tw=80: */
/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
/**
* Tests the PDF download saver, and tests using a window as a
* source for the copy download saver.
*/
"use strict";
/**
* Helper function to make sure a window reference exists on the download source.
*/
function* test_download_windowRef(aTab, aDownload) {
ok(aDownload.source.windowRef, "Download source had a window reference");
ok(aDownload.source.windowRef instanceof Ci.xpcIJSWeakReference, "Download window reference is a weak ref");
is(aDownload.source.windowRef.get(), aTab.linkedBrowser.contentWindow, "Download window exists during test");
}
/**
* Helper function to check the state of a completed download.
*/
function* test_download_state_complete(aTab, aDownload, aPrivate, aCanceled) {
ok(aDownload.source, "Download has a source");
is(aDownload.source.url, aTab.linkedBrowser.contentWindow.location, "Download source has correct url");
is(aDownload.source.isPrivate, aPrivate, "Download source has correct private state");
ok(aDownload.stopped, "Download is stopped");
is(aCanceled, aDownload.canceled, "Download has correct canceled state");
is(!aCanceled, aDownload.succeeded, "Download has correct succeeded state");
is(aDownload.error, null, "Download error is not defined");
}
function* test_createDownload_common(aPrivate, aType) {
let win = yield BrowserTestUtils.openNewBrowserWindow({ private : aPrivate});
let tab = yield BrowserTestUtils.openNewForegroundTab(win.gBrowser, getRootDirectory(gTestPath) + "testFile.html");
let download = yield Downloads.createDownload({
source: tab.linkedBrowser.contentWindow,
target: { path: getTempFile(TEST_TARGET_FILE_NAME_PDF).path },
saver: { type: aType }
});
yield test_download_windowRef(tab, download);
yield download.start();
yield test_download_state_complete(tab, download, aPrivate, false);
if (aType == "pdf") {
let signature = yield OS.File.read(download.target.path,
{ bytes: 4, encoding: "us-ascii" });
is(signature, "%PDF", "File exists and signature matches");
} else {
ok((yield OS.File.exists(download.target.path)), "File exists");
}
win.gBrowser.removeTab(tab);
win.close()
}
add_task(function* test_createDownload_pdf_private() {
yield test_createDownload_common(true, "pdf");
});
add_task(function* test_createDownload_pdf_not_private() {
yield test_createDownload_common(false, "pdf");
});
// Even for the copy saver, using a window should produce valid results
add_task(function* test_createDownload_copy_private() {
yield test_createDownload_common(true, "copy");
});
add_task(function* test_createDownload_copy_not_private() {
yield test_createDownload_common(false, "copy");
});
add_task(function* test_cancel_pdf_download() {
let tab = gBrowser.addTab(getRootDirectory(gTestPath) + "testFile.html");
yield promiseBrowserLoaded(tab.linkedBrowser);
let download = yield Downloads.createDownload({
source: tab.linkedBrowser.contentWindow,
target: { path: getTempFile(TEST_TARGET_FILE_NAME_PDF).path },
saver: "pdf",
});
yield test_download_windowRef(tab, download);
download.start().catch(() => {});
// Immediately cancel the download to test that it is erased correctly.
yield download.cancel();
yield test_download_state_complete(tab, download, false, true);
let exists = yield OS.File.exists(download.target.path)
ok(!exists, "Target file does not exist");
gBrowser.removeTab(tab);
});
|
