IndexedDB: Using a mock IndexedDBFactory for unit tests

content/browser/indexed_db/indexed_db_database_unittest.cc

 // Copyright 2013 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 "content/browser/indexed_db/indexed_db_database.h"
 
 #include "base/auto_reset.h"
 #include "base/logging.h"
 #include "base/run_loop.h"
 #include "base/strings/string16.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/browser/indexed_db/indexed_db.h"
 #include "content/browser/indexed_db/indexed_db_backing_store.h"
 #include "content/browser/indexed_db/indexed_db_callbacks.h"
 #include "content/browser/indexed_db/indexed_db_connection.h"
 #include "content/browser/indexed_db/indexed_db_cursor.h"
 #include "content/browser/indexed_db/indexed_db_database.h"
 #include "content/browser/indexed_db/indexed_db_factory.h"
 #include "content/browser/indexed_db/indexed_db_fake_backing_store.h"
 #include "content/browser/indexed_db/indexed_db_transaction.h"
 #include "content/browser/indexed_db/indexed_db_value.h"
 #include "content/browser/indexed_db/mock_indexed_db_callbacks.h"
 #include "content/browser/indexed_db/mock_indexed_db_database_callbacks.h"
+#include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::ASCIIToUTF16;
 
 namespace {
 const int kFakeChildProcessId = 0;
-}
+
+class MockIDBFactory : public content::IndexedDBFactory {
+ public:
+  MOCK_METHOD2(ReleaseDatabase,
+               void(const content::IndexedDBDatabase::Identifier& identifier,
+                    bool forcedClose));
+  MOCK_METHOD4(GetDatabaseNames,
+               void(scoped_refptr<content::IndexedDBCallbacks> callbacks,
+                    const GURL& origin_url,
+                    const base::FilePath& data_directory,
+                    net::URLRequestContext* request_context));
+  MOCK_METHOD5(Open,
+               void(const base::string16& name,
+                    const content::IndexedDBPendingConnection& connection,
+                    net::URLRequestContext* request_context,
+                    const GURL& origin_url,
+                    const base::FilePath& data_directory));
+  MOCK_METHOD5(DeleteDatabase,
+               void(const base::string16& name,
+                    net::URLRequestContext* request_context,
+                    scoped_refptr<content::IndexedDBCallbacks> callbacks,
+                    const GURL& origin_url,
+                    const base::FilePath& data_directory));
+  MOCK_METHOD1(HandleBackingStoreFailure, void(const GURL& origin_url));
+  MOCK_METHOD2(HandleBackingStoreCorruption,
+               void(const GURL& origin_url,
+                    const content::IndexedDBDatabaseError& error));
+  // mock template gets confused and needs typedef to compile.
+  typedef std::pair<OriginDBMapIterator, OriginDBMapIterator> OriginDBs;
+  MOCK_CONST_METHOD1(GetOpenDatabasesForOrigin,
+                     OriginDBs(const GURL& origin_url));
+  MOCK_METHOD1(ForceClose, void(const GURL& origin_url));
+  MOCK_METHOD0(ContextDestroyed, void());
+  MOCK_METHOD1(DatabaseDeleted,
+               void(const content::IndexedDBDatabase::Identifier& identifier));
+  MOCK_CONST_METHOD1(GetConnectionCount, size_t(const GURL& origin_url));
+  MOCK_METHOD2(ReportOutstandingBlobs,
+               void(const GURL& origin_url, bool blobs_outstanding));
+
+ protected:
+  virtual ~MockIDBFactory() {}
+
+  MOCK_METHOD6(OpenBackingStore,
+               scoped_refptr<content::IndexedDBBackingStore>(
+                   const GURL& origin_url,
+                   const base::FilePath& data_directory,
+                   net::URLRequestContext* request_context,
+                   blink::WebIDBDataLoss* data_loss,
+                   std::string* data_loss_reason,
+                   bool* disk_full));
+
+  MOCK_METHOD7(OpenBackingStoreHelper,
+               scoped_refptr<content::IndexedDBBackingStore>(
+                   const GURL& origin_url,
+                   const base::FilePath& data_directory,
+                   net::URLRequestContext* request_context,
+                   blink::WebIDBDataLoss* data_loss,
+                   std::string* data_loss_message,
+                   bool* disk_full,
+                   bool first_time));
+
+ private:
+  MOCK_CONST_METHOD1(IsBackingStoreOpen, bool(const GURL& origin_url));
+};
+
+}  // anonymous namespace
 
 namespace content {
 
 TEST(IndexedDBDatabaseTest, BackingStoreRetention) {
   scoped_refptr<IndexedDBFakeBackingStore> backing_store =
       new IndexedDBFakeBackingStore();
   EXPECT_TRUE(backing_store->HasOneRef());
 
-  IndexedDBFactory* factory = 0;
+  scoped_refptr<MockIDBFactory> factory = new MockIDBFactory();
   leveldb::Status s;
   scoped_refptr<IndexedDBDatabase> db =
       IndexedDBDatabase::Create(ASCIIToUTF16("db"),
                                 backing_store,
                                 factory,
                                 IndexedDBDatabase::Identifier(),
                                 &s);
   ASSERT_TRUE(s.ok());
   EXPECT_FALSE(backing_store->HasOneRef());  // local and db
   db = NULL;
   EXPECT_TRUE(backing_store->HasOneRef());  // local
 }
 
 TEST(IndexedDBDatabaseTest, ConnectionLifecycle) {
   scoped_refptr<IndexedDBFakeBackingStore> backing_store =
       new IndexedDBFakeBackingStore();
   EXPECT_TRUE(backing_store->HasOneRef());  // local
 
-  IndexedDBFactory* factory = 0;
+  scoped_refptr<MockIDBFactory> factory = new MockIDBFactory();
   leveldb::Status s;
   scoped_refptr<IndexedDBDatabase> db =
       IndexedDBDatabase::Create(ASCIIToUTF16("db"),
                                 backing_store,
                                 factory,
                                 IndexedDBDatabase::Identifier(),
                                 &s);
   ASSERT_TRUE(s.ok());
   EXPECT_FALSE(backing_store->HasOneRef());  // local and db
 
@@ skipping 37 matching lines @@
   EXPECT_FALSE(db->backing_store());
 
   db = NULL;
 }
 
 TEST(IndexedDBDatabaseTest, ForcedClose) {
   scoped_refptr<IndexedDBFakeBackingStore> backing_store =
       new IndexedDBFakeBackingStore();
   EXPECT_TRUE(backing_store->HasOneRef());
 
-  IndexedDBFactory* factory = 0;
+  scoped_refptr<MockIDBFactory> factory = new MockIDBFactory();
   leveldb::Status s;
   scoped_refptr<IndexedDBDatabase> database =
       IndexedDBDatabase::Create(ASCIIToUTF16("db"),
                                 backing_store,
                                 factory,
                                 IndexedDBDatabase::Identifier(),
                                 &s);
   ASSERT_TRUE(s.ok());
   EXPECT_FALSE(backing_store->HasOneRef());  // local and db
 
@@ skipping 45 matching lines @@
   bool success_called_;
 
   DISALLOW_COPY_AND_ASSIGN(MockDeleteCallbacks);
 };
 
 TEST(IndexedDBDatabaseTest, PendingDelete) {
   scoped_refptr<IndexedDBFakeBackingStore> backing_store =
       new IndexedDBFakeBackingStore();
   EXPECT_TRUE(backing_store->HasOneRef());  // local
 
-  IndexedDBFactory* factory = 0;
+  scoped_refptr<MockIDBFactory> factory = new MockIDBFactory();
   leveldb::Status s;
   scoped_refptr<IndexedDBDatabase> db =
       IndexedDBDatabase::Create(ASCIIToUTF16("db"),
                                 backing_store,
                                 factory,
                                 IndexedDBDatabase::Identifier(),
                                 &s);
   ASSERT_TRUE(s.ok());
   EXPECT_FALSE(backing_store->HasOneRef());  // local and db
 
@@ skipping 22 matching lines @@
   EXPECT_FALSE(db->backing_store());
   EXPECT_TRUE(backing_store->HasOneRef());  // local
   EXPECT_TRUE(request2->success_called());
 }
 
 void DummyOperation(IndexedDBTransaction* transaction) {
 }
 
 class IndexedDBDatabaseOperationTest : public testing::Test {
  public:
-  IndexedDBDatabaseOperationTest() : commit_success_(leveldb::Status::OK()) {}
+  IndexedDBDatabaseOperationTest()
+      : commit_success_(leveldb::Status::OK()),
+        factory_(new MockIDBFactory()) {}
 
   virtual void SetUp() {
     backing_store_ = new IndexedDBFakeBackingStore();
     leveldb::Status s;
     db_ = IndexedDBDatabase::Create(ASCIIToUTF16("db"),
                                     backing_store_,
-                                    NULL /*factory*/,
+                                    factory_,
                                     IndexedDBDatabase::Identifier(),
                                     &s);
     ASSERT_TRUE(s.ok());
 
     request_ = new MockIndexedDBCallbacks();
     callbacks_ = new MockIndexedDBDatabaseCallbacks();
     const int64 transaction_id = 1;
     db_->OpenConnection(IndexedDBPendingConnection(
         request_,
         callbacks_,
@@ skipping 24 matching lines @@
   scoped_refptr<IndexedDBFakeBackingStore> backing_store_;
   scoped_refptr<IndexedDBDatabase> db_;
   scoped_refptr<MockIndexedDBCallbacks> request_;
   scoped_refptr<MockIndexedDBDatabaseCallbacks> callbacks_;
   scoped_refptr<IndexedDBTransaction> transaction_;
 
   leveldb::Status commit_success_;
 
  private:
   base::MessageLoop message_loop_;
+  scoped_refptr<MockIDBFactory> factory_;
 
   DISALLOW_COPY_AND_ASSIGN(IndexedDBDatabaseOperationTest);
 };
 
 TEST_F(IndexedDBDatabaseOperationTest, CreateObjectStore) {
   EXPECT_EQ(0ULL, db_->metadata().object_stores.size());
   const int64 store_id = 1001;
   db_->CreateObjectStore(transaction_->id(),
                          store_id,
                          ASCIIToUTF16("store"),
@@ skipping 118 matching lines @@
   EXPECT_EQ(1ULL, db_->metadata().object_stores.size());
 
   // This will execute the Put then Delete.
   RunPostedTasks();
   EXPECT_EQ(0ULL, db_->metadata().object_stores.size());
 
   transaction_->Commit(); // Cleans up the object hierarchy.
 }
 
 }  // namespace content