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
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_SAFE_BROWSING_ANDROID_SAFE_BROWSING_API_HANDLER_TEST_UTIL_H_
#define COMPONENTS_SAFE_BROWSING_ANDROID_SAFE_BROWSING_API_HANDLER_TEST_UTIL_H_
#include "components/safe_browsing/android/safe_browsing_api_handler_util.h"
class GURL;
namespace safe_browsing::test {
// Test-only mirror of SafetyNetApiHandler.SafetyNetApiState in Java.
enum class SafetyNetApiInitializationState {
kNotAvailable = 0,
kInitialized = 1,
kInitializedFirstParty = 2,
};
// Mixin for unittests that need to mock various functionality of
// SafeBrowsingApiHandlerBridge, SafeBrowsingApiHandler, and
// SafetyNetApiHandler. Test fixtures should inherit from this class to add the
// mock behavior.
class WithMockSafeBrowsingApiHandler {
public:
// This value should be aligned with DEFAULT_CHECK_DELTA_MICROSECONDS in
// SafeBrowsingApiHandlerBridgeNativeUnitTestHelper.MockSafeBrowsingApiHandler
static constexpr int kExpectedSafeBrowsingCheckDeltaMicroseconds = 15;
// Call these in SetUp/TearDown for the test fixture.
void SetUp();
void TearDown();
// Fake the initialization state for SafetyNetApiHandler.
void SetSafetyNetApiInitializationState(
SafetyNetApiInitializationState state);
// Set a fake result for VerifyAppsEnabled.
void SetVerifyAppsResult(VerifyAppsEnabledResult result);
// Make the SafetyNetId show up as empty string.
void SetSafetyNetIdResultEmpty();
// Add a fake entry to one of the local allowlists. Note that the fake
// allowlists do not check for domain and path match; they only check that the
// URL spec is identical.
void AddLocalAllowlistEntry(const GURL& url,
bool is_download_allowlist,
bool is_match);
void AddSafeBrowsingResponse(
const GURL& url,
const SafeBrowsingApiLookupResult& returned_lookup_result,
const SafeBrowsingJavaThreatType& returned_threat_type,
const std::vector<SafeBrowsingJavaThreatAttribute>&
returned_threat_attributes,
const SafeBrowsingJavaResponseStatus& returned_response_status,
const std::vector<SafeBrowsingJavaThreatType>& expected_threat_types,
const SafeBrowsingJavaProtocol& expected_protocol);
void RunHashDatabaseUrlCheck(
const GURL& url,
const SBThreatTypeSet& threat_types,
SBThreatType expected_threat_type,
SubresourceFilterMatch expected_subresource_filter_match);
void RunHashRealTimeUrlCheck(const GURL& url,
const SBThreatTypeSet& threat_types,
SBThreatType expected_threat_type);
protected:
raw_ptr<JNIEnv> env_;
};
} // namespace safe_browsing::test
#endif // COMPONENTS_SAFE_BROWSING_ANDROID_SAFE_BROWSING_API_HANDLER_TEST_UTIL_H_
|
