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
3// found in the LICENSE file.
4
5#ifndef CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_
6#define CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_
7
8#include <queue>
9#include <string>
10
11#include "chromeos/dbus/update_engine_client.h"
12
13namespace chromeos {
14
15// A fake implementation of UpdateEngineClient. The user of this class can
16// use set_update_engine_client_status() to set a fake last Status and
17// GetLastStatus() returns the fake with no modification. Other methods do
18// nothing.
19class FakeUpdateEngineClient : public UpdateEngineClient {
20 public:
21  FakeUpdateEngineClient();
22  virtual ~FakeUpdateEngineClient();
23
24  // UpdateEngineClient overrides
25  virtual void Init(dbus::Bus* bus) OVERRIDE;
26  virtual void AddObserver(Observer* observer) OVERRIDE;
27  virtual void RemoveObserver(Observer* observer) OVERRIDE;
28  virtual bool HasObserver(Observer* observer) OVERRIDE;
29  virtual void RequestUpdateCheck(const UpdateCheckCallback& callback) OVERRIDE;
30  virtual void RebootAfterUpdate() OVERRIDE;
31  virtual void Rollback() OVERRIDE;
32  virtual void CanRollbackCheck(
33      const RollbackCheckCallback& callback) OVERRIDE;
34  virtual Status GetLastStatus() OVERRIDE;
35  virtual void SetChannel(const std::string& target_channel,
36                          bool is_powerwash_allowed) OVERRIDE;
37  virtual void GetChannel(bool get_current_channel,
38                          const GetChannelCallback& callback) OVERRIDE;
39
40  // Pushes UpdateEngineClient::Status in the queue to test changing status.
41  // GetLastStatus() returns the status set by this method in FIFO order.
42  // See set_default_status().
43  void PushLastStatus(const UpdateEngineClient::Status& status) {
44    status_queue_.push(status);
45  }
46
47  // Sends status change notification.
48  void NotifyObserversThatStatusChanged(
49      const UpdateEngineClient::Status& status);
50
51  // Sets the default UpdateEngineClient::Status. GetLastStatus() returns the
52  // value set here if |status_queue_| is empty.
53  void set_default_status(const UpdateEngineClient::Status& status);
54
55  // Sets a value returned by RequestUpdateCheck().
56  void set_update_check_result(
57      const UpdateEngineClient::UpdateCheckResult& result);
58
59  void set_can_rollback_check_result(bool result) {
60      can_rollback_stub_result_ = result;
61  }
62
63  // Returns how many times RebootAfterUpdate() is called.
64  int reboot_after_update_call_count() const {
65      return reboot_after_update_call_count_;
66  }
67
68  // Returns how many times RequestUpdateCheck() is called.
69  int request_update_check_call_count() const {
70    return request_update_check_call_count_;
71  }
72
73  // Returns how many times Rollback() is called.
74  int rollback_call_count() const { return rollback_call_count_; }
75
76  // Returns how many times Rollback() is called.
77  int can_rollback_call_count() const { return can_rollback_call_count_; }
78
79 private:
80  ObserverList<Observer> observers_;
81  std::queue<UpdateEngineClient::Status> status_queue_;
82  UpdateEngineClient::Status default_status_;
83  UpdateEngineClient::UpdateCheckResult update_check_result_;
84  bool can_rollback_stub_result_;
85  int reboot_after_update_call_count_;
86  int request_update_check_call_count_;
87  int rollback_call_count_;
88  int can_rollback_call_count_;
89};
90
91}  // namespace chromeos
92
93#endif  // CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_
94