OLD | NEW |
(Empty) | |
| 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 |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include <string> |
| 6 |
| 7 #include "base/memory/scoped_ptr.h" |
| 8 #include "chrome/browser/search/suggestions/image_manager_impl.h" |
| 9 #include "chrome/test/base/testing_profile.h" |
| 10 #include "components/leveldb_proto/proto_database.h" |
| 11 #include "components/leveldb_proto/testing/fake_db.h" |
| 12 #include "components/suggestions/proto/suggestions.pb.h" |
| 13 #include "content/public/test/test_browser_thread_bundle.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 #include "url/gurl.h" |
| 16 |
| 17 namespace { |
| 18 |
| 19 using leveldb_proto::test::FakeDB; |
| 20 using suggestions::ImageData; |
| 21 using suggestions::ImageManagerImpl; |
| 22 |
| 23 typedef base::hash_map<std::string, ImageData> EntryMap; |
| 24 |
| 25 const char kTestUrl[] = "http://go.com/"; |
| 26 const char kTestImageUrl[] = "http://thumb.com/anchor_download_test.png"; |
| 27 |
| 28 class ImageManagerImplTest : public testing::Test { |
| 29 protected: |
| 30 ImageManagerImpl* CreateImageManager(Profile* profile) { |
| 31 FakeDB<ImageData>* fake_db = new FakeDB<ImageData>(&db_model_); |
| 32 return new ImageManagerImpl( |
| 33 profile->GetRequestContext(), |
| 34 scoped_ptr<leveldb_proto::ProtoDatabase<ImageData> >(fake_db), |
| 35 FakeDB<ImageData>::DirectoryForTestDB()); |
| 36 } |
| 37 |
| 38 content::TestBrowserThreadBundle thread_bundle_; |
| 39 EntryMap db_model_; |
| 40 }; |
| 41 |
| 42 } // namespace |
| 43 |
| 44 namespace suggestions { |
| 45 |
| 46 TEST_F(ImageManagerImplTest, InitializeTest) { |
| 47 SuggestionsProfile suggestions_profile; |
| 48 ChromeSuggestion* suggestion = suggestions_profile.add_suggestions(); |
| 49 suggestion->set_url(kTestUrl); |
| 50 suggestion->set_thumbnail(kTestImageUrl); |
| 51 |
| 52 TestingProfile profile; |
| 53 scoped_ptr<ImageManagerImpl> image_manager( |
| 54 CreateImageManager(&profile)); |
| 55 image_manager->Initialize(suggestions_profile); |
| 56 |
| 57 GURL output; |
| 58 EXPECT_TRUE(image_manager->GetImageURL(GURL(kTestUrl), &output)); |
| 59 EXPECT_EQ(GURL(kTestImageUrl), output); |
| 60 |
| 61 EXPECT_FALSE(image_manager->GetImageURL(GURL("http://b.com"), &output)); |
| 62 } |
| 63 |
| 64 } // namespace suggestions |
OLD | NEW |