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 Rollback() is called.
69  int rollback_call_count() const { return rollback_call_count_; }
70
71  // Returns how many times Rollback() is called.
72  int can_rollback_call_count() const { return can_rollback_call_count_; }
73
74 private:
75  ObserverList<Observer> observers_;
76  std::queue<UpdateEngineClient::Status> status_queue_;
77  UpdateEngineClient::Status default_status_;
78  UpdateEngineClient::UpdateCheckResult update_check_result_;
79  bool can_rollback_stub_result_;
80  int reboot_after_update_call_count_;
81  int rollback_call_count_;
82  int can_rollback_call_count_;
83};
84
85}  // namespace chromeos
86
87#endif  // CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_
88