| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/bind.h" | 5 #include "base/bind.h" |
| 6 #include "base/location.h" | 6 #include "base/location.h" |
| 7 #include "base/macros.h" | 7 #include "base/macros.h" |
| 8 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
| 9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "media/base/video_frame.h" | 10 #include "media/base/video_frame.h" |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 }; | 27 }; |
| 28 | 28 |
| 29 class WebmMuxerTest : public testing::Test, public EventHandlerInterface { | 29 class WebmMuxerTest : public testing::Test, public EventHandlerInterface { |
| 30 public: | 30 public: |
| 31 WebmMuxerTest() | 31 WebmMuxerTest() |
| 32 : webm_muxer_(base::Bind(&WebmMuxerTest::WriteCallback, | 32 : webm_muxer_(base::Bind(&WebmMuxerTest::WriteCallback, |
| 33 base::Unretained(this))), | 33 base::Unretained(this))), |
| 34 last_encoded_length_(0), | 34 last_encoded_length_(0), |
| 35 accumulated_position_(0) { | 35 accumulated_position_(0) { |
| 36 EXPECT_EQ(webm_muxer_.Position(), 0); | 36 EXPECT_EQ(webm_muxer_.Position(), 0); |
| 37 const mkvmuxer::int64 kRandomNewPosition = 333; |
| 38 EXPECT_EQ(webm_muxer_.Position(kRandomNewPosition), -1); |
| 37 EXPECT_FALSE(webm_muxer_.Seekable()); | 39 EXPECT_FALSE(webm_muxer_.Seekable()); |
| 38 EXPECT_EQ(webm_muxer_.segment_.mode(), mkvmuxer::Segment::kLive); | |
| 39 } | 40 } |
| 40 | 41 |
| 41 MOCK_METHOD1(WriteCallback, void(const base::StringPiece&)); | 42 MOCK_METHOD1(WriteCallback, void(const base::StringPiece&)); |
| 42 | 43 |
| 43 void SaveEncodedDataLen(const base::StringPiece& encoded_data) { | 44 void SaveEncodedDataLen(const base::StringPiece& encoded_data) { |
| 44 last_encoded_length_ = encoded_data.size(); | 45 last_encoded_length_ = encoded_data.size(); |
| 45 accumulated_position_ += encoded_data.size(); | 46 accumulated_position_ += encoded_data.size(); |
| 46 } | 47 } |
| 47 | 48 |
| 48 mkvmuxer::int64 GetWebmMuxerPosition() const { | 49 mkvmuxer::int64 GetWebmMuxerPosition() const { |
| 49 return webm_muxer_.Position(); | 50 return webm_muxer_.Position(); |
| 50 } | 51 } |
| 51 | 52 |
| 53 mkvmuxer::Segment::Mode GetWebmSegmentMode() const { |
| 54 return webm_muxer_.segment_.mode(); |
| 55 } |
| 56 |
| 52 mkvmuxer::int32 WebmMuxerWrite(const void* buf, mkvmuxer::uint32 len) { | 57 mkvmuxer::int32 WebmMuxerWrite(const void* buf, mkvmuxer::uint32 len) { |
| 53 return webm_muxer_.Write(buf, len); | 58 return webm_muxer_.Write(buf, len); |
| 54 } | 59 } |
| 55 | 60 |
| 56 WebmMuxer webm_muxer_; | 61 WebmMuxer webm_muxer_; |
| 57 | 62 |
| 58 size_t last_encoded_length_; | 63 size_t last_encoded_length_; |
| 59 int64_t accumulated_position_; | 64 int64_t accumulated_position_; |
| 60 | 65 |
| 61 private: | 66 private: |
| (...skipping 26 matching lines...) Expand all Loading... |
| 88 webm_muxer_.OnEncodedVideo(video_frame, | 93 webm_muxer_.OnEncodedVideo(video_frame, |
| 89 encoded_data, | 94 encoded_data, |
| 90 base::TimeTicks::Now(), | 95 base::TimeTicks::Now(), |
| 91 false /* keyframe */); | 96 false /* keyframe */); |
| 92 | 97 |
| 93 // First time around WriteCallback() is pinged a number of times to write the | 98 // First time around WriteCallback() is pinged a number of times to write the |
| 94 // Matroska header, but at the end it dumps |encoded_data|. | 99 // Matroska header, but at the end it dumps |encoded_data|. |
| 95 EXPECT_EQ(last_encoded_length_, encoded_data.size()); | 100 EXPECT_EQ(last_encoded_length_, encoded_data.size()); |
| 96 EXPECT_EQ(GetWebmMuxerPosition(), accumulated_position_); | 101 EXPECT_EQ(GetWebmMuxerPosition(), accumulated_position_); |
| 97 EXPECT_GE(GetWebmMuxerPosition(), static_cast<int64_t>(last_encoded_length_)); | 102 EXPECT_GE(GetWebmMuxerPosition(), static_cast<int64_t>(last_encoded_length_)); |
| 103 EXPECT_EQ(GetWebmSegmentMode(), mkvmuxer::Segment::kLive); |
| 98 | 104 |
| 99 const int64_t begin_of_second_block = accumulated_position_; | 105 const int64_t begin_of_second_block = accumulated_position_; |
| 100 EXPECT_CALL(*this, WriteCallback(_)) | 106 EXPECT_CALL(*this, WriteCallback(_)) |
| 101 .Times(AtLeast(1)) | 107 .Times(AtLeast(1)) |
| 102 .WillRepeatedly(WithArgs<0>( | 108 .WillRepeatedly(WithArgs<0>( |
| 103 Invoke(this, &WebmMuxerTest::SaveEncodedDataLen))); | 109 Invoke(this, &WebmMuxerTest::SaveEncodedDataLen))); |
| 104 webm_muxer_.OnEncodedVideo(video_frame, | 110 webm_muxer_.OnEncodedVideo(video_frame, |
| 105 encoded_data, | 111 encoded_data, |
| 106 base::TimeTicks::Now(), | 112 base::TimeTicks::Now(), |
| 107 false /* keyframe */); | 113 false /* keyframe */); |
| 108 | 114 |
| 109 // The second time around the callbacks should include a SimpleBlock header, | 115 // The second time around the callbacks should include a SimpleBlock header, |
| 110 // namely the track index, a timestamp and a flags byte, for a total of 6B. | 116 // namely the track index, a timestamp and a flags byte, for a total of 6B. |
| 111 EXPECT_EQ(last_encoded_length_, encoded_data.size()); | 117 EXPECT_EQ(last_encoded_length_, encoded_data.size()); |
| 112 EXPECT_EQ(GetWebmMuxerPosition(), accumulated_position_); | 118 EXPECT_EQ(GetWebmMuxerPosition(), accumulated_position_); |
| 113 const uint32_t kSimpleBlockSize = 6u; | 119 const uint32_t kSimpleBlockSize = 6u; |
| 114 EXPECT_EQ(static_cast<int64_t>(begin_of_second_block + kSimpleBlockSize + | 120 EXPECT_EQ(static_cast<int64_t>(begin_of_second_block + kSimpleBlockSize + |
| 115 encoded_data.size()), | 121 encoded_data.size()), |
| 116 accumulated_position_); | 122 accumulated_position_); |
| 117 } | 123 } |
| 118 | 124 |
| 119 } // namespace media | 125 } // namespace media |
| OLD | NEW |