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_MOCK_OMAHA_REQUEST_PARAMS_H_ 18 #define UPDATE_ENGINE_MOCK_OMAHA_REQUEST_PARAMS_H_ 19 20 #include <string> 21 22 #include <gmock/gmock.h> 23 24 #include "update_engine/omaha_request_params.h" 25 26 namespace chromeos_update_engine { 27 28 class MockOmahaRequestParams : public OmahaRequestParams { 29 public: MockOmahaRequestParams(SystemState * system_state)30 explicit MockOmahaRequestParams(SystemState* system_state) 31 : OmahaRequestParams(system_state) { 32 // Delegate all calls to the parent instance by default. This helps the 33 // migration from tests using the real RequestParams when they should have 34 // use a fake or mock. 35 ON_CALL(*this, to_more_stable_channel()) 36 .WillByDefault(testing::Invoke( 37 this, &MockOmahaRequestParams::fake_to_more_stable_channel)); 38 ON_CALL(*this, GetAppId()) 39 .WillByDefault(testing::Invoke( 40 this, &MockOmahaRequestParams::FakeGetAppId)); 41 ON_CALL(*this, SetTargetChannel(testing::_, testing::_, testing::_)) 42 .WillByDefault(testing::Invoke( 43 this, &MockOmahaRequestParams::FakeSetTargetChannel)); 44 ON_CALL(*this, UpdateDownloadChannel()) 45 .WillByDefault(testing::Invoke( 46 this, &MockOmahaRequestParams::FakeUpdateDownloadChannel)); 47 ON_CALL(*this, is_powerwash_allowed()) 48 .WillByDefault(testing::Invoke( 49 this, &MockOmahaRequestParams::fake_is_powerwash_allowed)); 50 } 51 52 MOCK_CONST_METHOD0(to_more_stable_channel, bool(void)); 53 MOCK_CONST_METHOD0(GetAppId, std::string(void)); 54 MOCK_METHOD3(SetTargetChannel, bool(const std::string& channel, 55 bool is_powerwash_allowed, 56 std::string* error)); 57 MOCK_METHOD0(UpdateDownloadChannel, void(void)); 58 MOCK_CONST_METHOD0(is_powerwash_allowed, bool(void)); 59 MOCK_CONST_METHOD0(IsUpdateUrlOfficial, bool(void)); 60 61 private: 62 // Wrappers to call the parent class and behave like the real object by 63 // default. See "Delegating Calls to a Parent Class" in gmock's documentation. fake_to_more_stable_channel()64 bool fake_to_more_stable_channel() const { 65 return OmahaRequestParams::to_more_stable_channel(); 66 } 67 FakeGetAppId()68 std::string FakeGetAppId() const { 69 return OmahaRequestParams::GetAppId(); 70 } 71 FakeSetTargetChannel(const std::string & channel,bool is_powerwash_allowed,std::string * error)72 bool FakeSetTargetChannel(const std::string& channel, 73 bool is_powerwash_allowed, 74 std::string* error) { 75 return OmahaRequestParams::SetTargetChannel(channel, 76 is_powerwash_allowed, 77 error); 78 } 79 FakeUpdateDownloadChannel()80 void FakeUpdateDownloadChannel() { 81 return OmahaRequestParams::UpdateDownloadChannel(); 82 } 83 fake_is_powerwash_allowed()84 bool fake_is_powerwash_allowed() const { 85 return OmahaRequestParams::is_powerwash_allowed(); 86 } 87 }; 88 89 } // namespace chromeos_update_engine 90 91 #endif // UPDATE_ENGINE_MOCK_OMAHA_REQUEST_PARAMS_H_ 92