Updating histograms to allow for experiments & log origin-based histograms.

chrome/browser/prerender/prerender_util_unittest.cc

Index: chrome/browser/prerender/prerender_util_unittest.cc
===================================================================
--- chrome/browser/prerender/prerender_util_unittest.cc	(revision 0)
+++ chrome/browser/prerender/prerender_util_unittest.cc	(revision 0)
@@ -0,0 +1,52 @@
+// Copyright (c) 2011 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 "chrome/browser/prerender/prerender_util.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace prerender {
+
+class PrerenderUtilTest : public testing::Test {
+ public:
+  PrerenderUtilTest() {
+  }
+};
+
+// Ensure that extracting a urlencoded URL in the url= query string component
+// works.
+TEST_F(PrerenderUtilTest, ExtractURLInQueryStringTest) {
+  GURL result;
+  EXPECT_TRUE(MaybeGetQueryStringBasedAliasURL(
+      GURL("http://www.google.com/url?sa=t&source=web&cd=1&ved=0CBcQFjAA&url=http%3A%2F%2Fwww.abercrombie.com%2Fwebapp%2Fwcs%2Fstores%2Fservlet%2FStoreLocator%3FcatalogId%3D%26storeId%3D10051%26langId%3D-1&rct=j&q=allinurl%3A%26&ei=KLyUTYGSEdTWiAKUmLCdCQ&usg=AFQjCNF8nJ2MpBFfr1ijO39_f22bcKyccw&sig2=2ymyGpO0unJwU1d4kdCUjQ"),
+      &result));
+  ASSERT_EQ(GURL("http://www.abercrombie.com/webapp/wcs/stores/servlet/StoreLocator?catalogId=&storeId=10051&langId=-1").spec(), result.spec());
+  EXPECT_FALSE(MaybeGetQueryStringBasedAliasURL(
+      GURL("http://www.google.com/url?sadf=test&blah=blahblahblah"), &result));
+  EXPECT_FALSE(MaybeGetQueryStringBasedAliasURL(
+      GURL("http://www.google.com/?url=INVALIDurlsAREsoMUCHfun.com"), &result));
+  EXPECT_TRUE(MaybeGetQueryStringBasedAliasURL(
+      GURL("http://www.google.com/?url=http://validURLSareGREAT.com"),
+      &result));
+  ASSERT_EQ(GURL("http://validURLSareGREAT.com").spec(), result.spec());
+}
+
+// Ensure that extracting an experiment in the lpe= query string component
+// works.
+TEST_F(PrerenderUtilTest, ExtractExperimentInQueryStringTest) {
+  GURL result;
+  EXPECT_EQ(GetQueryStringBasedExperiment(
+      GURL("http://www.google.com/url?sa=t&source=web&cd=1&ved=0CBcQFjAA&url=http%3A%2F%2Fwww.abercrombie.com%2Fwebapp%2Fwcs%2Fstores%2Fservlet%2FStoreLocator%3FcatalogId%3D%26storeId%3D10051%26langId%3D-1&rct=j&q=allinurl%3A%26&ei=KLyUTYGSEdTWiAKUmLCdCQ&usg=AFQjCNF8nJ2MpBFfr1ijO39_f22bcKyccw&sig2=2ymyGpO0unJwU1d4kdCUjQ&lpe=4&asdf=test")), 4);
+  EXPECT_EQ(GetQueryStringBasedExperiment(
+      GURL("http://www.google.com/test.php?a=b")), kNoExperiment);
+  EXPECT_EQ(GetQueryStringBasedExperiment(
+      GURL("http://www.google.com/test.php?lpe=5")), 5);
+  EXPECT_EQ(GetQueryStringBasedExperiment(
+      GURL("http://www.google.com/test.php?lpe=50")), kNoExperiment);
+  EXPECT_EQ(GetQueryStringBasedExperiment(
+      GURL("http://www.google.com/test.php?lpe=0")), kNoExperiment);
+  EXPECT_EQ(GetQueryStringBasedExperiment(
+      GURL("http://www.google.com/test.php?lpe=10")), kNoExperiment);
+}
+
+}  // namespace prerender