Browsertests for improved reset options: powerwash and rollback.

chromeos/dbus/fake_update_engine_client.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chromeos/dbus/fake_update_engine_client.h"
 
 namespace chromeos {
 
 FakeUpdateEngineClient::FakeUpdateEngineClient()
     : update_check_result_(UpdateEngineClient::UPDATE_RESULT_SUCCESS),
-      reboot_after_update_call_count_(0) {
+      reboot_after_update_call_count_(0),
+      rollback_call_count_(0),
+      can_rollback_call_count_(0) {
 }
 
 FakeUpdateEngineClient::~FakeUpdateEngineClient() {
 }
 
 void FakeUpdateEngineClient::Init(dbus::Bus* bus) {
 }
 
 void FakeUpdateEngineClient::AddObserver(Observer* observer) {
 }
 
 void FakeUpdateEngineClient::RemoveObserver(Observer* observer) {
 }
 
 bool FakeUpdateEngineClient::HasObserver(Observer* observer) {
   return false;
 }
 
 void FakeUpdateEngineClient::RequestUpdateCheck(
     const UpdateCheckCallback& callback) {
   callback.Run(update_check_result_);
 }
 
 void FakeUpdateEngineClient::Rollback() {
+  rollback_call_count_++;
 }
 
 void FakeUpdateEngineClient::CanRollbackCheck(
     const RollbackCheckCallback& callback) {
+  can_rollback_call_count_++;
+  callback.Run(can_rollback_stub_result);
 }
 
 void FakeUpdateEngineClient::RebootAfterUpdate() {
   reboot_after_update_call_count_++;
 }
 
 UpdateEngineClient::Status FakeUpdateEngineClient::GetLastStatus() {
   if (status_queue_.empty())
     return default_status_;
 
@@ skipping 13 matching lines @@
 void FakeUpdateEngineClient::set_default_status(
     const UpdateEngineClient::Status& status) {
   default_status_ = status;
 }
 
 void FakeUpdateEngineClient::set_update_check_result(
     const UpdateEngineClient::UpdateCheckResult& result) {
   update_check_result_ = result;
 }
 
+void FakeUpdateEngineClient::set_can_callback_check_result(bool result) {
+  can_rollback_stub_result = result;
+}
+
 }  // namespace chromeos