OLD | NEW |
---|---|
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ | 5 #ifndef CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ |
6 #define CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ | 6 #define CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ |
7 | 7 |
8 #include <queue> | |
8 #include <string> | 9 #include <string> |
9 | 10 |
10 #include "chromeos/dbus/update_engine_client.h" | 11 #include "chromeos/dbus/update_engine_client.h" |
11 | 12 |
12 namespace chromeos { | 13 namespace chromeos { |
13 | 14 |
14 // A fake implementation of UpdateEngineClient. The user of this class can | 15 // A fake implementation of UpdateEngineClient. The user of this class can |
15 // use set_update_engine_client_status() to set a fake last Status and | 16 // use set_update_engine_client_status() to set a fake last Status and |
16 // GetLastStatus() returns the fake with no modification. Other methods do | 17 // GetLastStatus() returns the fake with no modification. Other methods do |
17 // nothing. | 18 // nothing. |
18 class FakeUpdateEngineClient : public UpdateEngineClient { | 19 class FakeUpdateEngineClient : public UpdateEngineClient { |
19 public: | 20 public: |
20 FakeUpdateEngineClient(); | 21 FakeUpdateEngineClient(); |
21 virtual ~FakeUpdateEngineClient(); | 22 virtual ~FakeUpdateEngineClient(); |
22 | 23 |
23 // Overrides | 24 // Overrides |
24 virtual void AddObserver(Observer* observer) OVERRIDE; | 25 virtual void AddObserver(Observer* observer) OVERRIDE; |
25 virtual void RemoveObserver(Observer* observer) OVERRIDE; | 26 virtual void RemoveObserver(Observer* observer) OVERRIDE; |
26 virtual bool HasObserver(Observer* observer) OVERRIDE; | 27 virtual bool HasObserver(Observer* observer) OVERRIDE; |
27 virtual void RequestUpdateCheck(const UpdateCheckCallback& callback) OVERRIDE; | 28 virtual void RequestUpdateCheck(const UpdateCheckCallback& callback) OVERRIDE; |
28 virtual void RebootAfterUpdate() OVERRIDE; | 29 virtual void RebootAfterUpdate() OVERRIDE; |
29 virtual void SetReleaseTrack(const std::string& track) OVERRIDE; | 30 virtual void SetReleaseTrack(const std::string& track) OVERRIDE; |
30 virtual void GetReleaseTrack(const GetReleaseTrackCallback& callback) | 31 virtual void GetReleaseTrack(const GetReleaseTrackCallback& callback) |
31 OVERRIDE; | 32 OVERRIDE; |
32 virtual Status GetLastStatus() OVERRIDE; | 33 virtual Status GetLastStatus() OVERRIDE; |
33 | 34 |
34 void set_update_engine_client_status( | 35 // 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!
| |
35 const UpdateEngineClient::Status& status); | 36 // GetLastStatus() returns the status set by this method in FIFO order. |
37 // See set_default_status(). | |
38 void PushLastStatus(const UpdateEngineClient::Status& status) { | |
39 status_queue_.push(status); | |
40 } | |
41 | |
42 // Sets the default UpdateEngineClient::Status. GetLastStatus() returns the | |
43 // value set here if |status_queue_| is empty. | |
44 void set_default_status(const UpdateEngineClient::Status& status); | |
36 | 45 |
37 // Sets a value returned by RequestUpdateCheck(). | 46 // Sets a value returned by RequestUpdateCheck(). |
38 void set_update_check_result( | 47 void set_update_check_result( |
39 const UpdateEngineClient::UpdateCheckResult& result); | 48 const UpdateEngineClient::UpdateCheckResult& result); |
40 | 49 |
41 // Returns how many times RebootAfterUpdate() is called. | 50 // Returns how many times RebootAfterUpdate() is called. |
42 int reboot_after_update_call_count() { | 51 int reboot_after_update_call_count() { |
43 return reboot_after_update_call_count_; | 52 return reboot_after_update_call_count_; |
44 } | 53 } |
45 | 54 |
46 private: | 55 private: |
47 UpdateEngineClient::Status update_engine_client_status_; | 56 std::queue<UpdateEngineClient::Status> status_queue_; |
57 UpdateEngineClient::Status default_status_; | |
48 UpdateEngineClient::UpdateCheckResult update_check_result_; | 58 UpdateEngineClient::UpdateCheckResult update_check_result_; |
49 int reboot_after_update_call_count_; | 59 int reboot_after_update_call_count_; |
50 }; | 60 }; |
51 | 61 |
52 } // namespace chromeos | 62 } // namespace chromeos |
53 | 63 |
54 #endif // CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ | 64 #endif // CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_ |
OLD | NEW |