src/chromeos/dbus/fake_update_engine_client.h

   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
  13 namespace 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.
  19 class 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 Status GetLastStatus() OVERRIDE;
  32   virtual void SetChannel(const std::string& target_channel,
  33                           bool is_powerwash_allowed) OVERRIDE;
  34   virtual void GetChannel(bool get_current_channel,
  35                           const GetChannelCallback& callback) OVERRIDE;
  36
  37   // Pushes UpdateEngineClient::Status in the queue to test changing status.
  38   // GetLastStatus() returns the status set by this method in FIFO order.
  39   // See set_default_status().
  40   void PushLastStatus(const UpdateEngineClient::Status& status) {
  41     status_queue_.push(status);
  42   }
  43
  44   // Sets the default UpdateEngineClient::Status. GetLastStatus() returns the
  45   // value set here if |status_queue_| is empty.
  46   void set_default_status(const UpdateEngineClient::Status& status);
  47
  48   // Sets a value returned by RequestUpdateCheck().
  49   void set_update_check_result(
  50       const UpdateEngineClient::UpdateCheckResult& result);
  51
  52   // Returns how many times RebootAfterUpdate() is called.
  53   int reboot_after_update_call_count() {
  54     return reboot_after_update_call_count_;
  55   }
  56
  57  private:
  58   std::queue<UpdateEngineClient::Status> status_queue_;
  59   UpdateEngineClient::Status default_status_;
  60   UpdateEngineClient::UpdateCheckResult update_check_result_;
  61   int reboot_after_update_call_count_;
  62 };
  63
  64 }  // namespace chromeos
  65
  66 #endif  // CHROMEOS_DBUS_FAKE_UPDATE_ENGINE_CLIENT_H_