1 //
2 // Copyright (C) 2014 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 //      http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16 
17 #ifndef UPDATE_ENGINE_UPDATE_MANAGER_MOCK_POLICY_H_
18 #define UPDATE_ENGINE_UPDATE_MANAGER_MOCK_POLICY_H_
19 
20 #include <string>
21 
22 #include <gmock/gmock.h>
23 
24 #include "update_engine/update_manager/default_policy.h"
25 #include "update_engine/update_manager/policy.h"
26 
27 namespace chromeos_update_manager {
28 
29 // A mocked implementation of Policy.
30 class MockPolicy : public Policy {
31  public:
32   explicit MockPolicy(chromeos_update_engine::ClockInterface* clock)
33       : default_policy_(clock) {
34     // We defer to the corresponding DefaultPolicy methods, by default.
35     ON_CALL(*this,
36             UpdateCheckAllowed(testing::_, testing::_, testing::_, testing::_))
37         .WillByDefault(testing::Invoke(&default_policy_,
38                                        &DefaultPolicy::UpdateCheckAllowed));
39     ON_CALL(*this,
40             UpdateCanBeApplied(
41                 testing::_, testing::_, testing::_, testing::_, testing::_))
42         .WillByDefault(testing::Invoke(&default_policy_,
43                                        &DefaultPolicy::UpdateCanBeApplied));
44     ON_CALL(*this,
45             UpdateCanStart(
46                 testing::_, testing::_, testing::_, testing::_, testing::_))
47         .WillByDefault(
48             testing::Invoke(&default_policy_, &DefaultPolicy::UpdateCanStart));
49     ON_CALL(
50         *this,
51         UpdateDownloadAllowed(testing::_, testing::_, testing::_, testing::_))
52         .WillByDefault(testing::Invoke(&default_policy_,
53                                        &DefaultPolicy::UpdateDownloadAllowed));
54     ON_CALL(*this, P2PEnabled(testing::_, testing::_, testing::_, testing::_))
55         .WillByDefault(
56             testing::Invoke(&default_policy_, &DefaultPolicy::P2PEnabled));
57     ON_CALL(*this,
58             P2PEnabledChanged(
59                 testing::_, testing::_, testing::_, testing::_, testing::_))
60         .WillByDefault(testing::Invoke(&default_policy_,
61                                        &DefaultPolicy::P2PEnabledChanged));
62   }
63 
64   MockPolicy() : MockPolicy(nullptr) {}
65   ~MockPolicy() override {}
66 
67   // Policy overrides.
68   MOCK_CONST_METHOD4(
69       UpdateCheckAllowed,
70       EvalStatus(EvaluationContext*, State*, std::string*, UpdateCheckParams*));
71 
72   MOCK_CONST_METHOD5(UpdateCanBeApplied,
73                      EvalStatus(EvaluationContext*,
74                                 State*,
75                                 std::string*,
76                                 chromeos_update_engine::ErrorCode*,
77                                 chromeos_update_engine::InstallPlan*));
78 
79   MOCK_CONST_METHOD5(UpdateCanStart,
80                      EvalStatus(EvaluationContext*,
81                                 State*,
82                                 std::string*,
83                                 UpdateDownloadParams*,
84                                 UpdateState));
85 
86   MOCK_CONST_METHOD4(
87       UpdateDownloadAllowed,
88       EvalStatus(EvaluationContext*, State*, std::string*, bool*));
89 
90   MOCK_CONST_METHOD4(
91       P2PEnabled, EvalStatus(EvaluationContext*, State*, std::string*, bool*));
92 
93   MOCK_CONST_METHOD5(
94       P2PEnabledChanged,
95       EvalStatus(EvaluationContext*, State*, std::string*, bool*, bool));
96 
97  protected:
98   // Policy override.
99   std::string PolicyName() const override { return "MockPolicy"; }
100 
101  private:
102   DefaultPolicy default_policy_;
103 
104   DISALLOW_COPY_AND_ASSIGN(MockPolicy);
105 };
106 
107 }  // namespace chromeos_update_manager
108 
109 #endif  // UPDATE_ENGINE_UPDATE_MANAGER_MOCK_POLICY_H_
110