OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "media/cast/net/rtp/frame_buffer.h" | 5 #include "media/cast/net/rtp/frame_buffer.h" |
6 #include "media/cast/net/cast_transport_defines.h" | 6 #include "media/cast/net/cast_transport_defines.h" |
7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
8 | 8 |
9 namespace media { | 9 namespace media { |
10 namespace cast { | 10 namespace cast { |
(...skipping 11 matching lines...) Expand all Loading... |
22 RtpCastHeader rtp_header_; | 22 RtpCastHeader rtp_header_; |
23 | 23 |
24 DISALLOW_COPY_AND_ASSIGN(FrameBufferTest); | 24 DISALLOW_COPY_AND_ASSIGN(FrameBufferTest); |
25 }; | 25 }; |
26 | 26 |
27 TEST_F(FrameBufferTest, OnePacketInsertSanity) { | 27 TEST_F(FrameBufferTest, OnePacketInsertSanity) { |
28 rtp_header_.rtp_timestamp = 3000; | 28 rtp_header_.rtp_timestamp = 3000; |
29 rtp_header_.is_key_frame = true; | 29 rtp_header_.is_key_frame = true; |
30 rtp_header_.frame_id = 5; | 30 rtp_header_.frame_id = 5; |
31 rtp_header_.reference_frame_id = 5; | 31 rtp_header_.reference_frame_id = 5; |
32 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 32 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
33 EncodedFrame frame; | 33 EncodedFrame frame; |
34 EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame)); | 34 EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame)); |
35 EXPECT_EQ(EncodedFrame::KEY, frame.dependency); | 35 EXPECT_EQ(EncodedFrame::KEY, frame.dependency); |
36 EXPECT_EQ(5u, frame.frame_id); | 36 EXPECT_EQ(5u, frame.frame_id); |
37 EXPECT_EQ(5u, frame.referenced_frame_id); | 37 EXPECT_EQ(5u, frame.referenced_frame_id); |
38 EXPECT_EQ(3000u, frame.rtp_timestamp); | 38 EXPECT_EQ(3000u, frame.rtp_timestamp); |
39 } | 39 } |
40 | 40 |
41 TEST_F(FrameBufferTest, EmptyBuffer) { | 41 TEST_F(FrameBufferTest, EmptyBuffer) { |
42 EXPECT_FALSE(buffer_.Complete()); | 42 EXPECT_FALSE(buffer_.Complete()); |
43 EncodedFrame frame; | 43 EncodedFrame frame; |
44 EXPECT_FALSE(buffer_.AssembleEncodedFrame(&frame)); | 44 EXPECT_FALSE(buffer_.AssembleEncodedFrame(&frame)); |
45 } | 45 } |
46 | 46 |
47 TEST_F(FrameBufferTest, DefaultOnePacketFrame) { | 47 TEST_F(FrameBufferTest, DefaultOnePacketFrame) { |
48 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 48 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
49 EXPECT_TRUE(buffer_.Complete()); | 49 EXPECT_TRUE(buffer_.Complete()); |
50 EXPECT_FALSE(buffer_.is_key_frame()); | 50 EXPECT_FALSE(buffer_.is_key_frame()); |
51 EncodedFrame frame; | 51 EncodedFrame frame; |
52 EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame)); | 52 EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame)); |
53 EXPECT_EQ(payload_.size(), frame.data.size()); | 53 EXPECT_EQ(payload_.size(), frame.data.size()); |
54 } | 54 } |
55 | 55 |
56 TEST_F(FrameBufferTest, MultiplePacketFrame) { | 56 TEST_F(FrameBufferTest, MultiplePacketFrame) { |
57 rtp_header_.is_key_frame = true; | 57 rtp_header_.is_key_frame = true; |
58 rtp_header_.max_packet_id = 2; | 58 rtp_header_.max_packet_id = 2; |
59 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 59 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
60 ++rtp_header_.packet_id; | 60 ++rtp_header_.packet_id; |
61 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 61 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
62 ++rtp_header_.packet_id; | 62 ++rtp_header_.packet_id; |
63 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 63 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
64 ++rtp_header_.packet_id; | 64 ++rtp_header_.packet_id; |
65 EXPECT_TRUE(buffer_.Complete()); | 65 EXPECT_TRUE(buffer_.Complete()); |
66 EXPECT_TRUE(buffer_.is_key_frame()); | 66 EXPECT_TRUE(buffer_.is_key_frame()); |
67 EncodedFrame frame; | 67 EncodedFrame frame; |
68 EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame)); | 68 EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame)); |
69 EXPECT_EQ(3 * payload_.size(), frame.data.size()); | 69 EXPECT_EQ(3 * payload_.size(), frame.data.size()); |
70 } | 70 } |
71 | 71 |
72 TEST_F(FrameBufferTest, IncompleteFrame) { | 72 TEST_F(FrameBufferTest, IncompleteFrame) { |
73 rtp_header_.max_packet_id = 4; | 73 rtp_header_.max_packet_id = 4; |
74 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 74 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
75 ++rtp_header_.packet_id; | 75 ++rtp_header_.packet_id; |
76 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 76 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
77 ++rtp_header_.packet_id; | 77 ++rtp_header_.packet_id; |
78 // Increment again - skip packet #2. | 78 // Increment again - skip packet #2. |
79 ++rtp_header_.packet_id; | 79 ++rtp_header_.packet_id; |
80 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 80 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
81 ++rtp_header_.packet_id; | 81 ++rtp_header_.packet_id; |
82 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 82 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
83 EXPECT_FALSE(buffer_.Complete()); | 83 EXPECT_FALSE(buffer_.Complete()); |
84 // Insert missing packet. | 84 // Insert missing packet. |
85 rtp_header_.packet_id = 2; | 85 rtp_header_.packet_id = 2; |
86 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 86 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
87 EXPECT_TRUE(buffer_.Complete()); | 87 EXPECT_TRUE(buffer_.Complete()); |
88 } | 88 } |
89 | 89 |
90 } // namespace media | 90 } // namespace media |
91 } // namespace cast | 91 } // namespace cast |
OLD | NEW |