1 files changed, 35 insertions, 0 deletions
diff --git a/chrome/browser/google_url_tracker_unittest.cc b/chrome/browser/google_url_tracker_unittest.cc
new file mode 100644
index 0000000..9de0b88
--- /dev/null
+++ b/chrome/browser/google_url_tracker_unittest.cc
@@ -0,0 +1,35 @@
+// Copyright (c) 2010 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/google_url_tracker.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+TEST(GoogleURLTrackerTest, CheckAndConvertURL) {
+  static const struct {
+    const char* const source_url;
+    const bool can_convert;
+    const char* const base_url;
+  } data[] = {
+    { "http://www.google.com/", true, "http://www.google.com/", },
+    { "http://google.fr/", true, "http://google.fr/", },
+    { "http://google.co.uk/", true, "http://google.co.uk/", },
+    { "http://www.google.com.by/", true, "http://www.google.com.by/", },
+    { "http://www.google.com/ig", true, "http://www.google.com/", },
+    { "http://www.google.com/intl/xx", true, "http://www.google.com/intl/xx", },
+    { "http://a.b.c.google.com/", true, "http://a.b.c.google.com/", },
+    { "http://www.yahoo.com/", false, "", },
+    { "http://google.evil.com/", false, "", },
+    { "http://google.com.com.com/", false, "", },
+    { "http://google/", false, "", },
+    { "http://wifi.corp.google.com/", false, "" },
+  };
+
+  for (size_t i = 0; i < ARRAYSIZE_UNSAFE(data); ++i) {
+    GURL base_url;
+    const bool can_convert = GoogleURLTracker::CheckAndConvertToGoogleBaseURL(
+        GURL(data[i].source_url), &base_url);
+    EXPECT_EQ(data[i].can_convert, can_convert);
+    EXPECT_STREQ(data[i].base_url, base_url.spec().c_str());
+  }
+}