Chromium Code Reviews| 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 ccf25041be4164e05b85bb6816f477a7f66e90e6..f98f6461d79b5853d9d4e5d3bae7f781e98485ea 100644 |
| --- a/chromeos/dbus/fake_update_engine_client.h |
| +++ b/chromeos/dbus/fake_update_engine_client.h |
| @@ -5,6 +5,7 @@ |
| #ifndef CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ |
| #define CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ |
| +#include <queue> |
| #include <string> |
| #include "chromeos/dbus/update_engine_client.h" |
| @@ -31,8 +32,16 @@ class FakeUpdateEngineClient : public UpdateEngineClient { |
| OVERRIDE; |
| virtual Status GetLastStatus() OVERRIDE; |
| - void set_update_engine_client_status( |
| - const UpdateEngineClient::Status& status); |
| + // Push UpdateEngineClient::Status in the queue to test changing status. |
|
hashimoto
2013/05/14 14:47:54
nit: s/Push/Pushes/?
Haruki Sato
2013/05/15 02:01:08
Done. Thanks!
|
| + // GetLastStatus() returns the status set by this method in FIFO order. |
| + // See set_default_status(). |
| + void PushLastStatus(const UpdateEngineClient::Status& status) { |
| + status_queue_.push(status); |
| + } |
| + |
| + // 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( |
| @@ -44,7 +53,8 @@ class FakeUpdateEngineClient : public UpdateEngineClient { |
| } |
| private: |
| - UpdateEngineClient::Status update_engine_client_status_; |
| + std::queue<UpdateEngineClient::Status> status_queue_; |
| + UpdateEngineClient::Status default_status_; |
| UpdateEngineClient::UpdateCheckResult update_check_result_; |
| int reboot_after_update_call_count_; |
| }; |