| 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 { |
| 11 | 11 |
| 12 class FrameBufferTest : public ::testing::Test { | 12 class FrameBufferTest : public ::testing::Test { |
| 13 protected: | 13 protected: |
| 14 FrameBufferTest() { | 14 FrameBufferTest() { |
| 15 payload_.assign(kMaxIpPacketSize, 0); | 15 payload_.assign(kMaxIpPacketSize, 0); |
| 16 } | 16 } |
| 17 | 17 |
| 18 ~FrameBufferTest() override {} | 18 ~FrameBufferTest() override {} |
| 19 | 19 |
| 20 FrameBuffer buffer_; | 20 FrameBuffer buffer_; |
| 21 std::vector<uint8> payload_; | 21 std::vector<uint8> payload_; |
| 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 = RtpTimeTicks().Expand(UINT32_C(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_[0], 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.lower_32_bits()); |
| 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_[0], payload_.size(), rtp_header_); | 48 buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_); |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 82 buffer_.InsertPacket(&payload_[0], 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_[0], 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 |