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
|
/* -*- 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/ */
"use strict";
Services.prefs.setBoolPref(
"toolkit.telemetry.testing.overrideProductsCheck",
true
);
const errors = [
Downloads.Error.BLOCK_VERDICT_MALWARE,
Downloads.Error.BLOCK_VERDICT_POTENTIALLY_UNWANTED,
Downloads.Error.BLOCK_VERDICT_INSECURE,
Downloads.Error.BLOCK_VERDICT_UNCOMMON,
];
add_task(async function test_confirm_block_download() {
for (let error of errors) {
info(`Testing block ${error} download`);
let histogram = TelemetryTestUtils.getAndClearKeyedHistogram(
"DOWNLOADS_USER_ACTION_ON_BLOCKED_DOWNLOAD"
);
let download;
try {
info(`Create ${error} download`);
if (error == Downloads.Error.BLOCK_VERDICT_INSECURE) {
download = await promiseStartLegacyDownload(null, {
downloadClassification: Ci.nsITransfer.DOWNLOAD_POTENTIALLY_UNSAFE,
});
} else {
download = await promiseBlockedDownload({
keepPartialData: true,
keepBlockedData: true,
useLegacySaver: false,
verdict: error,
});
}
await download.start();
do_throw("The download should have failed.");
} catch (ex) {
if (!(ex instanceof Downloads.Error)) {
throw ex;
}
}
// Test blocked download is recorded
TelemetryTestUtils.assertKeyedHistogramValue(histogram, error, 0, 1);
// Test confirm block
histogram = TelemetryTestUtils.getAndClearKeyedHistogram(
"DOWNLOADS_USER_ACTION_ON_BLOCKED_DOWNLOAD"
);
info(`Block ${error} download`);
await download.confirmBlock();
TelemetryTestUtils.assertKeyedHistogramValue(histogram, error, 1, 1);
}
});
add_task(async function test_confirm_unblock_download() {
for (let error of errors) {
info(`Testing unblock ${error} download`);
let histogram = TelemetryTestUtils.getAndClearKeyedHistogram(
"DOWNLOADS_USER_ACTION_ON_BLOCKED_DOWNLOAD"
);
let download;
try {
info(`Create ${error} download`);
if (error == Downloads.Error.BLOCK_VERDICT_INSECURE) {
download = await promiseStartLegacyDownload(null, {
downloadClassification: Ci.nsITransfer.DOWNLOAD_POTENTIALLY_UNSAFE,
});
} else {
download = await promiseBlockedDownload({
keepPartialData: true,
keepBlockedData: true,
useLegacySaver: false,
verdict: error,
});
}
await download.start();
do_throw("The download should have failed.");
} catch (ex) {
if (!(ex instanceof Downloads.Error)) {
throw ex;
}
}
// Test blocked download is recorded
TelemetryTestUtils.assertKeyedHistogramValue(histogram, error, 0, 1);
// Test unblock
histogram = TelemetryTestUtils.getAndClearKeyedHistogram(
"DOWNLOADS_USER_ACTION_ON_BLOCKED_DOWNLOAD"
);
info(`Unblock ${error} download`);
let promise = new Promise(r => (download.onchange = r));
await download.unblock();
// The environment is not set up properly for performing a real download, cancel
// the unblocked download so it doesn't affect the next testcase.
await download.cancel();
await promise;
if (error == Downloads.Error.BLOCK_VERDICT_INSECURE) {
Assert.ok(!download.error, "Ensure we didn't set download.error");
}
TelemetryTestUtils.assertKeyedHistogramValue(histogram, error, 2, 1);
}
});
|
