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