// Copyright 2019 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 REMOTING_TEST_FAKE_WEBRTC_CONNECTION_H_
#define REMOTING_TEST_FAKE_WEBRTC_CONNECTION_H_

#include "base/callback_forward.h"
#include "base/macros.h"
#include "base/memory/scoped_refptr.h"
#include "remoting/protocol/webrtc_transport.h"

namespace remoting {
namespace test {

class FakeWebrtcConnection final
    : public protocol::WebrtcTransport::EventHandler {
 public:
  FakeWebrtcConnection(
      scoped_refptr<protocol::TransportContext> transport_context,
      base::OnceClosure on_closed);
  ~FakeWebrtcConnection() override;

  protocol::Transport* transport() { return transport_.get(); }

 private:
  // protocol::WebrtcTransport::EventHandler implementations.
  void OnWebrtcTransportConnecting() override;
  void OnWebrtcTransportConnected() override;
  void OnWebrtcTransportError(protocol::ErrorCode error) override;
  void OnWebrtcTransportIncomingDataChannel(
      const std::string& name,
      std::unique_ptr<protocol::MessagePipe> pipe) override;
  void OnWebrtcTransportMediaStreamAdded(
      scoped_refptr<webrtc::MediaStreamInterface> stream) override;
  void OnWebrtcTransportMediaStreamRemoved(
      scoped_refptr<webrtc::MediaStreamInterface> stream) override;

  std::unique_ptr<protocol::WebrtcTransport> transport_;
  base::OnceClosure on_closed_;
  DISALLOW_COPY_AND_ASSIGN(FakeWebrtcConnection);
};

}  // namespace test
}  // namespace remoting

#endif  // REMOTING_TEST_FAKE_WEBRTC_CONNECTION_H_