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..c5ef59f45e5932ef593371f23934e40cd2733c80
--- /dev/null
+++ b/media/remoting/rpc/decoder_buffer_segment_unittest.cc
@@ -0,0 +1,106 @@
+// 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[] = {
+      0,   0,   0,   1,   9,   224, 0,   0,   0,   1,   103, 77,  64,  21,  217,
+      1,   177, 254, 78,  16,  0,   0,   62,  144, 0,   11,  184, 0,   241, 98,
+      228, 128, 0,   0,   0,   1,   104, 235, 143, 32,  0,   0,   0,   1,   103,
+      77,  64,  21,  217, 1,   177, 254, 78,  16,  0,   0,   62,  144, 0,   11,
+      184, 0,   241, 98,  228, 128, 0,   0,   0,   1,   104, 235, 143, 32,  0,
+      0,   0,   1,   101, 136, 132, 25,  255, 0,   191, 98,  0,   6,   29,  63,
+      252, 65,  246, 207, 255, 235, 63,  172, 35,  112, 198, 115, 222, 243, 159,
+      232, 208, 32,  0,   0,   3,   0,   0,   203, 255, 149, 20,  71,  203, 213,
+      40,  0,   0,   139, 0,   24,  117, 166, 249, 227, 68,  230, 177, 134, 161,
+      162, 1,   22,  105, 78,  66,  183, 130, 158, 108, 252, 112, 113, 58,  159,
+      72,  116, 78,  141, 133, 76,  225, 209, 13,  221, 49,  187, 83,  123, 193,
+      112, 123, 112, 74,  121, 133};
+  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);
+
+  // 1. To DecoderBuffer
+  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);
+
+  // 2. To DecoderBufferSegment
+  scoped_refptr<DecoderBufferSegment> from_rpc(
+      new DecoderBufferSegment(input_buffer));
+
+  // 3. To Byte Array
+  std::vector<uint8_t> data;
+  uint32_t data_size = from_rpc->ToBuffer(&data);
+  ASSERT_EQ(data_size, data.size());
+
+  // 4. Back To DecoderBufferSegment
+  // Convert proto buffer string back to RPC data structure.
+  scoped_refptr<DecoderBufferSegment> to_rpc =
+      DecoderBufferSegment::FromBuffer(data.data(), data.size());
+  DCHECK(to_rpc);
+
+  // 5. To DecoderBuffer
+  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