Index: chrome/browser/search/suggestions/image_manager_impl_browsertest.cc |
diff --git a/chrome/browser/search/suggestions/thumbnail_manager_browsertest.cc b/chrome/browser/search/suggestions/image_manager_impl_browsertest.cc |
similarity index 63% |
rename from chrome/browser/search/suggestions/thumbnail_manager_browsertest.cc |
rename to chrome/browser/search/suggestions/image_manager_impl_browsertest.cc |
index 166f770051dd06e4576c8c7207b88832efd8f56e..fcbfb405d06342bae533a9b4d203641ab24d274d 100644 |
--- a/chrome/browser/search/suggestions/thumbnail_manager_browsertest.cc |
+++ b/chrome/browser/search/suggestions/image_manager_impl_browsertest.cc |
@@ -7,7 +7,7 @@ |
#include "base/files/file_path.h" |
#include "base/run_loop.h" |
#include "chrome/browser/profiles/profile.h" |
-#include "chrome/browser/search/suggestions/thumbnail_manager.h" |
+#include "chrome/browser/search/suggestions/image_manager_impl.h" |
#include "chrome/browser/ui/browser.h" |
#include "chrome/test/base/in_process_browser_test.h" |
#include "components/leveldb_proto/proto_database.h" |
@@ -34,16 +34,16 @@ const base::FilePath::CharType kDocRoot[] = |
using chrome::BitmapFetcher; |
using content::BrowserThread; |
using leveldb_proto::test::FakeDB; |
-using suggestions::ThumbnailData; |
-using suggestions::ThumbnailManager; |
+using suggestions::ImageData; |
+using suggestions::ImageManagerImpl; |
-typedef base::hash_map<std::string, ThumbnailData> EntryMap; |
+typedef base::hash_map<std::string, ImageData> EntryMap; |
-void AddEntry(const ThumbnailData& d, EntryMap* map) { (*map)[d.url()] = d; } |
+void AddEntry(const ImageData& d, EntryMap* map) { (*map)[d.url()] = d; } |
-class ThumbnailManagerBrowserTest : public InProcessBrowserTest { |
+class ImageManagerImplBrowserTest : public InProcessBrowserTest { |
public: |
- ThumbnailManagerBrowserTest() |
+ ImageManagerImplBrowserTest() |
: num_callback_null_called_(0), |
num_callback_valid_called_(0), |
test_server_(net::SpawnedTestServer::TYPE_HTTP, |
@@ -60,20 +60,20 @@ class ThumbnailManagerBrowserTest : public InProcessBrowserTest { |
} |
virtual void SetUpOnMainThread() OVERRIDE { |
- fake_db_ = new FakeDB<ThumbnailData>(&db_model_); |
- thumbnail_manager_.reset(CreateThumbnailManager(fake_db_)); |
+ fake_db_ = new FakeDB<ImageData>(&db_model_); |
+ image_manager_.reset(CreateImageManagerImpl(fake_db_)); |
} |
virtual void TearDownOnMainThread() OVERRIDE { |
fake_db_ = NULL; |
db_model_.clear(); |
- thumbnail_manager_.reset(); |
- test_thumbnail_manager_.reset(); |
+ image_manager_.reset(); |
+ test_image_manager_.reset(); |
} |
void InitializeTestBitmapData() { |
- FakeDB<ThumbnailData>* test_fake_db = new FakeDB<ThumbnailData>(&db_model_); |
- test_thumbnail_manager_.reset(CreateThumbnailManager(test_fake_db)); |
+ FakeDB<ImageData>* test_fake_db = new FakeDB<ImageData>(&db_model_); |
+ test_image_manager_.reset(CreateImageManagerImpl(test_fake_db)); |
suggestions::SuggestionsProfile suggestions_profile; |
suggestions::ChromeSuggestion* suggestion = |
@@ -81,7 +81,7 @@ class ThumbnailManagerBrowserTest : public InProcessBrowserTest { |
suggestion->set_url(kTestBitmapUrl); |
suggestion->set_thumbnail(test_server_.GetURL(kTestImagePath).spec()); |
- test_thumbnail_manager_->Initialize(suggestions_profile); |
+ test_image_manager_->Initialize(suggestions_profile); |
// Initialize empty database. |
test_fake_db->InitCallback(true); |
@@ -89,24 +89,24 @@ class ThumbnailManagerBrowserTest : public InProcessBrowserTest { |
base::RunLoop run_loop; |
// Fetch existing URL. |
- test_thumbnail_manager_->GetImageForURL( |
+ test_image_manager_->GetImageForURL( |
GURL(kTestBitmapUrl), |
- base::Bind(&ThumbnailManagerBrowserTest::OnTestThumbnailAvailable, |
+ base::Bind(&ImageManagerImplBrowserTest::OnTestImageAvailable, |
base::Unretained(this), &run_loop)); |
run_loop.Run(); |
} |
- void OnTestThumbnailAvailable(base::RunLoop* loop, const GURL& url, |
- const SkBitmap* bitmap) { |
+ void OnTestImageAvailable(base::RunLoop* loop, const GURL& url, |
+ const SkBitmap* bitmap) { |
CHECK(bitmap); |
// Copy the resource locally. |
test_bitmap_ = *bitmap; |
loop->Quit(); |
} |
- void InitializeDefaultThumbnailMapAndDatabase( |
- ThumbnailManager* thumbnail_manager, FakeDB<ThumbnailData>* fake_db) { |
- CHECK(thumbnail_manager); |
+ void InitializeDefaultImageMapAndDatabase( |
+ ImageManagerImpl* image_manager, FakeDB<ImageData>* fake_db) { |
+ CHECK(image_manager); |
CHECK(fake_db); |
suggestions::SuggestionsProfile suggestions_profile; |
@@ -115,69 +115,70 @@ class ThumbnailManagerBrowserTest : public InProcessBrowserTest { |
suggestion->set_url(kTestUrl1); |
suggestion->set_thumbnail(test_server_.GetURL(kTestImagePath).spec()); |
- thumbnail_manager->Initialize(suggestions_profile); |
+ image_manager->Initialize(suggestions_profile); |
// Initialize empty database. |
fake_db->InitCallback(true); |
fake_db->LoadCallback(true); |
} |
- ThumbnailData GetSampleThumbnailData(const std::string& url) { |
- ThumbnailData data; |
+ ImageData GetSampleImageData(const std::string& url) { |
+ ImageData data; |
data.set_url(url); |
std::vector<unsigned char> encoded; |
- EXPECT_TRUE(ThumbnailManager::EncodeThumbnail(test_bitmap_, &encoded)); |
+ EXPECT_TRUE(ImageManagerImpl::EncodeImage(test_bitmap_, &encoded)); |
data.set_data(std::string(encoded.begin(), encoded.end())); |
return data; |
} |
- void OnThumbnailAvailable(base::RunLoop* loop, const GURL& url, |
+ void OnImageAvailable(base::RunLoop* loop, const GURL& url, |
const SkBitmap* bitmap) { |
if (bitmap) { |
num_callback_valid_called_++; |
- /*std::vector<unsigned char> actual; |
+ std::vector<unsigned char> actual; |
std::vector<unsigned char> expected; |
- EXPECT_TRUE(ThumbnailManager::EncodeThumbnail(*bitmap, &actual)); |
- EXPECT_TRUE(ThumbnailManager::EncodeThumbnail(test_bitmap_, &expected)); |
+ EXPECT_TRUE(ImageManagerImpl::EncodeImage(*bitmap, &actual)); |
+ EXPECT_TRUE(ImageManagerImpl::EncodeImage(test_bitmap_, &expected)); |
// Check first 100 bytes. |
std::string actual_str(actual.begin(), actual.begin() + 100); |
std::string expected_str(expected.begin(), expected.begin() + 100); |
- EXPECT_EQ(expected_str, actual_str);*/ |
+ EXPECT_EQ(expected_str, actual_str); |
} else { |
num_callback_null_called_++; |
} |
loop->Quit(); |
} |
- ThumbnailManager* CreateThumbnailManager(FakeDB<ThumbnailData>* fake_db) { |
- return new ThumbnailManager( |
+ ImageManagerImpl* CreateImageManagerImpl(FakeDB<ImageData>* fake_db) { |
+ return new ImageManagerImpl( |
browser()->profile()->GetRequestContext(), |
- scoped_ptr<leveldb_proto::ProtoDatabase<ThumbnailData> >(fake_db), |
- FakeDB<ThumbnailData>::DirectoryForTestDB()); |
+ scoped_ptr<leveldb_proto::ProtoDatabase<ImageData> >(fake_db), |
+ FakeDB<ImageData>::DirectoryForTestDB()); |
} |
EntryMap db_model_; |
- // Owned by the ThumbnailManager under test. |
- FakeDB<ThumbnailData>* fake_db_; |
+ // Owned by the ImageManagerImpl under test. |
+ FakeDB<ImageData>* fake_db_; |
SkBitmap test_bitmap_; |
- scoped_ptr<ThumbnailManager> test_thumbnail_manager_; |
+ scoped_ptr<ImageManagerImpl> test_image_manager_; |
int num_callback_null_called_; |
int num_callback_valid_called_; |
net::SpawnedTestServer test_server_; |
// Under test. |
- scoped_ptr<ThumbnailManager> thumbnail_manager_; |
+ scoped_ptr<ImageManagerImpl> image_manager_; |
}; |
-IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, GetImageForURLNetwork) { |
- InitializeDefaultThumbnailMapAndDatabase(thumbnail_manager_.get(), fake_db_); |
+IN_PROC_BROWSER_TEST_F(ImageManagerImplBrowserTest, GetImageForURLNetwork) { |
+ InitializeTestBitmapData(); |
+ InitializeDefaultImageMapAndDatabase(image_manager_.get(), fake_db_); |
base::RunLoop run_loop; |
// Fetch existing URL. |
- thumbnail_manager_->GetImageForURL( |
+ image_manager_->GetImageForURL( |
GURL(kTestUrl1), |
- base::Bind(&ThumbnailManagerBrowserTest::OnThumbnailAvailable, |
+ base::Bind(&ImageManagerImplBrowserTest::OnImageAvailable, |
base::Unretained(this), &run_loop)); |
run_loop.Run(); |
@@ -186,9 +187,9 @@ IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, GetImageForURLNetwork) { |
base::RunLoop run_loop2; |
// Fetch non-existing URL. |
- thumbnail_manager_->GetImageForURL( |
+ image_manager_->GetImageForURL( |
GURL(kTestUrl2), |
- base::Bind(&ThumbnailManagerBrowserTest::OnThumbnailAvailable, |
+ base::Bind(&ImageManagerImplBrowserTest::OnImageAvailable, |
base::Unretained(this), &run_loop2)); |
run_loop2.Run(); |
@@ -196,17 +197,18 @@ IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, GetImageForURLNetwork) { |
EXPECT_EQ(1, num_callback_valid_called_); |
} |
-IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, |
+IN_PROC_BROWSER_TEST_F(ImageManagerImplBrowserTest, |
GetImageForURLNetworkMultiple) { |
- InitializeDefaultThumbnailMapAndDatabase(thumbnail_manager_.get(), fake_db_); |
+ InitializeTestBitmapData(); |
+ InitializeDefaultImageMapAndDatabase(image_manager_.get(), fake_db_); |
// Fetch non-existing URL, and add more while request is in flight. |
base::RunLoop run_loop; |
for (int i = 0; i < 5; i++) { |
// Fetch existing URL. |
- thumbnail_manager_->GetImageForURL( |
+ image_manager_->GetImageForURL( |
GURL(kTestUrl1), |
- base::Bind(&ThumbnailManagerBrowserTest::OnThumbnailAvailable, |
+ base::Bind(&ImageManagerImplBrowserTest::OnImageAvailable, |
base::Unretained(this), &run_loop)); |
} |
run_loop.Run(); |
@@ -215,24 +217,24 @@ IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, |
EXPECT_EQ(5, num_callback_valid_called_); |
} |
-IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, |
+IN_PROC_BROWSER_TEST_F(ImageManagerImplBrowserTest, |
GetImageForURLNetworkInvalid) { |
SuggestionsProfile suggestions_profile; |
ChromeSuggestion* suggestion = suggestions_profile.add_suggestions(); |
suggestion->set_url(kTestUrl1); |
suggestion->set_thumbnail(test_server_.GetURL(kInvalidImagePath).spec()); |
- thumbnail_manager_->Initialize(suggestions_profile); |
+ image_manager_->Initialize(suggestions_profile); |
// Database will be initialized and loaded without anything in it. |
fake_db_->InitCallback(true); |
fake_db_->LoadCallback(true); |
base::RunLoop run_loop; |
- // Fetch existing URL that has invalid thumbnail. |
- thumbnail_manager_->GetImageForURL( |
+ // Fetch existing URL that has invalid image. |
+ image_manager_->GetImageForURL( |
GURL(kTestUrl1), |
- base::Bind(&ThumbnailManagerBrowserTest::OnThumbnailAvailable, |
+ base::Bind(&ImageManagerImplBrowserTest::OnImageAvailable, |
base::Unretained(this), &run_loop)); |
run_loop.Run(); |
@@ -240,7 +242,7 @@ IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, |
EXPECT_EQ(0, num_callback_valid_called_); |
} |
-IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, |
+IN_PROC_BROWSER_TEST_F(ImageManagerImplBrowserTest, |
GetImageForURLNetworkCacheHit) { |
InitializeTestBitmapData(); |
@@ -250,21 +252,21 @@ IN_PROC_BROWSER_TEST_F(ThumbnailManagerBrowserTest, |
// The URL we set is invalid, to show that it will fail from network. |
suggestion->set_thumbnail(test_server_.GetURL(kInvalidImagePath).spec()); |
- // Create the ThumbnailManager with an added entry in the database. |
- AddEntry(GetSampleThumbnailData(kTestUrl1), &db_model_); |
- FakeDB<ThumbnailData>* fake_db = new FakeDB<ThumbnailData>(&db_model_); |
- thumbnail_manager_.reset(CreateThumbnailManager(fake_db)); |
- thumbnail_manager_->Initialize(suggestions_profile); |
+ // Create the ImageManagerImpl with an added entry in the database. |
+ AddEntry(GetSampleImageData(kTestUrl1), &db_model_); |
+ FakeDB<ImageData>* fake_db = new FakeDB<ImageData>(&db_model_); |
+ image_manager_.reset(CreateImageManagerImpl(fake_db)); |
+ image_manager_->Initialize(suggestions_profile); |
fake_db->InitCallback(true); |
fake_db->LoadCallback(true); |
// Expect something in the cache. |
- SkBitmap* bitmap = thumbnail_manager_->GetBitmapFromCache(GURL(kTestUrl1)); |
+ SkBitmap* bitmap = image_manager_->GetBitmapFromCache(GURL(kTestUrl1)); |
EXPECT_FALSE(bitmap->isNull()); |
base::RunLoop run_loop; |
- thumbnail_manager_->GetImageForURL( |
+ image_manager_->GetImageForURL( |
GURL(kTestUrl1), |
- base::Bind(&ThumbnailManagerBrowserTest::OnThumbnailAvailable, |
+ base::Bind(&ImageManagerImplBrowserTest::OnImageAvailable, |
base::Unretained(this), &run_loop)); |
run_loop.Run(); |