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
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/safe_browsing/content/browser/download/download_stats.h"
#include <optional>
#include "base/files/file_path.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/metrics/user_action_tester.h"
#include "components/download/public/common/download_stats.h"
#include "components/safe_browsing/content/common/file_type_policies.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
// The below constants are based on download_file_types.asciipb.
const int kExeFileTypeUmaValue = 0;
const int kApkFileTypeUmaValue = 20;
} // namespace
namespace safe_browsing {
TEST(SafeBrowsingDownloadStatsTest, RecordDangerousDownloadWarningShown) {
base::HistogramTester histogram_tester;
base::UserActionTester user_action_tester;
RecordDangerousDownloadWarningShown(
download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT,
base::FilePath(FILE_PATH_LITERAL("file.apk")),
/*is_https=*/true, /*has_user_gesture=*/true);
histogram_tester.ExpectUniqueSample(
"SBClientDownload.Warning.FileType.Malicious.Shown",
/*sample=*/kApkFileTypeUmaValue, /*expected_bucket_count=*/1);
histogram_tester.ExpectUniqueSample(
"SBClientDownload.Warning.DownloadIsHttps.Malicious.Shown",
/*sample=*/1, /*expected_bucket_count=*/1);
histogram_tester.ExpectUniqueSample(
"SBClientDownload.Warning.DownloadHasUserGesture.Malicious.Shown",
/*sample=*/1, /*expected_bucket_count=*/1);
EXPECT_EQ(1, user_action_tester.GetActionCount(
"SafeBrowsing.Download.WarningShown"));
RecordDangerousDownloadWarningShown(
download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT,
base::FilePath(FILE_PATH_LITERAL("file.exe")),
/*is_https=*/false, /*has_user_gesture=*/false);
histogram_tester.ExpectBucketCount(
"SBClientDownload.Warning.FileType.Uncommon.Shown",
/*sample=*/kExeFileTypeUmaValue,
/*expected_count=*/1);
histogram_tester.ExpectBucketCount(
"SBClientDownload.Warning.DownloadIsHttps.Uncommon.Shown",
/*sample=*/0,
/*expected_count=*/1);
histogram_tester.ExpectBucketCount(
"SBClientDownload.Warning.DownloadHasUserGesture.Uncommon.Shown",
/*sample=*/0,
/*expected_count=*/1);
EXPECT_EQ(2, user_action_tester.GetActionCount(
"SafeBrowsing.Download.WarningShown"));
}
TEST(SafeBrowsingDownloadStatsTest, RecordDangerousDownloadWarningBypassed) {
base::HistogramTester histogram_tester;
base::UserActionTester user_action_tester;
RecordDangerousDownloadWarningBypassed(
download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE,
base::FilePath(FILE_PATH_LITERAL("file.apk")),
/*is_https=*/true, /*has_user_gesture=*/false);
histogram_tester.ExpectUniqueSample(
"SBClientDownload.Warning.FileType.DangerousFileType.Bypassed",
/*sample=*/kApkFileTypeUmaValue, /*expected_bucket_count=*/1);
histogram_tester.ExpectUniqueSample(
"SBClientDownload.Warning.DownloadIsHttps.DangerousFileType.Bypassed",
/*sample=*/1, /*expected_bucket_count=*/1);
histogram_tester.ExpectUniqueSample(
"SBClientDownload.Warning.DownloadHasUserGesture.DangerousFileType."
"Bypassed",
/*sample=*/0, /*expected_bucket_count=*/1);
EXPECT_EQ(1, user_action_tester.GetActionCount(
"SafeBrowsing.Download.WarningBypassed"));
}
TEST(SafeBrowsingDownloadStatsTest, RecordDownloadFileTypeAttributes) {
{
base::HistogramTester histogram_tester;
RecordDownloadFileTypeAttributes(DownloadFileType::ALLOW_ON_USER_GESTURE,
/*has_user_gesture=*/false,
/*visited_referrer_before=*/false,
/*latest_bypass_time=*/std::nullopt);
histogram_tester.ExpectUniqueSample(
"SBClientDownload.UserGestureFileType.Attributes",
/*sample=*/UserGestureFileTypeAttributes::TOTAL_TYPE_CHECKED,
/*expected_bucket_count=*/1);
}
{
base::HistogramTester histogram_tester;
RecordDownloadFileTypeAttributes(
DownloadFileType::ALLOW_ON_USER_GESTURE,
/*has_user_gesture=*/true,
/*visited_referrer_before=*/true,
/*latest_bypass_time=*/base::Time::Now() - base::Hours(1));
histogram_tester.ExpectBucketCount(
"SBClientDownload.UserGestureFileType.Attributes",
/*sample=*/UserGestureFileTypeAttributes::TOTAL_TYPE_CHECKED,
/*expected_count=*/1);
histogram_tester.ExpectBucketCount(
"SBClientDownload.UserGestureFileType.Attributes",
/*sample=*/UserGestureFileTypeAttributes::HAS_USER_GESTURE,
/*expected_count=*/1);
histogram_tester.ExpectBucketCount(
"SBClientDownload.UserGestureFileType.Attributes",
/*sample=*/UserGestureFileTypeAttributes::HAS_REFERRER_VISIT,
/*expected_count=*/1);
histogram_tester.ExpectBucketCount(
"SBClientDownload.UserGestureFileType.Attributes",
/*sample=*/
UserGestureFileTypeAttributes::HAS_BOTH_USER_GESTURE_AND_REFERRER_VISIT,
/*expected_count=*/1);
}
}
} // namespace safe_browsing
|
