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..c459bd7d3ea4834b69c3b746ae5949b95240b521 100644 |
--- a/chromeos/dbus/fake_update_engine_client.h |
+++ b/chromeos/dbus/fake_update_engine_client.h |
@@ -52,16 +52,31 @@ class FakeUpdateEngineClient : public UpdateEngineClient { |
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 |