[Icons NTP] Allow chrome-search:// large-icon and fallback-icon hosts to use <view_id>/<restricted_…

chrome/renderer/searchbox/searchbox_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/basictypes.h"
 #include "chrome/common/instant_types.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 namespace internal {
 
 // Defined in searchbox.cc
 bool GetRestrictedIDFromThumbnailUrl(int render_view_id,
                                      const GURL& url,
                                      InstantRestrictedID* id);
 
 // Defined in searchbox.cc
 bool GetRestrictedIDFromFaviconUrl(int render_view_id,
                                    const GURL& url,
                                    std::string* favicon_params,
                                    InstantRestrictedID* rid);
 
+// Defined in searchbox.cc
+bool GetRestrictedIDFromLargeIconUrl(int render_view_id,
+                                   const GURL& url,
+                                   std::string* favicon_params,
+                                   InstantRestrictedID* rid);
+
+// Defined in searchbox.cc
+bool GetRestrictedIDFromFallbackIconUrl(int render_view_id,
+                                        const GURL& url,
+                                        std::string* favicon_params,
+                                        InstantRestrictedID* rid);
+
 TEST(SearchBoxUtilTest, GetInstantRestrictedIDFromTransientURL) {
   const int kInvalidRenderViewID = 920;
   const int kValidRenderViewID = 1;
 
   const struct {
     int render_view_id;
     GURL transient_url;
     InstantRestrictedID expected_rid;
     bool expected_return_val;
   } test_cases[] = {
@@ skipping 190 matching lines @@
         &favicon_params,
         &rid);
     EXPECT_EQ(test_cases[i].expected_return_val, return_val);
     EXPECT_EQ(test_cases[i].expected_favicon_params, favicon_params);
     EXPECT_EQ(test_cases[i].expected_rid, rid);
     favicon_params = "";
     rid = 0;
   }
 }
 
+TEST(SearchBoxUtilTest, GetRestrictedIDFromLargeIconUrl) {
+  const int kInvalidRenderViewID = 920;
+  const int kValidRenderViewID = 1;
+
+  const struct {
+    int render_view_id;
+    GURL transient_url;
+    std::string expected_icon_params;
+    InstantRestrictedID expected_rid;
+    bool expected_return_val;
+  } test_cases[] = {
+    // RenderView ID matches the view id specified in the transient url.
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://big-icon/96/1/2"),
+      "96/",
+      2,
+      true
+    },
+
+    // RenderView ID does not match the view id specified in the transient url.
+    {
+      kInvalidRenderViewID,
+      GURL("chrome-search://big-icon/96/1/2"),
+      "96/",
+      0,
+      true
+    },
+
+    // Invalid transient urls.
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://big-icon"),
+      "",
+      0,
+      false
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://big-icon/"),
+      "",
+      0,
+      false
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://big-icon/96"),
+      "",
+      0,
+      false
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://big-icon/96/123"),
+      "96/",
+      0,
+      true
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://big-icon/96/xyz"),
+      "96/",
+      0,
+      true
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://big-icon/invalidparameter/96/1/2"),
+      "",
+      0,
+      false  // More stringent than chrome-search://favicon.
+    }
+  };
+
+  std::string icon_params = "";
+  InstantRestrictedID rid = 0;
+  for (size_t i = 0; i < arraysize(test_cases); ++i) {
+    bool return_val = GetRestrictedIDFromLargeIconUrl(
+        test_cases[i].render_view_id,
+        test_cases[i].transient_url,
+        &icon_params,
+        &rid);
+    EXPECT_EQ(test_cases[i].expected_return_val, return_val);
+    EXPECT_EQ(test_cases[i].expected_icon_params, icon_params);
+    EXPECT_EQ(test_cases[i].expected_rid, rid);
+    icon_params = "";
+    rid = 0;
+  }
+}
+
+TEST(SearchBoxUtilTest, GetRestrictedIDFromFallbackIconUrl) {
+  const int kInvalidRenderViewID = 920;
+  const int kValidRenderViewID = 1;
+
+  const struct {
+    int render_view_id;
+    GURL transient_url;
+    std::string expected_icon_params;
+    InstantRestrictedID expected_rid;
+    bool expected_return_val;
+  } test_cases[] = {
+    // RenderView ID matches the view id specified in the transient url.
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://fallback-icon/,,,,/1/2"),
+      ",,,,/",
+      2,
+      true
+    },
+
+    // RenderView ID does not match the view id specified in the transient url.
+    {
+      kInvalidRenderViewID,
+      GURL("chrome-search://fallback-icon/,,,,/1/2"),
+      ",,,,/",
+      0,
+      true
+    },
+
+    // Invalid transient urls.
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://fallback-icon"),
+      "",
+      0,
+      false
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://fallback-icon/"),
+      "",
+      0,
+      false
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://fallback-icon/,,,,"),
+      "",
+      0,
+      false
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://fallback-icon/,,,,/123"),
+      ",,,,/",
+      0,
+      true
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://fallback-icon/,,,,/xyz"),
+      ",,,,/",
+      0,
+      true
+    },
+    {
+      kValidRenderViewID,
+      GURL("chrome-search://fallback-icon/invalidparameter/,,,,/1/2"),
+      "",
+      0,
+      false  // More stringent than chrome-search://favicon.
+    }
+  };
+
+  std::string icon_params = "";
+  InstantRestrictedID rid = 0;
+  for (size_t i = 0; i < arraysize(test_cases); ++i) {
+    bool return_val = GetRestrictedIDFromFallbackIconUrl(
+        test_cases[i].render_view_id,
+        test_cases[i].transient_url,
+        &icon_params,
+        &rid);
+    EXPECT_EQ(test_cases[i].expected_return_val, return_val);
+    EXPECT_EQ(test_cases[i].expected_icon_params, icon_params);
+    EXPECT_EQ(test_cases[i].expected_rid, rid);
+    icon_params = "";
+    rid = 0;
+  }
+}
+
 }  // namespace internal