Revert 96974 - Remove mock_ffmpeg and update media unittests.

media/filters/ffmpeg_glue_unittest.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
+#include "media/base/mock_ffmpeg.h"
 #include "media/base/mock_filters.h"
 #include "media/ffmpeg/ffmpeg_common.h"
 #include "media/filters/ffmpeg_glue.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using ::testing::_;
 using ::testing::DoAll;
 using ::testing::InSequence;
 using ::testing::Return;
 using ::testing::SetArgumentPointee;
@@ skipping 11 matching lines @@
   MOCK_METHOD1(SetPosition, bool(int64 position));
   MOCK_METHOD1(GetSize, bool(int64* size_out));
   MOCK_METHOD0(IsStreaming, bool());
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockProtocol);
 };
 
 class FFmpegGlueTest : public ::testing::Test {
  public:
-  FFmpegGlueTest() : protocol_(NULL) {}
+  FFmpegGlueTest() {}
 
-  static void SetUpTestCase() {
+  virtual void SetUp() {
     // Singleton should initialize FFmpeg.
     CHECK(FFmpegGlue::GetInstance());
-  }
 
-  virtual void SetUp() {
     // Assign our static copy of URLProtocol for the rest of the tests.
-    protocol_ = FFmpegGlue::url_protocol();
+    protocol_ = MockFFmpeg::protocol();
     CHECK(protocol_);
   }
 
-  MOCK_METHOD1(CheckPoint, void(int val));
-
   // Helper to open a URLContext pointing to the given mocked protocol.
   // Callers are expected to close the context at the end of their test.
   virtual void OpenContext(MockProtocol* protocol, URLContext* context) {
     // IsStreaming() is called when opening.
     EXPECT_CALL(*protocol, IsStreaming()).WillOnce(Return(true));
 
     // Add the protocol to the glue layer and open a context.
     std::string key = FFmpegGlue::GetInstance()->AddProtocol(protocol);
     memset(context, 0, sizeof(*context));
     EXPECT_EQ(0, protocol_->url_open(context, key.c_str(), 0));
     FFmpegGlue::GetInstance()->RemoveProtocol(protocol);
   }
 
  protected:
   // Fixture members.
-  URLProtocol* protocol_;
+  MockFFmpeg mock_ffmpeg_;
+  static URLProtocol* protocol_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(FFmpegGlueTest);
 };
 
+URLProtocol* FFmpegGlueTest::protocol_ = NULL;
+
 TEST_F(FFmpegGlueTest, InitializeFFmpeg) {
   // Make sure URLProtocol was filled out correctly.
   EXPECT_STREQ("http", protocol_->name);
   EXPECT_TRUE(protocol_->url_close);
   EXPECT_TRUE(protocol_->url_open);
   EXPECT_TRUE(protocol_->url_read);
   EXPECT_TRUE(protocol_->url_seek);
   EXPECT_TRUE(protocol_->url_write);
 }
 
@@ skipping 27 matching lines @@
   std::string key_a2 = glue->AddProtocol(protocol_a.get());
   EXPECT_EQ(key_a, key_a2);
   glue->GetProtocol(key_a2, &protocol_c);
   EXPECT_EQ(protocol_a.get(), protocol_c);
 
   // Removes the protocols then releases our references.  They should be
   // destroyed.
   InSequence s;
   EXPECT_CALL(*protocol_a, OnDestroy());
   EXPECT_CALL(*protocol_b, OnDestroy());
-  EXPECT_CALL(*this, CheckPoint(0));
+  EXPECT_CALL(mock_ffmpeg_, CheckPoint(0));
 
   glue->RemoveProtocol(protocol_a.get());
   glue->GetProtocol(key_a, &protocol_c);
   EXPECT_FALSE(protocol_c);
   glue->GetProtocol(key_b, &protocol_d);
   EXPECT_EQ(protocol_b.get(), protocol_d);
   glue->RemoveProtocol(protocol_b.get());
   glue->GetProtocol(key_b, &protocol_d);
   EXPECT_FALSE(protocol_d);
   protocol_a.reset();
   protocol_b.reset();
 
   // Data sources should be deleted by this point.
-  CheckPoint(0);
+  mock_ffmpeg_.CheckPoint(0);
 }
 
 TEST_F(FFmpegGlueTest, OpenClose) {
   // Prepare testing data.
   FFmpegGlue* glue = FFmpegGlue::GetInstance();
 
   // Create our protocol and add them to the glue layer.
   scoped_ptr<StrictMock<Destroyable<MockProtocol> > > protocol(
       new StrictMock<Destroyable<MockProtocol> >());
   EXPECT_CALL(*protocol, IsStreaming()).WillOnce(Return(true));
   std::string key = glue->AddProtocol(protocol.get());
 
   // Prepare FFmpeg URLContext structure.
   URLContext context;
   memset(&context, 0, sizeof(context));
 
   // Test opening a URLContext with a protocol that doesn't exist.
   EXPECT_EQ(AVERROR(EIO), protocol_->url_open(&context, "foobar", 0));
 
   // Test opening a URLContext with our protocol.
   EXPECT_EQ(0, protocol_->url_open(&context, key.c_str(), 0));
   EXPECT_EQ(URL_RDONLY, context.flags);
   EXPECT_EQ(protocol.get(), context.priv_data);
   EXPECT_TRUE(context.is_streamed);
 
   // We're going to remove references one by one until the last reference is
   // held by FFmpeg.  Once we close the URLContext, the protocol should be
   // destroyed.
   InSequence s;
-  EXPECT_CALL(*this, CheckPoint(0));
-  EXPECT_CALL(*this, CheckPoint(1));
+  EXPECT_CALL(mock_ffmpeg_, CheckPoint(0));
+  EXPECT_CALL(mock_ffmpeg_, CheckPoint(1));
   EXPECT_CALL(*protocol, OnDestroy());
-  EXPECT_CALL(*this, CheckPoint(2));
+  EXPECT_CALL(mock_ffmpeg_, CheckPoint(2));
 
   // Remove the protocol from the glue layer, releasing a reference.
   glue->RemoveProtocol(protocol.get());
-  CheckPoint(0);
+  mock_ffmpeg_.CheckPoint(0);
 
   // Remove our own reference -- URLContext should maintain a reference.
-  CheckPoint(1);
+  mock_ffmpeg_.CheckPoint(1);
   protocol.reset();
 
   // Close the URLContext, which should release the final reference.
   EXPECT_EQ(0, protocol_->url_close(&context));
-  CheckPoint(2);
+  mock_ffmpeg_.CheckPoint(2);
 }
 
 TEST_F(FFmpegGlueTest, Write) {
   scoped_ptr<StrictMock<MockProtocol> > protocol(
       new StrictMock<MockProtocol>());
   URLContext context;
   OpenContext(protocol.get(), &context);
 
   const int kBufferSize = 16;
   uint8 buffer[kBufferSize];
@@ skipping 111 matching lines @@
 
 TEST_F(FFmpegGlueTest, Destroy) {
   // Create our protocol and add them to the glue layer.
   scoped_ptr<StrictMock<Destroyable<MockProtocol> > > protocol(
       new StrictMock<Destroyable<MockProtocol> >());
   std::string key = FFmpegGlue::GetInstance()->AddProtocol(protocol.get());
 
   // We should expect the protocol to get destroyed when the unit test
   // exits.
   InSequence s;
-  EXPECT_CALL(*this, CheckPoint(0));
+  EXPECT_CALL(mock_ffmpeg_, CheckPoint(0));
   EXPECT_CALL(*protocol, OnDestroy());
 
   // Remove our own reference, we shouldn't be destroyed yet.
-  CheckPoint(0);
+  mock_ffmpeg_.CheckPoint(0);
   protocol.reset();
 
   // ~FFmpegGlue() will be called when this unit test finishes execution.  By
   // leaving something inside FFmpegGlue's map we get to test our cleanup code.
 }
 
 }  // namespace media