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
|
// Copyright 2015 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/android/safe_browsing_api_handler_util.h"
#include <string>
#include "base/strings/stringprintf.h"
#include "components/safe_browsing/core/browser/db/util.h"
#include "components/safe_browsing/core/browser/db/v4_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace safe_browsing {
class SafeBrowsingApiHandlerUtilTest : public ::testing::Test {};
TEST_F(SafeBrowsingApiHandlerUtilTest, GetThreatMetadataFromSafeBrowsingApi) {
typedef SubresourceFilterLevel Level;
typedef SubresourceFilterType Type;
const struct {
SafeBrowsingJavaThreatType threat_type;
std::vector<int> threat_attributes;
SubresourceFilterMatch expected_match;
} test_cases[] = {
{SafeBrowsingJavaThreatType::SOCIAL_ENGINEERING, {}, {}},
{SafeBrowsingJavaThreatType::ABUSIVE_EXPERIENCE_VIOLATION,
{},
{{Type::ABUSIVE, Level::ENFORCE}}},
{SafeBrowsingJavaThreatType::ABUSIVE_EXPERIENCE_VIOLATION,
{static_cast<int>(SafeBrowsingJavaThreatAttribute::CANARY)},
{{Type::ABUSIVE, Level::WARN}}},
{SafeBrowsingJavaThreatType::ABUSIVE_EXPERIENCE_VIOLATION,
{static_cast<int>(SafeBrowsingJavaThreatAttribute::FRAME_ONLY)},
{{Type::ABUSIVE, Level::ENFORCE}}},
{SafeBrowsingJavaThreatType::BETTER_ADS_VIOLATION,
{},
{{Type::BETTER_ADS, Level::ENFORCE}}},
{SafeBrowsingJavaThreatType::BETTER_ADS_VIOLATION,
{static_cast<int>(SafeBrowsingJavaThreatAttribute::CANARY)},
{{Type::BETTER_ADS, Level::WARN}}},
{SafeBrowsingJavaThreatType::BETTER_ADS_VIOLATION,
{static_cast<int>(SafeBrowsingJavaThreatAttribute::FRAME_ONLY)},
{{Type::BETTER_ADS, Level::ENFORCE}}}};
for (const auto& test_case : test_cases) {
ThreatMetadata metadata = GetThreatMetadataFromSafeBrowsingApi(
test_case.threat_type, test_case.threat_attributes);
ThreatMetadata expected;
expected.subresource_filter_match = test_case.expected_match;
EXPECT_EQ(expected, metadata);
}
}
} // namespace safe_browsing
|
