Index: media/base/pipeline_impl_unittest.cc |
diff --git a/media/base/pipeline_impl_unittest.cc b/media/base/pipeline_impl_unittest.cc |
index 1e54794b4bfbc5f9287694fe2c7ec3980c76a230..84332ba404adcb7186ed15b37d83719f04b8b53e 100644 |
--- a/media/base/pipeline_impl_unittest.cc |
+++ b/media/base/pipeline_impl_unittest.cc |
@@ -77,7 +77,7 @@ class PipelineImplTest : public ::testing::Test { |
public: |
// Used for setting expectations on pipeline callbacks. Using a StrictMock |
// also lets us test for missing callbacks. |
- class CallbackHelper { |
+ class CallbackHelper : public Pipeline::Client { |
public: |
CallbackHelper() {} |
virtual ~CallbackHelper() {} |
@@ -87,11 +87,16 @@ class PipelineImplTest : public ::testing::Test { |
MOCK_METHOD1(OnSuspend, void(PipelineStatus)); |
MOCK_METHOD1(OnResume, void(PipelineStatus)); |
MOCK_METHOD0(OnStop, void()); |
- MOCK_METHOD0(OnEnded, void()); |
+ |
+ // Pipeline::Client overrides. |
MOCK_METHOD1(OnError, void(PipelineStatus)); |
+ MOCK_METHOD0(OnEnded, void()); |
MOCK_METHOD1(OnMetadata, void(PipelineMetadata)); |
MOCK_METHOD1(OnBufferingStateChange, void(BufferingState)); |
MOCK_METHOD0(OnDurationChange, void()); |
+ MOCK_METHOD2(OnAddTextTrack, |
+ void(const TextTrackConfig&, const AddTextTrackDoneCB&)); |
+ MOCK_METHOD0(OnWaitingForDecryptionKey, void()); |
private: |
DISALLOW_COPY_AND_ASSIGN(CallbackHelper); |
@@ -184,7 +189,7 @@ class PipelineImplTest : public ::testing::Test { |
} |
void AddTextStream() { |
- EXPECT_CALL(*this, OnAddTextTrack(_, _)) |
+ EXPECT_CALL(callbacks_, OnAddTextTrack(_, _)) |
.WillOnce(Invoke(this, &PipelineImplTest::DoOnAddTextTrack)); |
static_cast<DemuxerHost*>(pipeline_.get()) |
->AddTextStream(text_stream(), |
@@ -193,20 +198,10 @@ class PipelineImplTest : public ::testing::Test { |
} |
void StartPipeline() { |
- EXPECT_CALL(*this, OnWaitingForDecryptionKey()).Times(0); |
+ EXPECT_CALL(callbacks_, OnWaitingForDecryptionKey()).Times(0); |
pipeline_->Start( |
- demuxer_.get(), std::move(scoped_renderer_), |
- base::Bind(&CallbackHelper::OnEnded, base::Unretained(&callbacks_)), |
- base::Bind(&CallbackHelper::OnError, base::Unretained(&callbacks_)), |
- base::Bind(&CallbackHelper::OnStart, base::Unretained(&callbacks_)), |
- base::Bind(&CallbackHelper::OnMetadata, base::Unretained(&callbacks_)), |
- base::Bind(&CallbackHelper::OnBufferingStateChange, |
- base::Unretained(&callbacks_)), |
- base::Bind(&CallbackHelper::OnDurationChange, |
- base::Unretained(&callbacks_)), |
- base::Bind(&PipelineImplTest::OnAddTextTrack, base::Unretained(this)), |
- base::Bind(&PipelineImplTest::OnWaitingForDecryptionKey, |
- base::Unretained(this))); |
+ demuxer_.get(), std::move(scoped_renderer_), &callbacks_, |
+ base::Bind(&CallbackHelper::OnStart, base::Unretained(&callbacks_))); |
} |
// Sets up expectations on the callback and initializes the pipeline. Called |
@@ -334,10 +329,6 @@ class PipelineImplTest : public ::testing::Test { |
.WillOnce(Invoke(this, &PipelineImplTest::DestroyPipeline)); |
} |
- MOCK_METHOD2(OnAddTextTrack, |
- void(const TextTrackConfig&, const AddTextTrackDoneCB&)); |
- MOCK_METHOD0(OnWaitingForDecryptionKey, void(void)); |
- |
void DoOnAddTextTrack(const TextTrackConfig& config, |
const AddTextTrackDoneCB& done_cb) { |
std::unique_ptr<TextTrack> text_track(new MockTextTrack); |