| 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(5u, frame.frame_id); | 34 EXPECT_EQ(5u, frame.frame_id); |
| 35 EXPECT_TRUE(frame.key_frame); | 35 EXPECT_EQ(5u, frame.base_frame_id); |
| 36 EXPECT_EQ(3000u, frame.rtp_timestamp); | 36 EXPECT_EQ(3000u, frame.rtp_timestamp); |
| 37 } | 37 } |
| 38 | 38 |
| 39 TEST_F(FrameBufferTest, EmptyBuffer) { | 39 TEST_F(FrameBufferTest, EmptyBuffer) { |
| 40 EXPECT_FALSE(buffer_.Complete()); | 40 EXPECT_FALSE(buffer_.Complete()); |
| 41 EXPECT_FALSE(buffer_.is_key_frame()); | 41 transport::EncodedFrame frame; |
| 42 transport::EncodedVideoFrame frame; | 42 EXPECT_FALSE(buffer_.AssembleEncodedFrame(&frame)); |
| 43 EXPECT_FALSE(buffer_.GetEncodedVideoFrame(&frame)); | |
| 44 } | 43 } |
| 45 | 44 |
| 46 TEST_F(FrameBufferTest, DefaultOnePacketFrame) { | 45 TEST_F(FrameBufferTest, DefaultOnePacketFrame) { |
| 47 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 46 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 48 EXPECT_TRUE(buffer_.Complete()); | 47 EXPECT_TRUE(buffer_.Complete()); |
| 49 EXPECT_FALSE(buffer_.is_key_frame()); | 48 EXPECT_TRUE(buffer_.is_key_frame()); |
| 50 transport::EncodedVideoFrame frame; | 49 transport::EncodedFrame frame; |
| 51 EXPECT_TRUE(buffer_.GetEncodedVideoFrame(&frame)); | 50 EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame)); |
| 52 EXPECT_EQ(payload_.size(), frame.data.size()); | 51 EXPECT_EQ(payload_.size(), frame.data.size()); |
| 53 } | 52 } |
| 54 | 53 |
| 55 TEST_F(FrameBufferTest, MultiplePacketFrame) { | 54 TEST_F(FrameBufferTest, MultiplePacketFrame) { |
| 56 rtp_header_.is_key_frame = true; | 55 rtp_header_.is_key_frame = true; |
| 57 rtp_header_.max_packet_id = 2; | 56 rtp_header_.max_packet_id = 2; |
| 58 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 57 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 59 ++rtp_header_.packet_id; | 58 ++rtp_header_.packet_id; |
| 60 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 59 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 61 ++rtp_header_.packet_id; | 60 ++rtp_header_.packet_id; |
| 62 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 61 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 63 ++rtp_header_.packet_id; | 62 ++rtp_header_.packet_id; |
| 64 EXPECT_TRUE(buffer_.Complete()); | 63 EXPECT_TRUE(buffer_.Complete()); |
| 65 EXPECT_TRUE(buffer_.is_key_frame()); | 64 EXPECT_TRUE(buffer_.is_key_frame()); |
| 66 transport::EncodedVideoFrame frame; | 65 transport::EncodedFrame frame; |
| 67 EXPECT_TRUE(buffer_.GetEncodedVideoFrame(&frame)); | 66 EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame)); |
| 68 EXPECT_EQ(3 * payload_.size(), frame.data.size()); | 67 EXPECT_EQ(3 * payload_.size(), frame.data.size()); |
| 69 } | 68 } |
| 70 | 69 |
| 71 TEST_F(FrameBufferTest, IncompleteFrame) { | 70 TEST_F(FrameBufferTest, IncompleteFrame) { |
| 72 rtp_header_.max_packet_id = 4; | 71 rtp_header_.max_packet_id = 4; |
| 73 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 72 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 74 ++rtp_header_.packet_id; | 73 ++rtp_header_.packet_id; |
| 75 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 74 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 76 ++rtp_header_.packet_id; | 75 ++rtp_header_.packet_id; |
| 77 // Increment again - skip packet #2. | 76 // Increment again - skip packet #2. |
| 78 ++rtp_header_.packet_id; | 77 ++rtp_header_.packet_id; |
| 79 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 78 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 80 ++rtp_header_.packet_id; | 79 ++rtp_header_.packet_id; |
| 81 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 80 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 82 EXPECT_FALSE(buffer_.Complete()); | 81 EXPECT_FALSE(buffer_.Complete()); |
| 83 // Insert missing packet. | 82 // Insert missing packet. |
| 84 rtp_header_.packet_id = 2; | 83 rtp_header_.packet_id = 2; |
| 85 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); | 84 buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
| 86 EXPECT_TRUE(buffer_.Complete()); | 85 EXPECT_TRUE(buffer_.Complete()); |
| 87 } | 86 } |
| 88 | 87 |
| 89 } // namespace media | 88 } // namespace media |
| 90 } // namespace cast | 89 } // namespace cast |
| OLD | NEW |