[Fixed] [Suggestions] Move ImageManager to the component

components/suggestions/image_manager_unittest.cc

+// Copyright 2014 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 <string>
+
+#include "base/files/file_path.h"
+#include "base/run_loop.h"
+#include "components/leveldb_proto/proto_database.h"
+#include "components/leveldb_proto/testing/fake_db.h"
+#include "components/suggestions/image_fetcher.h"
+#include "components/suggestions/image_fetcher_delegate.h"
+#include "components/suggestions/image_manager.h"
+#include "components/suggestions/proto/suggestions.pb.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/gfx/image/image_skia.h"
+#include "url/gurl.h"
+
+using ::testing::Return;
+using ::testing::StrictMock;
+using ::testing::_;
+
+namespace suggestions {
+
+const char kTestUrl1[] = "http://go.com/";
+const char kTestUrl2[] = "http://goal.com/";
+const char kTestImagePath[] = "files/image_decoding/droids.png";
+const char kInvalidImagePath[] = "files/DOESNOTEXIST";
+
+using leveldb_proto::test::FakeDB;
+using suggestions::ImageData;
+using suggestions::ImageManager;
+
+typedef base::hash_map<std::string, ImageData> EntryMap;
+
+void AddEntry(const ImageData& d, EntryMap* map) { (*map)[d.url()] = d; }
+
+class MockImageFetcher : public suggestions::ImageFetcher {
+ public:
+  MockImageFetcher() {}
+  virtual ~MockImageFetcher() {}
+  MOCK_METHOD3(StartOrQueueNetworkRequest,
+               void(const GURL&, const GURL&,
+                    base::Callback<void(const GURL&, const SkBitmap*)>));
+  MOCK_METHOD1(SetImageFetcherDelegate, void(ImageFetcherDelegate*));
+};
+
+class ImageManagerTest : public testing::Test {
+ public:
+  ImageManagerTest()
+      : mock_image_fetcher_(NULL),
+        num_callback_null_called_(0),
+        num_callback_valid_called_(0) {}
+
+  virtual void SetUp() OVERRIDE {
+    fake_db_ = new FakeDB<ImageData>(&db_model_);
+    image_manager_.reset(CreateImageManager(fake_db_));
+  }
+
+  virtual void TearDown() OVERRIDE {
+    fake_db_ = NULL;
+    db_model_.clear();
+    image_manager_.reset();
+  }
+
+  void InitializeDefaultImageMapAndDatabase(ImageManager* image_manager,
+                                            FakeDB<ImageData>* fake_db) {
+    CHECK(image_manager);
+    CHECK(fake_db);
+
+    suggestions::SuggestionsProfile suggestions_profile;
+    suggestions::ChromeSuggestion* suggestion =
+        suggestions_profile.add_suggestions();
+    suggestion->set_url(kTestUrl1);
+    suggestion->set_thumbnail(kTestImagePath);
+
+    image_manager->Initialize(suggestions_profile);
+
+    // Initialize empty database.
+    fake_db->InitCallback(true);
+    fake_db->LoadCallback(true);
+  }
+
+  ImageData GetSampleImageData(const std::string& url) {
+    // Create test bitmap.
+    SkBitmap bm;
+    // Being careful with the Bitmap. There are memory-related issue in
+    // crbug.com/101781.
+    bm.allocN32Pixels(4, 4);
+    bm.eraseColor(SK_ColorRED);
+    ImageData data;
+    data.set_url(url);
+    std::vector<unsigned char> encoded;
+    EXPECT_TRUE(ImageManager::EncodeImage(bm, &encoded));
+    data.set_data(std::string(encoded.begin(), encoded.end()));
+    return data;
+  }
+
+  void OnImageAvailable(base::RunLoop* loop, const GURL& url,
+                        const SkBitmap* bitmap) {
+    if (bitmap) {
+      num_callback_valid_called_++;
+    } else {
+      num_callback_null_called_++;
+    }
+    loop->Quit();
+  }
+
+  ImageManager* CreateImageManager(FakeDB<ImageData>* fake_db) {
+    mock_image_fetcher_ = new StrictMock<MockImageFetcher>();
+    EXPECT_CALL(*mock_image_fetcher_, SetImageFetcherDelegate(_));
+    return new ImageManager(
+        scoped_ptr<ImageFetcher>(mock_image_fetcher_),
+        scoped_ptr<leveldb_proto::ProtoDatabase<ImageData> >(fake_db),
+        FakeDB<ImageData>::DirectoryForTestDB());
+  }
+
+  EntryMap db_model_;
+  // Owned by the ImageManager under test.
+  FakeDB<ImageData>* fake_db_;
+
+  MockImageFetcher* mock_image_fetcher_;
+
+  int num_callback_null_called_;
+  int num_callback_valid_called_;
+  // Under test.
+  scoped_ptr<ImageManager> image_manager_;
+};
+
+TEST_F(ImageManagerTest, InitializeTest) {
+  SuggestionsProfile suggestions_profile;
+  ChromeSuggestion* suggestion = suggestions_profile.add_suggestions();
+  suggestion->set_url(kTestUrl1);
+  suggestion->set_thumbnail(kTestImagePath);
+
+  image_manager_->Initialize(suggestions_profile);
+
+  GURL output;
+  EXPECT_TRUE(image_manager_->GetImageURL(GURL(kTestUrl1), &output));
+  EXPECT_EQ(GURL(kTestImagePath), output);
+
+  EXPECT_FALSE(image_manager_->GetImageURL(GURL("http://b.com"), &output));
+}
+
+TEST_F(ImageManagerTest, GetImageForURLNetwork) {
+  InitializeDefaultImageMapAndDatabase(image_manager_.get(), fake_db_);
+
+  // We expect the fetcher to go to network and call the callback.
+  EXPECT_CALL(*mock_image_fetcher_, StartOrQueueNetworkRequest(_, _, _));
+
+  // Fetch existing URL.
+  base::RunLoop run_loop;
+  image_manager_->GetImageForURL(GURL(kTestUrl1),
+                                 base::Bind(&ImageManagerTest::OnImageAvailable,
+                                            base::Unretained(this), &run_loop));
+
+  // Will not go to network and use the fetcher since URL is invalid.
+  // Fetch non-existing URL.
+  image_manager_->GetImageForURL(GURL(kTestUrl2),
+                                 base::Bind(&ImageManagerTest::OnImageAvailable,
+                                            base::Unretained(this), &run_loop));
+  run_loop.Run();
+
+  EXPECT_EQ(1, num_callback_null_called_);
+}
+
+TEST_F(ImageManagerTest, GetImageForURLNetworkCacheHit) {
+  SuggestionsProfile suggestions_profile;
+  ChromeSuggestion* suggestion = suggestions_profile.add_suggestions();
+  suggestion->set_url(kTestUrl1);
+  // The URL we set is invalid, to show that it will fail from network.
+  suggestion->set_thumbnail(kInvalidImagePath);
+
+  // Create the ImageManager with an added entry in the database.
+  AddEntry(GetSampleImageData(kTestUrl1), &db_model_);
+  FakeDB<ImageData>* fake_db = new FakeDB<ImageData>(&db_model_);
+  image_manager_.reset(CreateImageManager(fake_db));
+  image_manager_->Initialize(suggestions_profile);
+  fake_db->InitCallback(true);
+  fake_db->LoadCallback(true);
+  // Expect something in the cache.
+  SkBitmap* bitmap = image_manager_->GetBitmapFromCache(GURL(kTestUrl1));
+  EXPECT_FALSE(bitmap->isNull());
+
+  base::RunLoop run_loop;
+  image_manager_->GetImageForURL(GURL(kTestUrl1),
+                                 base::Bind(&ImageManagerTest::OnImageAvailable,
+                                            base::Unretained(this), &run_loop));
+  run_loop.Run();
+
+  EXPECT_EQ(0, num_callback_null_called_);
+  EXPECT_EQ(1, num_callback_valid_called_);
+}
+
+}  // namespace suggestions