Home | History | Annotate | Download | only in update_engine 
      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:
     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.
     64   bool fake_to_more_stable_channel() const {
     65     return OmahaRequestParams::to_more_stable_channel();
     66   }
     67 
     68   std::string FakeGetAppId() const {
     69     return OmahaRequestParams::GetAppId();
     70   }
     71 
     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 
     80   void FakeUpdateDownloadChannel() {
     81     return OmahaRequestParams::UpdateDownloadChannel();
     82   }
     83 
     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