Define remote playback proto buffer

media/remoting/rpc/decoder_buffer_segment_unittest.cc

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