Index: media/base/pipeline_impl_unittest.cc |
diff --git a/media/base/pipeline_impl_unittest.cc b/media/base/pipeline_impl_unittest.cc |
index d08ecf1c173c58526069edb5d0c1cf23059e88ad..b0408d7774a0eee706454d36838d61559be58e31 100644 |
--- a/media/base/pipeline_impl_unittest.cc |
+++ b/media/base/pipeline_impl_unittest.cc |
@@ -76,7 +76,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() {} |
@@ -86,11 +86,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); |
@@ -183,7 +188,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(), |
@@ -192,20 +197,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 |
@@ -333,10 +328,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) { |
scoped_ptr<TextTrack> text_track(new MockTextTrack); |