Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(155)

Side by Side Diff: media/cast/rtp_sender/rtp_packetizer/rtp_packetizer_unittest.cc

Issue 112133002: Cast:Moving netwrok sender related code to a designated folder (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: moving const' to .cc Created 7 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
(Empty)
1 // Copyright 2013 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/rtp_sender/rtp_packetizer/rtp_packetizer.h"
6
7 #include "base/memory/scoped_ptr.h"
8 #include "base/test/simple_test_tick_clock.h"
9 #include "media/cast/cast_config.h"
10 #include "media/cast/pacing/paced_sender.h"
11 #include "media/cast/rtp_common/rtp_defines.h"
12 #include "media/cast/rtp_sender/packet_storage/packet_storage.h"
13 #include "media/cast/rtp_sender/rtp_packetizer/test/rtp_header_parser.h"
14 #include "testing/gmock/include/gmock/gmock.h"
15
16 namespace media {
17 namespace cast {
18
19 static const int kPayload = 127;
20 static const uint32 kTimestampMs = 10;
21 static const uint16 kSeqNum = 33;
22 static const int kMaxPacketLength = 1500;
23 static const int kSsrc = 0x12345;
24 static const unsigned int kFrameSize = 5000;
25 static const int kMaxPacketStorageTimeMs = 300;
26
27 class TestRtpPacketTransport : public PacedPacketSender {
28 public:
29 explicit TestRtpPacketTransport(RtpPacketizerConfig config)
30 : config_(config),
31 sequence_number_(kSeqNum),
32 packets_sent_(0),
33 expected_number_of_packets_(0),
34 expected_packet_id_(0),
35 expected_frame_id_(0) {}
36
37 void VerifyRtpHeader(const RtpCastHeader& rtp_header) {
38 VerifyCommonRtpHeader(rtp_header);
39 VerifyCastRtpHeader(rtp_header);
40 }
41
42 void VerifyCommonRtpHeader(const RtpCastHeader& rtp_header) {
43 EXPECT_EQ(expected_number_of_packets_ == packets_sent_,
44 rtp_header.webrtc.header.markerBit);
45 EXPECT_EQ(kPayload, rtp_header.webrtc.header.payloadType);
46 EXPECT_EQ(sequence_number_, rtp_header.webrtc.header.sequenceNumber);
47 EXPECT_EQ(kTimestampMs * 90, rtp_header.webrtc.header.timestamp);
48 EXPECT_EQ(config_.ssrc, rtp_header.webrtc.header.ssrc);
49 EXPECT_EQ(0, rtp_header.webrtc.header.numCSRCs);
50 }
51
52 void VerifyCastRtpHeader(const RtpCastHeader& rtp_header) {
53 EXPECT_FALSE(rtp_header.is_key_frame);
54 EXPECT_EQ(expected_frame_id_, rtp_header.frame_id);
55 EXPECT_EQ(expected_packet_id_, rtp_header.packet_id);
56 EXPECT_EQ(expected_number_of_packets_ - 1, rtp_header.max_packet_id);
57 EXPECT_TRUE(rtp_header.is_reference);
58 EXPECT_EQ(expected_frame_id_ - 1u, rtp_header.reference_frame_id);
59 }
60
61 virtual bool SendPackets(const PacketList& packets) OVERRIDE {
62 EXPECT_EQ(expected_number_of_packets_, static_cast<int>(packets.size()));
63 PacketList::const_iterator it = packets.begin();
64 for (; it != packets.end(); ++it) {
65 ++packets_sent_;
66 RtpHeaderParser parser(it->data(), it->size());
67 RtpCastHeader rtp_header;
68 parser.Parse(&rtp_header);
69 VerifyRtpHeader(rtp_header);
70 ++sequence_number_;
71 ++expected_packet_id_;
72 }
73 return true;
74 }
75
76 virtual bool ResendPackets(const PacketList& packets) OVERRIDE {
77 EXPECT_TRUE(false);
78 return false;
79 }
80
81 virtual bool SendRtcpPacket(const std::vector<uint8>& packet) OVERRIDE {
82 EXPECT_TRUE(false);
83 return false;
84 }
85
86 void SetExpectedNumberOfPackets(int num) {
87 expected_number_of_packets_ = num;
88 }
89
90 RtpPacketizerConfig config_;
91 uint32 sequence_number_;
92 int packets_sent_;
93 int expected_number_of_packets_;
94 // Assuming packets arrive in sequence.
95 int expected_packet_id_;
96 uint32 expected_frame_id_;
97 };
98
99 class RtpPacketizerTest : public ::testing::Test {
100 protected:
101 RtpPacketizerTest()
102 :video_frame_(),
103 packet_storage_(&testing_clock_, kMaxPacketStorageTimeMs) {
104 config_.sequence_number = kSeqNum;
105 config_.ssrc = kSsrc;
106 config_.payload_type = kPayload;
107 config_.max_payload_length = kMaxPacketLength;
108 transport_.reset(new TestRtpPacketTransport(config_));
109 rtp_packetizer_.reset(
110 new RtpPacketizer(transport_.get(), &packet_storage_, config_));
111 }
112
113 virtual ~RtpPacketizerTest() {}
114
115 virtual void SetUp() {
116 video_frame_.key_frame = false;
117 video_frame_.frame_id = 0;
118 video_frame_.last_referenced_frame_id = kStartFrameId;
119 video_frame_.data.assign(kFrameSize, 123);
120 }
121
122 base::SimpleTestTickClock testing_clock_;
123 scoped_ptr<RtpPacketizer> rtp_packetizer_;
124 RtpPacketizerConfig config_;
125 scoped_ptr<TestRtpPacketTransport> transport_;
126 EncodedVideoFrame video_frame_;
127 PacketStorage packet_storage_;
128 };
129
130 TEST_F(RtpPacketizerTest, SendStandardPackets) {
131 int expected_num_of_packets = kFrameSize / kMaxPacketLength + 1;
132 transport_->SetExpectedNumberOfPackets(expected_num_of_packets);
133
134 base::TimeTicks time;
135 time += base::TimeDelta::FromMilliseconds(kTimestampMs);
136 rtp_packetizer_->IncomingEncodedVideoFrame(&video_frame_, time);
137 }
138
139 TEST_F(RtpPacketizerTest, Stats) {
140 EXPECT_FALSE(rtp_packetizer_->send_packets_count());
141 EXPECT_FALSE(rtp_packetizer_->send_octet_count());
142 // Insert packets at varying lengths.
143 int expected_num_of_packets = kFrameSize / kMaxPacketLength + 1;
144 transport_->SetExpectedNumberOfPackets(expected_num_of_packets);
145
146 testing_clock_.Advance(base::TimeDelta::FromMilliseconds(kTimestampMs));
147 rtp_packetizer_->IncomingEncodedVideoFrame(&video_frame_,
148 testing_clock_.NowTicks());
149 EXPECT_EQ(expected_num_of_packets, rtp_packetizer_->send_packets_count());
150 EXPECT_EQ(kFrameSize, rtp_packetizer_->send_octet_count());
151 }
152
153 } // namespace cast
154 } // namespace media
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698