Index: media/cast/framer/frame_buffer_unittest.cc |
diff --git a/media/cast/framer/frame_buffer_unittest.cc b/media/cast/framer/frame_buffer_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..26998f5fd7ea3cde311b8747637e2fdd5834fdf8 |
--- /dev/null |
+++ b/media/cast/framer/frame_buffer_unittest.cc |
@@ -0,0 +1,88 @@ |
+// Copyright 2013 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "media/cast/framer/frame_buffer.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace media { |
+namespace cast { |
+ |
+class FrameBufferTest : public ::testing::Test { |
+ protected: |
+ FrameBufferTest() {} |
+ |
+ ~FrameBufferTest() {} |
+ |
+ void SetUp() { |
+ payload_.assign(kIpPacketSize, 0); |
+ |
+ // Build a default one packet frame - populate webrtc header. |
+ rtp_header_.is_key_frame = false; |
+ rtp_header_.frame_id = 0; |
+ rtp_header_.packet_id = 0; |
+ rtp_header_.max_packet_id = 0; |
+ rtp_header_.is_reference = false; |
+ rtp_header_.reference_frame_id = 0; |
+ } |
+ |
+ FrameBuffer buffer_; |
+ std::vector<uint8> payload_; |
+ RtpCastHeader rtp_header_; |
+}; |
+ |
+TEST_F(FrameBufferTest, EmptyBuffer) { |
+ EXPECT_FALSE(buffer_.Complete()); |
+ EXPECT_FALSE(buffer_.is_key_frame()); |
+ EncodedVideoFrame frame; |
+ uint32 rtp_timestamp; |
+ EXPECT_FALSE(buffer_.GetEncodedVideoFrame(&frame, &rtp_timestamp)); |
+} |
+ |
+TEST_F(FrameBufferTest, DefaultOnePacketFrame) { |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ EXPECT_TRUE(buffer_.Complete()); |
+ EXPECT_FALSE(buffer_.is_key_frame()); |
+ EncodedVideoFrame frame; |
+ uint32 rtp_timestamp; |
+ EXPECT_TRUE(buffer_.GetEncodedVideoFrame(&frame, &rtp_timestamp)); |
+ EXPECT_EQ(payload_.size(), frame.data.size()); |
+} |
+ |
+TEST_F(FrameBufferTest, MultiplePacketFrame) { |
+ rtp_header_.is_key_frame = true; |
+ rtp_header_.max_packet_id = 2; |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ ++rtp_header_.packet_id; |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ ++rtp_header_.packet_id; |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ ++rtp_header_.packet_id; |
+ EXPECT_TRUE(buffer_.Complete()); |
+ EXPECT_TRUE(buffer_.is_key_frame()); |
+ EncodedVideoFrame frame; |
+ uint32 rtp_timestamp; |
+ EXPECT_TRUE(buffer_.GetEncodedVideoFrame(&frame, &rtp_timestamp)); |
+ EXPECT_EQ(3 * payload_.size(), frame.data.size()); |
+} |
+ |
+TEST_F(FrameBufferTest, InCompleteFrame) { |
+ rtp_header_.max_packet_id = 4; |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ ++rtp_header_.packet_id; |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ ++rtp_header_.packet_id; |
+ // Increment again - skip packet #2. |
+ ++rtp_header_.packet_id; |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ ++rtp_header_.packet_id; |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ EXPECT_FALSE(buffer_.Complete()); |
+ // Insert missing packet. |
+ rtp_header_.packet_id = 2; |
+ buffer_.InsertPacket(payload_.data(), payload_.size(), rtp_header_); |
+ EXPECT_TRUE(buffer_.Complete()); |
+} |
+ |
+} // namespace media |
+} // namespace cast |