src/content/browser/browser_url_handler_impl_unittest.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/browser/browser_url_handler_impl.h"
   6 #include "content/public/test/test_browser_context.h"
   7 #include "testing/gtest/include/gtest/gtest.h"
   8 #include "url/gurl.h"
   9
  10 namespace content {
  11
  12 class BrowserURLHandlerImplTest : public testing::Test {
  13 };
  14
  15 // Test URL rewriter that rewrites all "foo://" URLs to "bar://bar".
  16 static bool FooRewriter(GURL* url, BrowserContext* browser_context) {
  17   if (url->scheme() == "foo") {
  18     *url = GURL("bar://bar");
  19     return true;
  20   }
  21   return false;
  22 }
  23
  24 // Test URL rewriter that rewrites all "bar://" URLs to "foo://foo".
  25 static bool BarRewriter(GURL* url, BrowserContext* browser_context) {
  26   if (url->scheme() == "bar") {
  27     *url = GURL("foo://foo");
  28     return true;
  29   }
  30   return false;
  31 }
  32
  33 TEST_F(BrowserURLHandlerImplTest, BasicRewriteAndReverse) {
  34   TestBrowserContext browser_context;
  35   BrowserURLHandlerImpl handler;
  36
  37   handler.AddHandlerPair(FooRewriter, BarRewriter);
  38
  39   GURL url("foo://bar");
  40   GURL original_url(url);
  41   bool reverse_on_redirect = false;
  42   handler.RewriteURLIfNecessary(&url, &browser_context, &reverse_on_redirect);
  43   ASSERT_TRUE(reverse_on_redirect);
  44   ASSERT_EQ("bar://bar", url.spec());
  45
  46   // Check that reversing the URL works.
  47   GURL saved_url(url);
  48   bool reversed = handler.ReverseURLRewrite(&url,
  49                                             original_url,
  50                                             &browser_context);
  51   ASSERT_TRUE(reversed);
  52   ASSERT_EQ("foo://foo", url.spec());
  53
  54   // Check that reversing the URL only works with a matching |original_url|.
  55   url = saved_url;
  56   original_url = GURL("bam://bam");  // Won't be matched by FooRewriter.
  57   reversed = handler.ReverseURLRewrite(&url, original_url, &browser_context);
  58   ASSERT_FALSE(reversed);
  59   ASSERT_EQ(saved_url, url);
  60 }
  61
  62 TEST_F(BrowserURLHandlerImplTest, NullHandlerReverse) {
  63   TestBrowserContext browser_context;
  64   BrowserURLHandlerImpl handler;
  65
  66   GURL url("bar://foo");
  67   GURL original_url(url);
  68
  69   handler.AddHandlerPair(BrowserURLHandlerImpl::null_handler(), FooRewriter);
  70   bool reversed = handler.ReverseURLRewrite(&url,
  71                                             original_url,
  72                                             &browser_context);
  73   ASSERT_FALSE(reversed);
  74   ASSERT_EQ(original_url, url);
  75
  76   handler.AddHandlerPair(BrowserURLHandlerImpl::null_handler(), BarRewriter);
  77   reversed = handler.ReverseURLRewrite(&url, original_url, &browser_context);
  78   ASSERT_TRUE(reversed);
  79   ASSERT_EQ("foo://foo", url.spec());
  80 }
  81
  82 }  // namespace content