| Index: chromeos/dbus/fake_update_engine_client.h
|
| diff --git a/chromeos/dbus/fake_update_engine_client.h b/chromeos/dbus/fake_update_engine_client.h
|
| index 8768e31c360c14572a598e6d522687802246b34d..146db8b6c8d9e95aa93b94352906cdf7171f2146 100644
|
| --- a/chromeos/dbus/fake_update_engine_client.h
|
| +++ b/chromeos/dbus/fake_update_engine_client.h
|
| @@ -52,16 +52,29 @@ class FakeUpdateEngineClient : public UpdateEngineClient {
|
| void set_update_check_result(
|
| const UpdateEngineClient::UpdateCheckResult& result);
|
|
|
| + void set_can_rollback_check_result(bool result) {
|
| + can_rollback_stub_result_ = result;
|
| + }
|
| +
|
| // Returns how many times RebootAfterUpdate() is called.
|
| - int reboot_after_update_call_count() {
|
| - return reboot_after_update_call_count_;
|
| + int reboot_after_update_call_count() const {
|
| + return reboot_after_update_call_count_;
|
| }
|
|
|
| + // Returns how many times Rollback() is called.
|
| + int rollback_call_count() const { return rollback_call_count_; }
|
| +
|
| + // Returns how many times Rollback() is called.
|
| + int can_rollback_call_count() const { return can_rollback_call_count_; }
|
| +
|
| private:
|
| std::queue<UpdateEngineClient::Status> status_queue_;
|
| UpdateEngineClient::Status default_status_;
|
| UpdateEngineClient::UpdateCheckResult update_check_result_;
|
| + bool can_rollback_stub_result_;
|
| int reboot_after_update_call_count_;
|
| + int rollback_call_count_;
|
| + int can_rollback_call_count_;
|
| };
|
|
|
| } // namespace chromeos
|
|
|