Browsertests for improved reset options: powerwash and rollback.

chromeos/dbus/fake_update_engine_client.h

 // 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.
 
 #ifndef CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_
 #define CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_
 
 #include <queue>
 #include <string>
 
@@ skipping 34 matching lines @@
   }
 
   // Sets the default UpdateEngineClient::Status. GetLastStatus() returns the
   // value set here if |status_queue_| is empty.
   void set_default_status(const UpdateEngineClient::Status& status);
 
   // Sets a value returned by RequestUpdateCheck().
   void set_update_check_result(
       const UpdateEngineClient::UpdateCheckResult& result);
 
+  void set_can_callback_check_result(bool result);
+
   // Returns how many times RebootAfterUpdate() is called.
   int reboot_after_update_call_count() {
     return reboot_after_update_call_count_;
   }
 
+  // Returns how many times Rollback() is called.
+  int rollback_call_count() {

[ygorshenin1, 2014/03/31 13:57:57]

Both methods can be folded to a single line.

[merkulova, 2014/03/31 15:33:58]

Done.

+    return rollback_call_count_;
+  }
+
+  // Returns how many times Rollback() is called.
+  int can_rollback_call_count() {
+    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
 
 #endif  // CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_