Index: media/base/mock_callback.cc |
diff --git a/media/base/mock_callback.cc b/media/base/mock_callback.cc |
index eeedd15672d53b133436338418605178a1a44aa2..cc4f141c1f9c7ab228662e33dafde4bb3c4c4bc7 100644 |
--- a/media/base/mock_callback.cc |
+++ b/media/base/mock_callback.cc |
@@ -4,6 +4,8 @@ |
#include "media/base/mock_callback.h" |
+#include "base/bind.h" |
+ |
using ::testing::_; |
using ::testing::StrictMock; |
@@ -51,4 +53,22 @@ MockStatusCallback* NewExpectedStatusCallback(PipelineStatus status) { |
return callback; |
} |
+class MockStatusCB : public base::RefCountedThreadSafe<MockStatusCB> { |
+ public: |
+ MockStatusCB() {} |
+ virtual ~MockStatusCB() {} |
+ |
+ MOCK_METHOD1(Run, void(PipelineStatus)); |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(MockStatusCB); |
+}; |
+ |
+base::Callback<void(PipelineStatus)> NewExpectedStatusCB( |
+ PipelineStatus status) { |
+ StrictMock<MockStatusCB>* callback = new StrictMock<MockStatusCB>(); |
+ EXPECT_CALL(*callback, Run(status)); |
+ return base::Bind(&MockStatusCB::Run, callback); |
+} |
+ |
} // namespace media |