chromeos/components/tether/fake_tether_component.h

// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROMEOS_COMPONENTS_TETHER_FAKE_TETHER_COMPONENT_H_
#define CHROMEOS_COMPONENTS_TETHER_FAKE_TETHER_COMPONENT_H_

#include "base/macros.h"
#include "chromeos/components/tether/tether_component.h"
#include "chromeos/components/tether/tether_disconnector.h"

namespace chromeos {

namespace tether {

// Test double for TetherComponent.
class FakeTetherComponent : public TetherComponent {
 public:
  explicit FakeTetherComponent(bool has_asynchronous_shutdown);
  ~FakeTetherComponent() override;

  void set_has_asynchronous_shutdown(bool has_asynchronous_shutdown) {
    has_asynchronous_shutdown_ = has_asynchronous_shutdown;
  }

  ShutdownReason* last_shutdown_reason() { return last_shutdown_reason_.get(); }

  // Should only be called when status() == SHUTTING_DOWN.
  void FinishAsynchronousShutdown();

  // Initializer:
  void RequestShutdown(const ShutdownReason& shutdown_reason) override;

 private:
  bool has_asynchronous_shutdown_;
  std::unique_ptr<ShutdownReason> last_shutdown_reason_;

  DISALLOW_COPY_AND_ASSIGN(FakeTetherComponent);
};

}  // namespace tether

}  // namespace chromeos

#endif  // CHROMEOS_COMPONENTS_TETHER_FAKE_TETHER_COMPONENT_H_