Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(15)

Side by Side Diff: components/suggestions/image_manager_unittest.cc

Issue 634273002: Revert of [Suggestions] Create ImageEncoder, to abstract away image encode/decode. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 6 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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
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
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
OLDNEW
« no previous file with comments | « components/suggestions/image_manager.cc ('k') | components/suggestions/jpeg/jpeg_image_encoder.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698