Browsertests for improved reset options: powerwash and rollback.

chromeos/dbus/fake_update_engine_client.cc

Index: chromeos/dbus/fake_update_engine_client.cc
diff --git a/chromeos/dbus/fake_update_engine_client.cc b/chromeos/dbus/fake_update_engine_client.cc
index b87b798b04a13f7073784d53a14d8d838dcf4e84..b523f82edf29d48da587c674bc40f5030bb1cfe0 100644
--- a/chromeos/dbus/fake_update_engine_client.cc
+++ b/chromeos/dbus/fake_update_engine_client.cc
@@ -8,7 +8,9 @@ namespace chromeos {
 
 FakeUpdateEngineClient::FakeUpdateEngineClient()
     : update_check_result_(UpdateEngineClient::UPDATE_RESULT_SUCCESS),
-      reboot_after_update_call_count_(0) {
+      reboot_after_update_call_count_(0),

[oshima, 2014/03/31 20:41:54]

initialize can_rollback_stub_result_

[merkulova, 2014/04/01 11:43:20]

Done.

+      rollback_call_count_(0),
+      can_rollback_call_count_(0) {
 }
 
 FakeUpdateEngineClient::~FakeUpdateEngineClient() {
@@ -33,10 +35,13 @@ void FakeUpdateEngineClient::RequestUpdateCheck(
 }
 
 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() {
@@ -70,4 +75,8 @@ void FakeUpdateEngineClient::set_update_check_result(
   update_check_result_ = result;
 }
 
+void FakeUpdateEngineClient::set_can_callback_check_result(bool result) {
+  can_rollback_stub_result = result;
+}
+
 }  // namespace chromeos