Index: media/remoting/rpc/decoder_buffer_segment_unittest.cc |
diff --git a/media/remoting/rpc/decoder_buffer_segment_unittest.cc b/media/remoting/rpc/decoder_buffer_segment_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..3f4540de7ba7026ca8cead05fcd4fff38c5802ed |
--- /dev/null |
+++ b/media/remoting/rpc/decoder_buffer_segment_unittest.cc |
@@ -0,0 +1,89 @@ |
+// Copyright 2016 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/remoting/rpc/decoder_buffer_segment.h" |
+ |
+#include <string> |
+#include <vector> |
+ |
+#include "base/macros.h" |
+#include "base/memory/ref_counted.h" |
+#include "media/base/decoder_buffer.h" |
+#include "media/remoting/remoting_rpc_message.pb.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace media { |
+namespace remoting { |
+ |
+class DecoderBufferSegmentTest : public testing::Test { |
+ protected: |
+ void SetUp() override {} |
+}; |
+ |
+TEST_F(DecoderBufferSegmentTest, PassEOSDecoderBuffer) { |
+ scoped_refptr<::media::DecoderBuffer> input_buffer = |
+ ::media::DecoderBuffer::CreateEOSBuffer(); |
+ |
+ scoped_refptr<DecoderBufferSegment> from_rpc( |
+ new DecoderBufferSegment(input_buffer)); |
+ std::vector<uint8_t> data; |
+ uint32_t data_size = from_rpc->ToBuffer(&data); |
+ ASSERT_EQ(data_size, data.size()); |
+ |
+ // Convert proto buffer string back to RPC data structure. |
+ scoped_refptr<DecoderBufferSegment> to_rpc = |
+ DecoderBufferSegment::FromBuffer(&data[0], data.size()); |
+ DCHECK(to_rpc); |
+ |
+ scoped_refptr<::media::DecoderBuffer> output_buffer = |
+ to_rpc->decoder_buffer(); |
+ |
+ ASSERT_TRUE(output_buffer->end_of_stream()); |
+} |
+ |
+TEST_F(DecoderBufferSegmentTest, PassValidDecoderBuffer) { |
+ const uint8_t buffer[] = "ABCDE"; |
+ size_t buffer_size = sizeof(buffer) / sizeof(uint8_t); |
+ const uint8_t side_buffer[] = "XX"; |
+ size_t side_buffer_size = sizeof(side_buffer) / sizeof(uint8_t); |
+ base::TimeDelta pts = base::TimeDelta::FromMilliseconds(5); |
+ scoped_refptr<::media::DecoderBuffer> input_buffer = |
+ ::media::DecoderBuffer::CopyFrom(buffer, buffer_size, side_buffer, |
+ side_buffer_size); |
+ input_buffer->set_timestamp(pts); |
+ input_buffer->set_is_key_frame(true); |
+ |
+ scoped_refptr<DecoderBufferSegment> from_rpc( |
+ new DecoderBufferSegment(input_buffer)); |
+ std::vector<uint8_t> data; |
+ uint32_t data_size = from_rpc->ToBuffer(&data); |
+ ASSERT_EQ(data_size, data.size()); |
+ |
+ // Convert proto buffer string back to RPC data structure. |
+ scoped_refptr<DecoderBufferSegment> to_rpc = |
+ DecoderBufferSegment::FromBuffer(&data[0], data.size()); |
+ DCHECK(to_rpc); |
+ |
+ scoped_refptr<::media::DecoderBuffer> output_buffer = |
+ to_rpc->decoder_buffer(); |
+ |
+ ASSERT_FALSE(output_buffer->end_of_stream()); |
+ ASSERT_TRUE(output_buffer->is_key_frame()); |
+ ASSERT_EQ(output_buffer->timestamp(), pts); |
+ ASSERT_EQ(output_buffer->data_size(), buffer_size); |
+ const uint8_t* output_data = output_buffer->data(); |
+ ; |
+ for (size_t i = 0; i < buffer_size; i++) { |
+ ASSERT_EQ(output_data[i], buffer[i]); |
+ } |
+ ASSERT_EQ(output_buffer->side_data_size(), side_buffer_size); |
+ const uint8_t* output_side_data = output_buffer->side_data(); |
+ ; |
+ for (size_t i = 0; i < side_buffer_size; i++) { |
+ ASSERT_EQ(output_side_data[i], side_buffer[i]); |
+ } |
+} |
+ |
+} // namespace remoting |
+} // namespace media |