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: MockPolicy(chromeos_update_engine::ClockInterface * clock)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 MockPolicy()64 MockPolicy() : MockPolicy(nullptr) {} ~MockPolicy()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. PolicyName()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