| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <string> | 5 #include <string> |
| 6 #include <vector> | |
| 7 | 6 |
| 8 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
| 9 #include "base/run_loop.h" | 8 #include "base/run_loop.h" |
| 10 #include "components/leveldb_proto/proto_database.h" | 9 #include "components/leveldb_proto/proto_database.h" |
| 11 #include "components/leveldb_proto/testing/fake_db.h" | 10 #include "components/leveldb_proto/testing/fake_db.h" |
| 12 #include "components/suggestions/image_encoder.h" | |
| 13 #include "components/suggestions/image_fetcher.h" | 11 #include "components/suggestions/image_fetcher.h" |
| 14 #include "components/suggestions/image_fetcher_delegate.h" | 12 #include "components/suggestions/image_fetcher_delegate.h" |
| 15 #include "components/suggestions/image_manager.h" | 13 #include "components/suggestions/image_manager.h" |
| 16 #include "components/suggestions/jpeg/jpeg_image_encoder.h" | |
| 17 #include "components/suggestions/proto/suggestions.pb.h" | 14 #include "components/suggestions/proto/suggestions.pb.h" |
| 18 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
| 19 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 20 #include "ui/gfx/image/image_skia.h" | 17 #include "ui/gfx/image/image_skia.h" |
| 21 #include "url/gurl.h" | 18 #include "url/gurl.h" |
| 22 | 19 |
| 23 using ::testing::Return; | 20 using ::testing::Return; |
| 24 using ::testing::StrictMock; | 21 using ::testing::StrictMock; |
| 25 using ::testing::_; | 22 using ::testing::_; |
| 26 | 23 |
| (...skipping 15 matching lines...) Expand all Loading... |
| 42 class MockImageFetcher : public suggestions::ImageFetcher { | 39 class MockImageFetcher : public suggestions::ImageFetcher { |
| 43 public: | 40 public: |
| 44 MockImageFetcher() {} | 41 MockImageFetcher() {} |
| 45 virtual ~MockImageFetcher() {} | 42 virtual ~MockImageFetcher() {} |
| 46 MOCK_METHOD3(StartOrQueueNetworkRequest, | 43 MOCK_METHOD3(StartOrQueueNetworkRequest, |
| 47 void(const GURL&, const GURL&, | 44 void(const GURL&, const GURL&, |
| 48 base::Callback<void(const GURL&, const SkBitmap*)>)); | 45 base::Callback<void(const GURL&, const SkBitmap*)>)); |
| 49 MOCK_METHOD1(SetImageFetcherDelegate, void(ImageFetcherDelegate*)); | 46 MOCK_METHOD1(SetImageFetcherDelegate, void(ImageFetcherDelegate*)); |
| 50 }; | 47 }; |
| 51 | 48 |
| 52 class MockImageEncoder : public suggestions::ImageEncoder { | |
| 53 public: | |
| 54 MockImageEncoder() {} | |
| 55 virtual ~MockImageEncoder() {} | |
| 56 MOCK_METHOD1(DecodeImage, | |
| 57 SkBitmap*(const std::vector<unsigned char>&)); | |
| 58 MOCK_METHOD2(EncodeImage, bool(const SkBitmap&, std::vector<unsigned char>*)); | |
| 59 }; | |
| 60 | |
| 61 class ImageManagerTest : public testing::Test { | 49 class ImageManagerTest : public testing::Test { |
| 62 public: | 50 public: |
| 63 ImageManagerTest() | 51 ImageManagerTest() |
| 64 : mock_image_fetcher_(NULL), | 52 : mock_image_fetcher_(NULL), |
| 65 num_callback_null_called_(0), | 53 num_callback_null_called_(0), |
| 66 num_callback_valid_called_(0) {} | 54 num_callback_valid_called_(0) {} |
| 67 | 55 |
| 68 virtual void SetUp() override { | 56 virtual void SetUp() override { |
| 69 fake_db_ = new FakeDB<ImageData>(&db_model_); | 57 fake_db_ = new FakeDB<ImageData>(&db_model_); |
| 70 jpeg_image_encoder_ = new JpegImageEncoder(); | 58 image_manager_.reset(CreateImageManager(fake_db_)); |
| 71 image_manager_.reset(CreateImageManager(fake_db_, jpeg_image_encoder_)); | |
| 72 } | 59 } |
| 73 | 60 |
| 74 virtual void TearDown() override { | 61 virtual void TearDown() override { |
| 75 fake_db_ = NULL; | 62 fake_db_ = NULL; |
| 76 db_model_.clear(); | 63 db_model_.clear(); |
| 77 image_manager_.reset(); | 64 image_manager_.reset(); |
| 78 } | 65 } |
| 79 | 66 |
| 80 void InitializeDefaultImageMapAndDatabase(ImageManager* image_manager, | 67 void InitializeDefaultImageMapAndDatabase(ImageManager* image_manager, |
| 81 FakeDB<ImageData>* fake_db) { | 68 FakeDB<ImageData>* fake_db) { |
| (...skipping 16 matching lines...) Expand all Loading... |
| 98 ImageData GetSampleImageData(const std::string& url) { | 85 ImageData GetSampleImageData(const std::string& url) { |
| 99 // Create test bitmap. | 86 // Create test bitmap. |
| 100 SkBitmap bm; | 87 SkBitmap bm; |
| 101 // Being careful with the Bitmap. There are memory-related issue in | 88 // Being careful with the Bitmap. There are memory-related issue in |
| 102 // crbug.com/101781. | 89 // crbug.com/101781. |
| 103 bm.allocN32Pixels(4, 4); | 90 bm.allocN32Pixels(4, 4); |
| 104 bm.eraseColor(SK_ColorRED); | 91 bm.eraseColor(SK_ColorRED); |
| 105 ImageData data; | 92 ImageData data; |
| 106 data.set_url(url); | 93 data.set_url(url); |
| 107 std::vector<unsigned char> encoded; | 94 std::vector<unsigned char> encoded; |
| 108 EXPECT_TRUE(jpeg_image_encoder_->EncodeImage(bm, &encoded)); | 95 EXPECT_TRUE(ImageManager::EncodeImage(bm, &encoded)); |
| 109 data.set_data(std::string(encoded.begin(), encoded.end())); | 96 data.set_data(std::string(encoded.begin(), encoded.end())); |
| 110 return data; | 97 return data; |
| 111 } | 98 } |
| 112 | 99 |
| 113 void OnImageAvailable(base::RunLoop* loop, const GURL& url, | 100 void OnImageAvailable(base::RunLoop* loop, const GURL& url, |
| 114 const SkBitmap* bitmap) { | 101 const SkBitmap* bitmap) { |
| 115 if (bitmap) { | 102 if (bitmap) { |
| 116 num_callback_valid_called_++; | 103 num_callback_valid_called_++; |
| 117 } else { | 104 } else { |
| 118 num_callback_null_called_++; | 105 num_callback_null_called_++; |
| 119 } | 106 } |
| 120 loop->Quit(); | 107 loop->Quit(); |
| 121 } | 108 } |
| 122 | 109 |
| 123 ImageManager* CreateImageManager(FakeDB<ImageData>* fake_db, | 110 ImageManager* CreateImageManager(FakeDB<ImageData>* fake_db) { |
| 124 JpegImageEncoder* jpeg_image_encoder) { | |
| 125 mock_image_fetcher_ = new StrictMock<MockImageFetcher>(); | 111 mock_image_fetcher_ = new StrictMock<MockImageFetcher>(); |
| 126 EXPECT_CALL(*mock_image_fetcher_, SetImageFetcherDelegate(_)); | 112 EXPECT_CALL(*mock_image_fetcher_, SetImageFetcherDelegate(_)); |
| 127 return new ImageManager( | 113 return new ImageManager( |
| 128 scoped_ptr<ImageFetcher>(mock_image_fetcher_), | 114 scoped_ptr<ImageFetcher>(mock_image_fetcher_), |
| 129 scoped_ptr<ImageEncoder>(jpeg_image_encoder), | |
| 130 scoped_ptr<leveldb_proto::ProtoDatabase<ImageData> >(fake_db), | 115 scoped_ptr<leveldb_proto::ProtoDatabase<ImageData> >(fake_db), |
| 131 FakeDB<ImageData>::DirectoryForTestDB()); | 116 FakeDB<ImageData>::DirectoryForTestDB()); |
| 132 } | 117 } |
| 133 | 118 |
| 134 EntryMap db_model_; | 119 EntryMap db_model_; |
| 135 // Owned by the ImageManager under test. | 120 // Owned by the ImageManager under test. |
| 136 FakeDB<ImageData>* fake_db_; | 121 FakeDB<ImageData>* fake_db_; |
| 137 JpegImageEncoder* jpeg_image_encoder_; | |
| 138 | 122 |
| 139 MockImageFetcher* mock_image_fetcher_; | 123 MockImageFetcher* mock_image_fetcher_; |
| 140 | 124 |
| 141 int num_callback_null_called_; | 125 int num_callback_null_called_; |
| 142 int num_callback_valid_called_; | 126 int num_callback_valid_called_; |
| 143 // Under test. | 127 // Under test. |
| 144 scoped_ptr<ImageManager> image_manager_; | 128 scoped_ptr<ImageManager> image_manager_; |
| 145 }; | 129 }; |
| 146 | 130 |
| 147 TEST_F(ImageManagerTest, InitializeTest) { | 131 TEST_F(ImageManagerTest, InitializeTest) { |
| (...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 184 TEST_F(ImageManagerTest, GetImageForURLNetworkCacheHit) { | 168 TEST_F(ImageManagerTest, GetImageForURLNetworkCacheHit) { |
| 185 SuggestionsProfile suggestions_profile; | 169 SuggestionsProfile suggestions_profile; |
| 186 ChromeSuggestion* suggestion = suggestions_profile.add_suggestions(); | 170 ChromeSuggestion* suggestion = suggestions_profile.add_suggestions(); |
| 187 suggestion->set_url(kTestUrl1); | 171 suggestion->set_url(kTestUrl1); |
| 188 // The URL we set is invalid, to show that it will fail from network. | 172 // The URL we set is invalid, to show that it will fail from network. |
| 189 suggestion->set_thumbnail(kInvalidImagePath); | 173 suggestion->set_thumbnail(kInvalidImagePath); |
| 190 | 174 |
| 191 // Create the ImageManager with an added entry in the database. | 175 // Create the ImageManager with an added entry in the database. |
| 192 AddEntry(GetSampleImageData(kTestUrl1), &db_model_); | 176 AddEntry(GetSampleImageData(kTestUrl1), &db_model_); |
| 193 FakeDB<ImageData>* fake_db = new FakeDB<ImageData>(&db_model_); | 177 FakeDB<ImageData>* fake_db = new FakeDB<ImageData>(&db_model_); |
| 194 image_manager_.reset(CreateImageManager(fake_db, new JpegImageEncoder())); | 178 image_manager_.reset(CreateImageManager(fake_db)); |
| 195 image_manager_->Initialize(suggestions_profile); | 179 image_manager_->Initialize(suggestions_profile); |
| 196 fake_db->InitCallback(true); | 180 fake_db->InitCallback(true); |
| 197 fake_db->LoadCallback(true); | 181 fake_db->LoadCallback(true); |
| 198 // Expect something in the cache. | 182 // Expect something in the cache. |
| 199 SkBitmap* bitmap = image_manager_->GetBitmapFromCache(GURL(kTestUrl1)); | 183 SkBitmap* bitmap = image_manager_->GetBitmapFromCache(GURL(kTestUrl1)); |
| 200 EXPECT_FALSE(bitmap->isNull()); | 184 EXPECT_FALSE(bitmap->isNull()); |
| 201 | 185 |
| 202 base::RunLoop run_loop; | 186 base::RunLoop run_loop; |
| 203 image_manager_->GetImageForURL(GURL(kTestUrl1), | 187 image_manager_->GetImageForURL(GURL(kTestUrl1), |
| 204 base::Bind(&ImageManagerTest::OnImageAvailable, | 188 base::Bind(&ImageManagerTest::OnImageAvailable, |
| 205 base::Unretained(this), &run_loop)); | 189 base::Unretained(this), &run_loop)); |
| 206 run_loop.Run(); | 190 run_loop.Run(); |
| 207 | 191 |
| 208 EXPECT_EQ(0, num_callback_null_called_); | 192 EXPECT_EQ(0, num_callback_null_called_); |
| 209 EXPECT_EQ(1, num_callback_valid_called_); | 193 EXPECT_EQ(1, num_callback_valid_called_); |
| 210 } | 194 } |
| 211 | 195 |
| 212 } // namespace suggestions | 196 } // namespace suggestions |
| OLD | NEW |