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
|
// 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.
#ifndef CHROME_BROWSER_CHROMEOS_POLICY_DLP_TEST_MOCK_DLP_RULES_MANAGER_H_
#define CHROME_BROWSER_CHROMEOS_POLICY_DLP_TEST_MOCK_DLP_RULES_MANAGER_H_
#include <utility>
#include "chrome/browser/chromeos/policy/dlp/dlp_rules_manager.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"
namespace policy {
class MockDlpRulesManager : public DlpRulesManager {
public:
explicit MockDlpRulesManager(Profile* profile);
~MockDlpRulesManager() override;
MOCK_CONST_METHOD2(IsRestricted,
Level(const GURL& source, Restriction restriction));
MOCK_CONST_METHOD4(IsRestrictedByAnyRule,
Level(const GURL& source,
Restriction restriction,
std::string* out_source_pattern,
RuleMetadata* out_rule_metadata));
MOCK_CONST_METHOD6(IsRestrictedDestination,
Level(const GURL& source,
const GURL& destination,
Restriction restriction,
std::string* out_source_pattern,
std::string* out_destination_pattern,
RuleMetadata* out_rule_metadata));
MOCK_CONST_METHOD5(IsRestrictedComponent,
Level(const GURL& source,
const data_controls::Component& destination,
Restriction restriction,
std::string* out_source_pattern,
RuleMetadata* out_rule_metadata));
MOCK_METHOD(AggregatedDestinations,
GetAggregatedDestinations,
(const GURL& source, Restriction restriction),
(override, const));
MOCK_METHOD(AggregatedComponents,
GetAggregatedComponents,
(const GURL& source, Restriction restriction),
(override, const));
MOCK_CONST_METHOD0(IsReportingEnabled, bool());
MOCK_CONST_METHOD0(GetReportingManager,
data_controls::DlpReportingManager*());
MOCK_CONST_METHOD0(GetDlpFilesController, DlpFilesController*());
MOCK_CONST_METHOD4(GetSourceUrlPattern,
std::string(const GURL& source_url,
Restriction restriction,
Level level,
RuleMetadata* out_rule_metadata));
MOCK_CONST_METHOD0(GetClipboardCheckSizeLimitInBytes, size_t());
MOCK_METHOD(bool, IsFilesPolicyEnabled, (), (override, const));
MOCK_CONST_METHOD2(
GetDisallowedFileTransfers,
std::vector<uint64_t>(const std::vector<FileMetadata>& files_entries,
const GURL& destination));
MOCK_METHOD(void, OnDataControlsRulesUpdate, ());
MOCK_METHOD(void, OnDataLeakPreventionRulesUpdate, ());
};
} // namespace policy
#endif // CHROME_BROWSER_CHROMEOS_POLICY_DLP_TEST_MOCK_DLP_RULES_MANAGER_H_
|
