| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "chromecast/media/cma/test/mock_frame_consumer.h" | 5 #include "chromecast/media/cma/test/mock_frame_consumer.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/location.h" | 8 #include "base/location.h" |
| 9 #include "base/message_loop/message_loop_proxy.h" | 9 #include "base/single_thread_task_runner.h" |
| 10 #include "base/thread_task_runner_handle.h" |
| 10 #include "base/time/time.h" | 11 #include "base/time/time.h" |
| 11 #include "chromecast/media/cma/base/coded_frame_provider.h" | 12 #include "chromecast/media/cma/base/coded_frame_provider.h" |
| 12 #include "chromecast/media/cma/base/decoder_buffer_base.h" | 13 #include "chromecast/media/cma/base/decoder_buffer_base.h" |
| 13 #include "chromecast/media/cma/test/frame_generator_for_test.h" | 14 #include "chromecast/media/cma/test/frame_generator_for_test.h" |
| 14 #include "media/base/video_decoder_config.h" | 15 #include "media/base/video_decoder_config.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 17 |
| 17 namespace chromecast { | 18 namespace chromecast { |
| 18 namespace media { | 19 namespace media { |
| 19 | 20 |
| (...skipping 14 matching lines...) Expand all Loading... |
| 34 delayed_task_pattern_ = delayed_task_pattern; | 35 delayed_task_pattern_ = delayed_task_pattern; |
| 35 last_read_aborted_by_flush_ = last_read_aborted_by_flush; | 36 last_read_aborted_by_flush_ = last_read_aborted_by_flush; |
| 36 frame_generator_.reset(frame_generator.release()); | 37 frame_generator_.reset(frame_generator.release()); |
| 37 } | 38 } |
| 38 | 39 |
| 39 void MockFrameConsumer::Start(const base::Closure& done_cb) { | 40 void MockFrameConsumer::Start(const base::Closure& done_cb) { |
| 40 done_cb_ = done_cb; | 41 done_cb_ = done_cb; |
| 41 | 42 |
| 42 pattern_idx_ = 0; | 43 pattern_idx_ = 0; |
| 43 | 44 |
| 44 base::MessageLoopProxy::current()->PostTask( | 45 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 45 FROM_HERE, | 46 FROM_HERE, |
| 46 base::Bind(&MockFrameConsumer::ReadFrame, base::Unretained(this))); | 47 base::Bind(&MockFrameConsumer::ReadFrame, base::Unretained(this))); |
| 47 } | 48 } |
| 48 | 49 |
| 49 void MockFrameConsumer::ReadFrame() { | 50 void MockFrameConsumer::ReadFrame() { |
| 50 // Once all the frames have been read, flush the frame provider. | 51 // Once all the frames have been read, flush the frame provider. |
| 51 if (frame_generator_->RemainingFrameCount() == 0 && | 52 if (frame_generator_->RemainingFrameCount() == 0 && |
| 52 !last_read_aborted_by_flush_) { | 53 !last_read_aborted_by_flush_) { |
| 53 coded_frame_provider_->Flush( | 54 coded_frame_provider_->Flush( |
| 54 base::Bind(&MockFrameConsumer::OnFlushCompleted, | 55 base::Bind(&MockFrameConsumer::OnFlushCompleted, |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 87 EXPECT_EQ(buffer->timestamp(), ref_buffer->timestamp()); | 88 EXPECT_EQ(buffer->timestamp(), ref_buffer->timestamp()); |
| 88 ASSERT_EQ(buffer->data_size(), ref_buffer->data_size()); | 89 ASSERT_EQ(buffer->data_size(), ref_buffer->data_size()); |
| 89 for (size_t k = 0; k < ref_buffer->data_size(); k++) | 90 for (size_t k = 0; k < ref_buffer->data_size(); k++) |
| 90 EXPECT_EQ(buffer->data()[k], ref_buffer->data()[k]); | 91 EXPECT_EQ(buffer->data()[k], ref_buffer->data()[k]); |
| 91 } | 92 } |
| 92 | 93 |
| 93 bool delayed = delayed_task_pattern_[pattern_idx_]; | 94 bool delayed = delayed_task_pattern_[pattern_idx_]; |
| 94 pattern_idx_ = (pattern_idx_ + 1) % delayed_task_pattern_.size(); | 95 pattern_idx_ = (pattern_idx_ + 1) % delayed_task_pattern_.size(); |
| 95 | 96 |
| 96 if (delayed) { | 97 if (delayed) { |
| 97 base::MessageLoopProxy::current()->PostDelayedTask( | 98 base::ThreadTaskRunnerHandle::Get()->PostDelayedTask( |
| 98 FROM_HERE, | 99 FROM_HERE, |
| 99 base::Bind(&MockFrameConsumer::ReadFrame, | 100 base::Bind(&MockFrameConsumer::ReadFrame, base::Unretained(this)), |
| 100 base::Unretained(this)), | |
| 101 base::TimeDelta::FromMilliseconds(1)); | 101 base::TimeDelta::FromMilliseconds(1)); |
| 102 } else { | 102 } else { |
| 103 base::MessageLoopProxy::current()->PostTask( | 103 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 104 FROM_HERE, | 104 FROM_HERE, |
| 105 base::Bind(&MockFrameConsumer::ReadFrame, | 105 base::Bind(&MockFrameConsumer::ReadFrame, base::Unretained(this))); |
| 106 base::Unretained(this))); | |
| 107 } | 106 } |
| 108 } | 107 } |
| 109 | 108 |
| 110 void MockFrameConsumer::OnFlushCompleted() { | 109 void MockFrameConsumer::OnFlushCompleted() { |
| 111 EXPECT_EQ(frame_generator_->RemainingFrameCount(), 0u); | 110 EXPECT_EQ(frame_generator_->RemainingFrameCount(), 0u); |
| 112 done_cb_.Run(); | 111 done_cb_.Run(); |
| 113 } | 112 } |
| 114 | 113 |
| 115 } // namespace media | 114 } // namespace media |
| 116 } // namespace chromecast | 115 } // namespace chromecast |
| OLD | NEW |