1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ 6 #define CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ 7 8 #include <queue> 9 #include <string> 10 11 #include "chromeos/dbus/update_engine_client.h" 12 13 namespace chromeos { 14 15 // A fake implementation of UpdateEngineClient. The user of this class can 16 // use set_update_engine_client_status() to set a fake last Status and 17 // GetLastStatus() returns the fake with no modification. Other methods do 18 // nothing. 19 class FakeUpdateEngineClient : public UpdateEngineClient { 20 public: 21 FakeUpdateEngineClient(); 22 virtual ~FakeUpdateEngineClient(); 23 24 // UpdateEngineClient overrides 25 virtual void Init(dbus::Bus* bus) OVERRIDE; 26 virtual void AddObserver(Observer* observer) OVERRIDE; 27 virtual void RemoveObserver(Observer* observer) OVERRIDE; 28 virtual bool HasObserver(Observer* observer) OVERRIDE; 29 virtual void RequestUpdateCheck(const UpdateCheckCallback& callback) OVERRIDE; 30 virtual void RebootAfterUpdate() OVERRIDE; 31 virtual void Rollback() OVERRIDE; 32 virtual void CanRollbackCheck( 33 const RollbackCheckCallback& callback) OVERRIDE; 34 virtual Status GetLastStatus() OVERRIDE; 35 virtual void SetChannel(const std::string& target_channel, 36 bool is_powerwash_allowed) OVERRIDE; 37 virtual void GetChannel(bool get_current_channel, 38 const GetChannelCallback& callback) OVERRIDE; 39 40 // Pushes UpdateEngineClient::Status in the queue to test changing status. 41 // GetLastStatus() returns the status set by this method in FIFO order. 42 // See set_default_status(). 43 void PushLastStatus(const UpdateEngineClient::Status& status) { 44 status_queue_.push(status); 45 } 46 47 // Sends status change notification. 48 void NotifyObserversThatStatusChanged( 49 const UpdateEngineClient::Status& status); 50 51 // Sets the default UpdateEngineClient::Status. GetLastStatus() returns the 52 // value set here if |status_queue_| is empty. 53 void set_default_status(const UpdateEngineClient::Status& status); 54 55 // Sets a value returned by RequestUpdateCheck(). 56 void set_update_check_result( 57 const UpdateEngineClient::UpdateCheckResult& result); 58 59 void set_can_rollback_check_result(bool result) { 60 can_rollback_stub_result_ = result; 61 } 62 63 // Returns how many times RebootAfterUpdate() is called. 64 int reboot_after_update_call_count() const { 65 return reboot_after_update_call_count_; 66 } 67 68 // Returns how many times Rollback() is called. 69 int rollback_call_count() const { return rollback_call_count_; } 70 71 // Returns how many times Rollback() is called. 72 int can_rollback_call_count() const { return can_rollback_call_count_; } 73 74 private: 75 ObserverList<Observer> observers_; 76 std::queue<UpdateEngineClient::Status> status_queue_; 77 UpdateEngineClient::Status default_status_; 78 UpdateEngineClient::UpdateCheckResult update_check_result_; 79 bool can_rollback_stub_result_; 80 int reboot_after_update_call_count_; 81 int rollback_call_count_; 82 int can_rollback_call_count_; 83 }; 84 85 } // namespace chromeos 86 87 #endif // CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ 88