media/cast/test/loopback_transport.h

   1 // Copyright 2014 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 MEDIA_CAST_TEST_LOOPBACK_TRANSPORT_H_
   6 #define MEDIA_CAST_TEST_LOOPBACK_TRANSPORT_H_
   7
   8 #include "base/memory/ref_counted.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "media/cast/cast_environment.h"
  11 #include "media/cast/net/cast_transport_config.h"
  12
  13 namespace base {
  14 class SingleThreadTaskRunner;
  15 class TickClock;
  16 }  // namespace base
  17
  18 namespace media {
  19 namespace cast {
  20
  21 namespace test {
  22 class PacketPipe;
  23 }  // namespace test
  24
  25 // Class that sends the packet to a receiver through a stack of PacketPipes.
  26 class LoopBackTransport : public PacketSender {
  27  public:
  28   explicit LoopBackTransport(
  29       scoped_refptr<CastEnvironment> cast_environment);
  30   ~LoopBackTransport() override;
  31
  32   bool SendPacket(PacketRef packet, const base::Closure& cb) override;
  33
  34   int64 GetBytesSent() override;
  35
  36   // Initiailize this loopback transport.
  37   // Establish a flow of packets from |pipe| to |packet_receiver|.
  38   // The data flow looks like:
  39   // SendPacket() -> |pipe| -> Fake loopback pipe -> |packet_receiver|.
  40   void Initialize(
  41       scoped_ptr<test::PacketPipe> pipe,
  42       const PacketReceiverCallback& packet_receiver,
  43       const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
  44       base::TickClock* clock);
  45
  46  private:
  47   const scoped_refptr<CastEnvironment> cast_environment_;
  48   scoped_ptr<test::PacketPipe> packet_pipe_;
  49   int64 bytes_sent_;
  50
  51   DISALLOW_COPY_AND_ASSIGN(LoopBackTransport);
  52 };
  53
  54 }  // namespace cast
  55 }  // namespace media
  56
  57 #endif  // MEDIA_CAST_TEST_LOOPBACK_TRANSPORT_H_