OLD | NEW |
(Empty) | |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "media/remoting/rpc/decoder_buffer_segment.h" |
| 6 |
| 7 #include <string> |
| 8 #include <vector> |
| 9 |
| 10 #include "base/macros.h" |
| 11 #include "base/memory/ref_counted.h" |
| 12 #include "media/base/decoder_buffer.h" |
| 13 #include "media/remoting/remoting_rpc_message.pb.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 |
| 16 namespace media { |
| 17 namespace remoting { |
| 18 |
| 19 class DecoderBufferSegmentTest : public testing::Test { |
| 20 protected: |
| 21 void SetUp() override {} |
| 22 }; |
| 23 |
| 24 TEST_F(DecoderBufferSegmentTest, PassEOSDecoderBuffer) { |
| 25 scoped_refptr<::media::DecoderBuffer> input_buffer = |
| 26 ::media::DecoderBuffer::CreateEOSBuffer(); |
| 27 |
| 28 scoped_refptr<DecoderBufferSegment> from_rpc( |
| 29 new DecoderBufferSegment(input_buffer)); |
| 30 std::vector<uint8_t> data; |
| 31 uint32_t data_size = from_rpc->ToBuffer(&data); |
| 32 ASSERT_EQ(data_size, data.size()); |
| 33 |
| 34 // Convert proto buffer string back to RPC data structure. |
| 35 scoped_refptr<DecoderBufferSegment> to_rpc = |
| 36 DecoderBufferSegment::FromBuffer(&data[0], data.size()); |
| 37 DCHECK(to_rpc); |
| 38 |
| 39 scoped_refptr<::media::DecoderBuffer> output_buffer = |
| 40 to_rpc->decoder_buffer(); |
| 41 |
| 42 ASSERT_TRUE(output_buffer->end_of_stream()); |
| 43 } |
| 44 |
| 45 TEST_F(DecoderBufferSegmentTest, PassValidDecoderBuffer) { |
| 46 const uint8_t buffer[] = "ABCDE"; |
| 47 size_t buffer_size = sizeof(buffer) / sizeof(uint8_t); |
| 48 const uint8_t side_buffer[] = "XX"; |
| 49 size_t side_buffer_size = sizeof(side_buffer) / sizeof(uint8_t); |
| 50 base::TimeDelta pts = base::TimeDelta::FromMilliseconds(5); |
| 51 scoped_refptr<::media::DecoderBuffer> input_buffer = |
| 52 ::media::DecoderBuffer::CopyFrom(buffer, buffer_size, side_buffer, |
| 53 side_buffer_size); |
| 54 input_buffer->set_timestamp(pts); |
| 55 input_buffer->set_is_key_frame(true); |
| 56 |
| 57 scoped_refptr<DecoderBufferSegment> from_rpc( |
| 58 new DecoderBufferSegment(input_buffer)); |
| 59 std::vector<uint8_t> data; |
| 60 uint32_t data_size = from_rpc->ToBuffer(&data); |
| 61 ASSERT_EQ(data_size, data.size()); |
| 62 |
| 63 // Convert proto buffer string back to RPC data structure. |
| 64 scoped_refptr<DecoderBufferSegment> to_rpc = |
| 65 DecoderBufferSegment::FromBuffer(&data[0], data.size()); |
| 66 DCHECK(to_rpc); |
| 67 |
| 68 scoped_refptr<::media::DecoderBuffer> output_buffer = |
| 69 to_rpc->decoder_buffer(); |
| 70 |
| 71 ASSERT_FALSE(output_buffer->end_of_stream()); |
| 72 ASSERT_TRUE(output_buffer->is_key_frame()); |
| 73 ASSERT_EQ(output_buffer->timestamp(), pts); |
| 74 ASSERT_EQ(output_buffer->data_size(), buffer_size); |
| 75 const uint8_t* output_data = output_buffer->data(); |
| 76 ; |
| 77 for (size_t i = 0; i < buffer_size; i++) { |
| 78 ASSERT_EQ(output_data[i], buffer[i]); |
| 79 } |
| 80 ASSERT_EQ(output_buffer->side_data_size(), side_buffer_size); |
| 81 const uint8_t* output_side_data = output_buffer->side_data(); |
| 82 ; |
| 83 for (size_t i = 0; i < side_buffer_size; i++) { |
| 84 ASSERT_EQ(output_side_data[i], side_buffer[i]); |
| 85 } |
| 86 } |
| 87 |
| 88 } // namespace remoting |
| 89 } // namespace media |
OLD | NEW |