| Index: components/suggestions/image_manager_unittest.cc
|
| diff --git a/components/suggestions/image_manager_unittest.cc b/components/suggestions/image_manager_unittest.cc
|
| index e9d16e31ceaf642dc935ff7c9e2ff87662f8f16c..01b0f1eb080377b5dcb35f09521c4422cbc94978 100644
|
| --- a/components/suggestions/image_manager_unittest.cc
|
| +++ b/components/suggestions/image_manager_unittest.cc
|
| @@ -13,6 +13,7 @@
|
| #include "components/suggestions/image_fetcher_delegate.h"
|
| #include "components/suggestions/image_manager.h"
|
| #include "components/suggestions/proto/suggestions.pb.h"
|
| +#include "content/public/test/test_browser_thread.h"
|
| #include "testing/gmock/include/gmock/gmock.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| #include "ui/gfx/image/image_skia.h"
|
| @@ -21,6 +22,7 @@
|
| using ::testing::Return;
|
| using ::testing::StrictMock;
|
| using ::testing::_;
|
| +using content::BrowserThread;
|
|
|
| namespace suggestions {
|
|
|
| @@ -52,7 +54,10 @@ class ImageManagerTest : public testing::Test {
|
| ImageManagerTest()
|
| : mock_image_fetcher_(NULL),
|
| num_callback_null_called_(0),
|
| - num_callback_valid_called_(0) {}
|
| + num_callback_valid_called_(0),
|
| + db_thread_(BrowserThread::DB) {
|
| + db_thread_.Start();
|
| + }
|
|
|
| void SetUp() override {
|
| fake_db_ = new FakeDB<ImageData>(&db_model_);
|
| @@ -125,6 +130,10 @@ class ImageManagerTest : public testing::Test {
|
|
|
| int num_callback_null_called_;
|
| int num_callback_valid_called_;
|
| +
|
| + base::MessageLoop message_loop_;
|
| + content::TestBrowserThread db_thread_;
|
| +
|
| // Under test.
|
| scoped_ptr<ImageManager> image_manager_;
|
| };
|
| @@ -181,8 +190,9 @@ TEST_F(ImageManagerTest, GetImageForURLNetworkCacheHit) {
|
| fake_db->InitCallback(true);
|
| fake_db->LoadCallback(true);
|
| // Expect something in the cache.
|
| - SkBitmap* bitmap = image_manager_->GetBitmapFromCache(GURL(kTestUrl1));
|
| - EXPECT_FALSE(bitmap->isNull());
|
| + auto encoded_image =
|
| + image_manager_->GetEncodedImageFromCache(GURL(kTestUrl1));
|
| + EXPECT_NE(nullptr, encoded_image);
|
|
|
| base::RunLoop run_loop;
|
| image_manager_->GetImageForURL(GURL(kTestUrl1),
|
|
|