| // Copyright (c) 2014 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef UPDATE_ENGINE_UPDATE_MANAGER_MOCK_POLICY_H_ |
| #define UPDATE_ENGINE_UPDATE_MANAGER_MOCK_POLICY_H_ |
| |
| #include <string> |
| |
| #include <gmock/gmock.h> |
| |
| #include "update_engine/update_manager/policy.h" |
| |
| namespace chromeos_update_manager { |
| |
| // A mocked implementation of Policy. |
| class MockPolicy : public Policy { |
| public: |
| MockPolicy() {} |
| virtual ~MockPolicy() {} |
| |
| // Policy overrides. |
| MOCK_CONST_METHOD4(UpdateCheckAllowed, |
| EvalStatus(EvaluationContext*, State*, std::string*, |
| UpdateCheckParams*)); |
| |
| MOCK_CONST_METHOD6(UpdateCanStart, |
| EvalStatus(EvaluationContext*, State*, std::string*, |
| UpdateCanStartResult*, |
| const bool, const UpdateState&)); |
| |
| MOCK_CONST_METHOD4(UpdateCanStart, |
| EvalStatus(EvaluationContext*, State*, std::string*, |
| bool*)); |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(MockPolicy); |
| }; |
| |
| } // namespace chromeos_update_manager |
| |
| #endif // UPDATE_ENGINE_UPDATE_MANAGER_MOCK_POLICY_H_ |