| 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 |