[NTP Snippets] Cache images in a LevelDB

components/ntp_snippets/ntp_snippets_database_unittest.cc

 // Copyright 2016 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 "components/ntp_snippets/ntp_snippets_database.h"
 
 #include <memory>
 
 #include "base/bind.h"
 #include "base/callback.h"
@@ skipping 62 matching lines @@
     // Explicitly destroy any existing database first, so it releases the lock
     // on the file.
     db_.reset();
 
     db_.reset(new NTPSnippetsDatabase(database_dir_.path(),
                                       base::ThreadTaskRunnerHandle::Get()));
   }
 
   NTPSnippetsDatabase* db() { return db_.get(); }
 
-  bool db_inited() { return db_->database_initialized_; }
-
   void OnSnippetsLoaded(NTPSnippet::PtrVector snippets) {
     OnSnippetsLoadedImpl(snippets);
   }
 
   MOCK_METHOD1(OnSnippetsLoadedImpl,
                void(const NTPSnippet::PtrVector& snippets));
 
  private:
   base::MessageLoop message_loop_;
   base::ScopedTempDir database_dir_;
   std::unique_ptr<NTPSnippetsDatabase> db_;
 
   DISALLOW_COPY_AND_ASSIGN(NTPSnippetsDatabaseTest);
 };
 
 TEST_F(NTPSnippetsDatabaseTest, Init) {
   ASSERT_FALSE(db());
 
   CreateDatabase();
-  EXPECT_FALSE(db_inited());
+  EXPECT_FALSE(db()->IsInitialized());
 
   base::RunLoop().RunUntilIdle();
-  EXPECT_TRUE(db_inited());
+  EXPECT_TRUE(db()->IsInitialized());
 }
 
 TEST_F(NTPSnippetsDatabaseTest, LoadBeforeInit) {
   CreateDatabase();
-  EXPECT_FALSE(db_inited());
+  EXPECT_FALSE(db()->IsInitialized());
 
-  db()->Load(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
-                        base::Unretained(this)));
+  db()->LoadSnippets(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
+                                base::Unretained(this)));
 
   EXPECT_CALL(*this, OnSnippetsLoadedImpl(_));
   base::RunLoop().RunUntilIdle();
-  EXPECT_TRUE(db_inited());
+  EXPECT_TRUE(db()->IsInitialized());
 }
 
 TEST_F(NTPSnippetsDatabaseTest, LoadAfterInit) {
   CreateDatabase();
-  EXPECT_FALSE(db_inited());
+  EXPECT_FALSE(db()->IsInitialized());
 
   EXPECT_CALL(*this, OnSnippetsLoadedImpl(_)).Times(0);
   base::RunLoop().RunUntilIdle();
-  EXPECT_TRUE(db_inited());
+  EXPECT_TRUE(db()->IsInitialized());
 
   Mock::VerifyAndClearExpectations(this);
 
-  db()->Load(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
-                        base::Unretained(this)));
+  db()->LoadSnippets(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
+                                base::Unretained(this)));
 
   EXPECT_CALL(*this, OnSnippetsLoadedImpl(_));
   base::RunLoop().RunUntilIdle();
 }
 
 TEST_F(NTPSnippetsDatabaseTest, Save) {
   CreateDatabase();
   base::RunLoop().RunUntilIdle();
-  ASSERT_TRUE(db_inited());
+  ASSERT_TRUE(db()->IsInitialized());
 
   std::unique_ptr<NTPSnippet> snippet = CreateTestSnippet();
 
-  db()->Save(*snippet);
+  db()->SaveSnippet(*snippet);
   base::RunLoop().RunUntilIdle();
 
-  db()->Load(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
-                        base::Unretained(this)));
+  db()->LoadSnippets(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
+                                base::Unretained(this)));
 
   EXPECT_CALL(*this,
               OnSnippetsLoadedImpl(ElementsAre(SnippetEq(snippet.get()))));
   base::RunLoop().RunUntilIdle();
 
   Mock::VerifyAndClearExpectations(this);
 
   // The snippet should still exist after recreating the database.
   CreateDatabase();
 
-  db()->Load(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
-                        base::Unretained(this)));
+  db()->LoadSnippets(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
+                                base::Unretained(this)));
 
   EXPECT_CALL(*this,
               OnSnippetsLoadedImpl(ElementsAre(SnippetEq(snippet.get()))));
   base::RunLoop().RunUntilIdle();
 }
 
 TEST_F(NTPSnippetsDatabaseTest, Update) {
   CreateDatabase();
   base::RunLoop().RunUntilIdle();
-  ASSERT_TRUE(db_inited());
+  ASSERT_TRUE(db()->IsInitialized());
 
   std::unique_ptr<NTPSnippet> snippet = CreateTestSnippet();
 
-  db()->Save(*snippet);
+  db()->SaveSnippet(*snippet);
   base::RunLoop().RunUntilIdle();
 
   const std::string text("some text");
   snippet->set_snippet(text);
 
-  db()->Save(*snippet);
+  db()->SaveSnippet(*snippet);
   base::RunLoop().RunUntilIdle();
 
-  db()->Load(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
-                        base::Unretained(this)));
+  db()->LoadSnippets(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
+                                base::Unretained(this)));
 
   EXPECT_CALL(*this,
               OnSnippetsLoadedImpl(ElementsAre(SnippetEq(snippet.get()))));
   base::RunLoop().RunUntilIdle();
 }
 
 TEST_F(NTPSnippetsDatabaseTest, Delete) {
   CreateDatabase();
   base::RunLoop().RunUntilIdle();
-  ASSERT_TRUE(db_inited());
+  ASSERT_TRUE(db()->IsInitialized());
 
   std::unique_ptr<NTPSnippet> snippet = CreateTestSnippet();
 
-  db()->Save(*snippet);
+  db()->SaveSnippet(*snippet);
   base::RunLoop().RunUntilIdle();
 
-  db()->Delete(snippet->id());
+  db()->DeleteSnippet(snippet->id());
   base::RunLoop().RunUntilIdle();
 
-  db()->Load(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
-                        base::Unretained(this)));
+  db()->LoadSnippets(base::Bind(&NTPSnippetsDatabaseTest::OnSnippetsLoaded,
+                                base::Unretained(this)));
 
   EXPECT_CALL(*this, OnSnippetsLoadedImpl(IsEmpty()));
   base::RunLoop().RunUntilIdle();
 }
 
 }  // namespace ntp_snippets