Add a getter for search provider logos.

chrome/browser/search_engines/template_url_prepopulate_data_unittest.cc

Index: chrome/browser/search_engines/template_url_prepopulate_data_unittest.cc
diff --git a/chrome/browser/search_engines/template_url_prepopulate_data_unittest.cc b/chrome/browser/search_engines/template_url_prepopulate_data_unittest.cc
index f0e33c0c15b153c6d053000d4000beccd2aa5798..963e251fbe3954e90531faa32c581f00b02fc62a 100644
--- a/chrome/browser/search_engines/template_url_prepopulate_data_unittest.cc
+++ b/chrome/browser/search_engines/template_url_prepopulate_data_unittest.cc
@@ -176,3 +176,45 @@ TEST_F(TemplateURLPrepopulateDataTest, GetEngineTypeAdvanced) {
   EXPECT_EQ(SEARCH_ENGINE_OTHER,
             TemplateURLPrepopulateData::GetEngineType("invalid:search:url"));
 }
+
+TEST(TemplateURLPrepopulateDataTest, GetLogoURLGoogle100Percent) {
+  TemplateURLData data;
+  data.SetURL("http://www.google.com/");
+  TemplateURL turl(NULL, data);
+  GURL logo_url = TemplateURLPrepopulateData::GetLogoURL(
+      turl, TemplateURLPrepopulateData::LOGO_100_PERCENT);
+
+  EXPECT_EQ("www.google.com", logo_url.host());
+  EXPECT_TRUE(logo_url.SchemeIsSecure());

[Peter Kasting, 2012/08/28 23:20:20]

Why is testing for a secure scheme something the unittest should ensure?  Seems
like this should be omitted.

[samarth, 2012/08/28 23:29:20]

Done.

+}
+
+TEST(TemplateURLPrepopulateDataTest, GetLogoURLGoogle200Percent) {

[Peter Kasting, 2012/08/28 23:20:20]

Nit: Combine these first two tests, a la:

  ...
  GURL logo_100 = ...;
  EXPECT_EQ("www.google.com", logo_100.host());

  GURL logo_200 = ...;
  EXPECT_EQ("www.google.com", logo_200.host());

  EXPECT_NE(logo_100, logo_200);

[samarth, 2012/08/28 23:29:20]

Done.

+  TemplateURLData data;
+  data.SetURL("http://www.google.com/");
+  TemplateURL turl(NULL, data);
+  GURL logo_url = TemplateURLPrepopulateData::GetLogoURL(
+      turl, TemplateURLPrepopulateData::LOGO_200_PERCENT);
+
+  EXPECT_EQ("www.google.com", logo_url.host());
+  EXPECT_TRUE(logo_url.SchemeIsSecure());
+}
+
+TEST(TemplateURLPrepopulateDataTest, GetLogoURLUnknown) {
+  TemplateURLData data;
+  data.SetURL("http://webalta.ru/");
+  TemplateURL turl(NULL, data);
+  GURL logo_url = TemplateURLPrepopulateData::GetLogoURL(
+      turl, TemplateURLPrepopulateData::LOGO_100_PERCENT);
+
+  EXPECT_TRUE(logo_url.is_empty());
+}
+
+TEST(TemplateURLPrepopulateDataTest, GetLogoURLInvalid) {
+  TemplateURLData data;
+  data.SetURL("http://invalid:search:url/");
+  TemplateURL turl(NULL, data);
+  GURL logo_url = TemplateURLPrepopulateData::GetLogoURL(
+      turl, TemplateURLPrepopulateData::LOGO_100_PERCENT);
+
+  EXPECT_TRUE(logo_url.is_empty());
+}