From 825b16687e0de3ef4ec3b7dd60fc1dfb237947c9 Mon Sep 17 00:00:00 2001
From: "jam@chromium.org"
 <jam@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>
Date: Mon, 12 Mar 2012 19:07:31 +0000
Subject: Add a Content API around BrowserURLHandler.

BUG=98716
Review URL: https://chromiumcodereview.appspot.com/9688019

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@126180 0039d316-1c4b-4281-b951-d872f2087c98
---
 .../browser/browser_url_handler_impl_unittest.cc   | 78 ++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 content/browser/browser_url_handler_impl_unittest.cc

(limited to 'content/browser/browser_url_handler_impl_unittest.cc')

diff --git a/content/browser/browser_url_handler_impl_unittest.cc b/content/browser/browser_url_handler_impl_unittest.cc
new file mode 100644
index 0000000..7109d5d
--- /dev/null
+++ b/content/browser/browser_url_handler_impl_unittest.cc
@@ -0,0 +1,78 @@
+// 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 "content/browser/browser_url_handler_impl.h"
+#include "content/test/test_browser_context.h"
+#include "googleurl/src/gurl.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+class BrowserURLHandlerImplTest : public testing::Test {
+};
+
+// Test URL rewriter that rewrites all "foo://" URLs to "bar://bar".
+static bool FooRewriter(GURL* url, content::BrowserContext* browser_context) {
+  if (url->scheme() == "foo") {
+    *url = GURL("bar://bar");
+    return true;
+  }
+  return false;
+}
+
+// Test URL rewriter that rewrites all "bar://" URLs to "foo://foo".
+static bool BarRewriter(GURL* url, content::BrowserContext* browser_context) {
+  if (url->scheme() == "bar") {
+    *url = GURL("foo://foo");
+    return true;
+  }
+  return false;
+}
+
+TEST_F(BrowserURLHandlerImplTest, BasicRewriteAndReverse) {
+  TestBrowserContext browser_context;
+  BrowserURLHandlerImpl handler;
+
+  handler.AddHandlerPair(FooRewriter, BarRewriter);
+
+  GURL url("foo://bar");
+  GURL original_url(url);
+  bool reverse_on_redirect = false;
+  handler.RewriteURLIfNecessary(&url, &browser_context, &reverse_on_redirect);
+  ASSERT_TRUE(reverse_on_redirect);
+  ASSERT_EQ("bar://bar", url.spec());
+
+  // Check that reversing the URL works.
+  GURL saved_url(url);
+  bool reversed = handler.ReverseURLRewrite(&url,
+                                            original_url,
+                                            &browser_context);
+  ASSERT_TRUE(reversed);
+  ASSERT_EQ("foo://foo", url.spec());
+
+  // Check that reversing the URL only works with a matching |original_url|.
+  url = saved_url;
+  original_url = GURL("bam://bam");  // Won't be matched by FooRewriter.
+  reversed = handler.ReverseURLRewrite(&url, original_url, &browser_context);
+  ASSERT_FALSE(reversed);
+  ASSERT_EQ(saved_url, url);
+}
+
+TEST_F(BrowserURLHandlerImplTest, NullHandlerReverse) {
+  TestBrowserContext browser_context;
+  BrowserURLHandlerImpl handler;
+
+  GURL url("bar://foo");
+  GURL original_url(url);
+
+  handler.AddHandlerPair(BrowserURLHandlerImpl::null_handler(), FooRewriter);
+  bool reversed = handler.ReverseURLRewrite(&url,
+                                            original_url,
+                                            &browser_context);
+  ASSERT_FALSE(reversed);
+  ASSERT_EQ(original_url, url);
+
+  handler.AddHandlerPair(BrowserURLHandlerImpl::null_handler(), BarRewriter);
+  reversed = handler.ReverseURLRewrite(&url, original_url, &browser_context);
+  ASSERT_TRUE(reversed);
+  ASSERT_EQ("foo://foo", url.spec());
+}
-- 
cgit v1.1