| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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 <vector> | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "media/base/data_buffer.h" | |
| 9 #include "media/base/mock_callback.h" | |
| 10 #include "media/base/mock_filters.h" | |
| 11 #include "media/base/mock_task.h" | |
| 12 #include "media/base/pipeline.h" | |
| 13 #include "media/filters/decoder_base.h" | |
| 14 #include "testing/gmock/include/gmock/gmock.h" | |
| 15 #include "testing/gtest/include/gtest/gtest.h" | |
| 16 | |
| 17 using ::testing::_; | |
| 18 using ::testing::AnyNumber; | |
| 19 using ::testing::NotNull; | |
| 20 using ::testing::StrictMock; | |
| 21 using ::testing::WithArg; | |
| 22 | |
| 23 namespace media { | |
| 24 | |
| 25 class GTEST_TEST_CLASS_NAME_(DecoderBaseTest, FlowControl); | |
| 26 | |
| 27 class MockDecoderCallback { | |
| 28 public: | |
| 29 MOCK_METHOD1(OnReadComplete, void(scoped_refptr<Buffer> output)); | |
| 30 }; | |
| 31 | |
| 32 class MockDecoderImpl : public DecoderBase<MockAudioDecoder, Buffer> { | |
| 33 public: | |
| 34 explicit MockDecoderImpl(MessageLoop* message_loop) | |
| 35 : DecoderBase<MockAudioDecoder, Buffer>(message_loop) { | |
| 36 } | |
| 37 | |
| 38 virtual ~MockDecoderImpl() {} | |
| 39 | |
| 40 // DecoderBase implementation. | |
| 41 MOCK_METHOD3(DoInitialize, | |
| 42 void(DemuxerStream* demuxer_stream, | |
| 43 bool* success, | |
| 44 Task* done_cb)); | |
| 45 MOCK_METHOD1(DoStop, void(Task* done_cb)); | |
| 46 MOCK_METHOD2(DoSeek, void(base::TimeDelta time, Task* done_cb)); | |
| 47 MOCK_METHOD1(DoDecode, void(Buffer* input)); | |
| 48 | |
| 49 private: | |
| 50 FRIEND_TEST(DecoderBaseTest, FlowControl); | |
| 51 | |
| 52 DISALLOW_COPY_AND_ASSIGN(MockDecoderImpl); | |
| 53 }; | |
| 54 | |
| 55 ACTION(Initialize) { | |
| 56 base::ScopedTaskRunner done_runner(arg2); | |
| 57 *arg1 = true; | |
| 58 } | |
| 59 | |
| 60 ACTION_P(SaveDecodeRequest, list) { | |
| 61 scoped_refptr<Buffer> buffer(arg0); | |
| 62 list->push_back(buffer); | |
| 63 } | |
| 64 | |
| 65 ACTION(CompleteDemuxRequest) { | |
| 66 arg0.Run(new DataBuffer(0)); | |
| 67 } | |
| 68 | |
| 69 // Test the flow control of decoder base by the following sequence of actions: | |
| 70 // - Read() -> DecoderStream | |
| 71 // \ Read() -> DemuxerStream | |
| 72 // - Read() -> DecoderBase | |
| 73 // \ Read() -> DemixerStream | |
| 74 // - ReadCallback() -> DecoderBase | |
| 75 // \ DoDecode() -> Decoder | |
| 76 // - ReadCallback() -> DecoderBase | |
| 77 // \ DoDecode() -> Decoder | |
| 78 // - DecodeCallback() -> DecoderBase | |
| 79 // \ ReadCallback() -> client | |
| 80 // - DecodeCallback() -> DecoderBase | |
| 81 // \ ReadCallback() -> client | |
| 82 TEST(DecoderBaseTest, FlowControl) { | |
| 83 MessageLoop message_loop; | |
| 84 scoped_refptr<MockDecoderImpl> decoder(new MockDecoderImpl(&message_loop)); | |
| 85 | |
| 86 MockDecoderCallback read_callback; | |
| 87 decoder->set_consume_audio_samples_callback( | |
| 88 base::Bind(&MockDecoderCallback::OnReadComplete, | |
| 89 base::Unretained(&read_callback))); | |
| 90 | |
| 91 scoped_refptr<MockDemuxerStream> demuxer_stream(new MockDemuxerStream()); | |
| 92 MockStatisticsCallback stats_callback_object; | |
| 93 EXPECT_CALL(stats_callback_object, OnStatistics(_)) | |
| 94 .Times(AnyNumber()); | |
| 95 | |
| 96 // Initialize. | |
| 97 EXPECT_CALL(*decoder, DoInitialize(NotNull(), NotNull(), NotNull())) | |
| 98 .WillOnce(Initialize()); | |
| 99 decoder->Initialize(demuxer_stream.get(), NewExpectedCallback(), | |
| 100 NewCallback(&stats_callback_object, | |
| 101 &MockStatisticsCallback::OnStatistics)); | |
| 102 message_loop.RunAllPending(); | |
| 103 | |
| 104 // Read. | |
| 105 std::vector<scoped_refptr<Buffer> > decode_requests; | |
| 106 EXPECT_CALL(*demuxer_stream, Read(_)) | |
| 107 .Times(2) | |
| 108 .WillRepeatedly(CompleteDemuxRequest()); | |
| 109 EXPECT_CALL(*decoder, DoDecode(NotNull())) | |
| 110 .Times(2) | |
| 111 .WillRepeatedly(SaveDecodeRequest(&decode_requests)); | |
| 112 scoped_refptr<DataBuffer> output; | |
| 113 decoder->PostReadTaskHack(output); | |
| 114 decoder->PostReadTaskHack(output); | |
| 115 message_loop.RunAllPending(); | |
| 116 | |
| 117 // Fulfill the decode request. | |
| 118 EXPECT_CALL(read_callback, OnReadComplete(_)).Times(2); | |
| 119 PipelineStatistics statistics; | |
| 120 for (size_t i = 0; i < decode_requests.size(); ++i) { | |
| 121 decoder->EnqueueResult(new DataBuffer(0)); | |
| 122 decoder->OnDecodeComplete(statistics); | |
| 123 } | |
| 124 decode_requests.clear(); | |
| 125 message_loop.RunAllPending(); | |
| 126 | |
| 127 // Stop. | |
| 128 EXPECT_CALL(*decoder, DoStop(_)) | |
| 129 .WillOnce(WithArg<0>(InvokeRunnable())); | |
| 130 decoder->Stop(NewExpectedCallback()); | |
| 131 message_loop.RunAllPending(); | |
| 132 } | |
| 133 | |
| 134 } // namespace media | |
| OLD | NEW |