| Index: components/suggestions/image_manager_unittest.cc
|
| diff --git a/components/suggestions/image_manager_unittest.cc b/components/suggestions/image_manager_unittest.cc
|
| deleted file mode 100644
|
| index 757b810f52a83442888c6d97ccbbbccdfff9e973..0000000000000000000000000000000000000000
|
| --- a/components/suggestions/image_manager_unittest.cc
|
| +++ /dev/null
|
| @@ -1,193 +0,0 @@
|
| -// 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;
|
| - bm.allocN32Pixels(2, 2);
|
| - 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
|
|
|