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..b0328fbbd1868acc200b4486c0bd20b25236464d 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_; |
+ const int reboot_after_update_call_count() { |
oshima
2014/04/01 13:51:06
Sorry if I wasn't clear. I meant this should be co
|
+ return reboot_after_update_call_count_; |
} |
+ // Returns how many times Rollback() is called. |
+ const int rollback_call_count() { return rollback_call_count_; } |
+ |
+ // Returns how many times Rollback() is called. |
+ const 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 |