src/media/cast/net/rtp/rtp_packetizer_unittest.cc

   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 #include "media/cast/net/rtp/rtp_packetizer.h"
   6
   7 #include <stdint.h>
   8
   9 #include "base/memory/scoped_ptr.h"
  10 #include "base/test/simple_test_tick_clock.h"
  11 #include "media/cast/logging/logging_impl.h"
  12 #include "media/cast/logging/simple_event_subscriber.h"
  13 #include "media/cast/net/pacing/paced_sender.h"
  14 #include "media/cast/net/rtp/packet_storage.h"
  15 #include "media/cast/net/rtp/rtp_header_parser.h"
  16 #include "media/cast/test/fake_single_thread_task_runner.h"
  17 #include "testing/gmock/include/gmock/gmock.h"
  18
  19 namespace media {
  20 namespace cast {
  21
  22 namespace {
  23 static const int kPayload = 127;
  24 static const uint32 kTimestampMs = 10;
  25 static const uint16 kSeqNum = 33;
  26 static const int kMaxPacketLength = 1500;
  27 static const int kSsrc = 0x12345;
  28 static const unsigned int kFrameSize = 5000;
  29 }
  30
  31 class TestRtpPacketTransport : public PacketSender {
  32  public:
  33   explicit TestRtpPacketTransport(RtpPacketizerConfig config)
  34       : config_(config),
  35         sequence_number_(kSeqNum),
  36         packets_sent_(0),
  37         expected_number_of_packets_(0),
  38         expected_packet_id_(0),
  39         expected_frame_id_(0),
  40         expectd_rtp_timestamp_(0) {}
  41
  42   void VerifyRtpHeader(const RtpCastTestHeader& rtp_header) {
  43     VerifyCommonRtpHeader(rtp_header);
  44     VerifyCastRtpHeader(rtp_header);
  45   }
  46
  47   void VerifyCommonRtpHeader(const RtpCastTestHeader& rtp_header) {
  48     EXPECT_EQ(kPayload, rtp_header.payload_type);
  49     EXPECT_EQ(sequence_number_, rtp_header.sequence_number);
  50     EXPECT_EQ(expectd_rtp_timestamp_, rtp_header.rtp_timestamp);
  51     EXPECT_EQ(config_.ssrc, rtp_header.ssrc);
  52     EXPECT_EQ(0, rtp_header.num_csrcs);
  53   }
  54
  55   void VerifyCastRtpHeader(const RtpCastTestHeader& rtp_header) {
  56     EXPECT_FALSE(rtp_header.is_key_frame);
  57     EXPECT_EQ(expected_frame_id_, rtp_header.frame_id);
  58     EXPECT_EQ(expected_packet_id_, rtp_header.packet_id);
  59     EXPECT_EQ(expected_number_of_packets_ - 1, rtp_header.max_packet_id);
  60     EXPECT_TRUE(rtp_header.is_reference);
  61     EXPECT_EQ(expected_frame_id_ - 1u, rtp_header.reference_frame_id);
  62   }
  63
  64   virtual bool SendPacket(PacketRef packet, const base::Closure& cb) OVERRIDE {
  65     ++packets_sent_;
  66     RtpHeaderParser parser(&packet->data[0], packet->data.size());
  67     RtpCastTestHeader rtp_header;
  68     parser.Parse(&rtp_header);
  69     VerifyRtpHeader(rtp_header);
  70     ++sequence_number_;
  71     ++expected_packet_id_;
  72     return true;
  73   }
  74
  75   virtual int64 GetBytesSent() OVERRIDE {
  76     return 0;
  77   }
  78
  79   size_t number_of_packets_received() const { return packets_sent_; }
  80
  81   void set_expected_number_of_packets(size_t expected_number_of_packets) {
  82     expected_number_of_packets_ = expected_number_of_packets;
  83   }
  84
  85   void set_rtp_timestamp(uint32 rtp_timestamp) {
  86     expectd_rtp_timestamp_ = rtp_timestamp;
  87   }
  88
  89   RtpPacketizerConfig config_;
  90   uint32 sequence_number_;
  91   size_t packets_sent_;
  92   size_t number_of_packets_;
  93   size_t expected_number_of_packets_;
  94   // Assuming packets arrive in sequence.
  95   int expected_packet_id_;
  96   uint32 expected_frame_id_;
  97   uint32 expectd_rtp_timestamp_;
  98
  99   DISALLOW_COPY_AND_ASSIGN(TestRtpPacketTransport);
 100 };
 101
 102 class RtpPacketizerTest : public ::testing::Test {
 103  protected:
 104   RtpPacketizerTest()
 105       : task_runner_(new test::FakeSingleThreadTaskRunner(&testing_clock_)) {
 106     config_.sequence_number = kSeqNum;
 107     config_.ssrc = kSsrc;
 108     config_.payload_type = kPayload;
 109     config_.max_payload_length = kMaxPacketLength;
 110     transport_.reset(new TestRtpPacketTransport(config_));
 111     pacer_.reset(new PacedSender(kTargetBurstSize,
 112                                  kMaxBurstSize,
 113                                  &testing_clock_,
 114                                  &logging_,
 115                                  transport_.get(),
 116                                  task_runner_));
 117     pacer_->RegisterVideoSsrc(config_.ssrc);
 118     rtp_packetizer_.reset(new RtpPacketizer(
 119         pacer_.get(), &packet_storage_, config_));
 120     video_frame_.dependency = EncodedFrame::DEPENDENT;
 121     video_frame_.frame_id = 0;
 122     video_frame_.referenced_frame_id = kStartFrameId;
 123     video_frame_.data.assign(kFrameSize, 123);
 124     video_frame_.rtp_timestamp =
 125         GetVideoRtpTimestamp(testing_clock_.NowTicks());
 126   }
 127
 128   void RunTasks(int during_ms) {
 129     for (int i = 0; i < during_ms; ++i) {
 130       // Call process the timers every 1 ms.
 131       testing_clock_.Advance(base::TimeDelta::FromMilliseconds(1));
 132       task_runner_->RunTasks();
 133     }
 134   }
 135
 136   base::SimpleTestTickClock testing_clock_;
 137   scoped_refptr<test::FakeSingleThreadTaskRunner> task_runner_;
 138   EncodedFrame video_frame_;
 139   PacketStorage packet_storage_;
 140   RtpPacketizerConfig config_;
 141   scoped_ptr<TestRtpPacketTransport> transport_;
 142   LoggingImpl logging_;
 143   scoped_ptr<PacedSender> pacer_;
 144   scoped_ptr<RtpPacketizer> rtp_packetizer_;
 145
 146   DISALLOW_COPY_AND_ASSIGN(RtpPacketizerTest);
 147 };
 148
 149 TEST_F(RtpPacketizerTest, SendStandardPackets) {
 150   size_t expected_num_of_packets = kFrameSize / kMaxPacketLength + 1;
 151   transport_->set_expected_number_of_packets(expected_num_of_packets);
 152   transport_->set_rtp_timestamp(video_frame_.rtp_timestamp);
 153
 154   testing_clock_.Advance(base::TimeDelta::FromMilliseconds(kTimestampMs));
 155   video_frame_.reference_time = testing_clock_.NowTicks();
 156   rtp_packetizer_->SendFrameAsPackets(video_frame_);
 157   RunTasks(33 + 1);
 158   EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received());
 159 }
 160
 161 TEST_F(RtpPacketizerTest, Stats) {
 162   EXPECT_FALSE(rtp_packetizer_->send_packet_count());
 163   EXPECT_FALSE(rtp_packetizer_->send_octet_count());
 164   // Insert packets at varying lengths.
 165   size_t expected_num_of_packets = kFrameSize / kMaxPacketLength + 1;
 166   transport_->set_expected_number_of_packets(expected_num_of_packets);
 167   transport_->set_rtp_timestamp(video_frame_.rtp_timestamp);
 168
 169   testing_clock_.Advance(base::TimeDelta::FromMilliseconds(kTimestampMs));
 170   video_frame_.reference_time = testing_clock_.NowTicks();
 171   rtp_packetizer_->SendFrameAsPackets(video_frame_);
 172   RunTasks(33 + 1);
 173   EXPECT_EQ(expected_num_of_packets, rtp_packetizer_->send_packet_count());
 174   EXPECT_EQ(kFrameSize, rtp_packetizer_->send_octet_count());
 175   EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received());
 176 }
 177
 178 }  // namespace cast
 179 }  // namespace media