Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(288)

Side by Side Diff: content/browser/browser_url_handler_impl_unittest.cc

Issue 9688019: Add a Content API around BrowserURLHandler. (Closed) Base URL: svn://chrome-svn/chrome/trunk/src/
Patch Set: fix gypis Created 8 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 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 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "content/browser/browser_url_handler.h" 5 #include "content/browser/browser_url_handler_impl.h"
6 #include "content/test/test_browser_context.h" 6 #include "content/test/test_browser_context.h"
7 #include "googleurl/src/gurl.h" 7 #include "googleurl/src/gurl.h"
8 #include "testing/gtest/include/gtest/gtest.h" 8 #include "testing/gtest/include/gtest/gtest.h"
9 9
10 class BrowserURLHandlerTest : public testing::Test { 10 class BrowserURLHandlerImplTest : public testing::Test {
11 }; 11 };
12 12
13 // Test URL rewriter that rewrites all "foo://" URLs to "bar://bar". 13 // Test URL rewriter that rewrites all "foo://" URLs to "bar://bar".
14 static bool FooRewriter(GURL* url, content::BrowserContext* browser_context) { 14 static bool FooRewriter(GURL* url, content::BrowserContext* browser_context) {
15 if (url->scheme() == "foo") { 15 if (url->scheme() == "foo") {
16 *url = GURL("bar://bar"); 16 *url = GURL("bar://bar");
17 return true; 17 return true;
18 } 18 }
19 return false; 19 return false;
20 } 20 }
21 21
22 // Test URL rewriter that rewrites all "bar://" URLs to "foo://foo". 22 // Test URL rewriter that rewrites all "bar://" URLs to "foo://foo".
23 static bool BarRewriter(GURL* url, content::BrowserContext* browser_context) { 23 static bool BarRewriter(GURL* url, content::BrowserContext* browser_context) {
24 if (url->scheme() == "bar") { 24 if (url->scheme() == "bar") {
25 *url = GURL("foo://foo"); 25 *url = GURL("foo://foo");
26 return true; 26 return true;
27 } 27 }
28 return false; 28 return false;
29 } 29 }
30 30
31 TEST_F(BrowserURLHandlerTest, BasicRewriteAndReverse) { 31 TEST_F(BrowserURLHandlerImplTest, BasicRewriteAndReverse) {
32 TestBrowserContext browser_context; 32 TestBrowserContext browser_context;
33 BrowserURLHandler handler; 33 BrowserURLHandlerImpl handler;
34 34
35 handler.AddHandlerPair(FooRewriter, BarRewriter); 35 handler.AddHandlerPair(FooRewriter, BarRewriter);
36 36
37 GURL url("foo://bar"); 37 GURL url("foo://bar");
38 GURL original_url(url); 38 GURL original_url(url);
39 bool reverse_on_redirect = false; 39 bool reverse_on_redirect = false;
40 handler.RewriteURLIfNecessary(&url, &browser_context, &reverse_on_redirect); 40 handler.RewriteURLIfNecessary(&url, &browser_context, &reverse_on_redirect);
41 ASSERT_TRUE(reverse_on_redirect); 41 ASSERT_TRUE(reverse_on_redirect);
42 ASSERT_EQ("bar://bar", url.spec()); 42 ASSERT_EQ("bar://bar", url.spec());
43 43
44 // Check that reversing the URL works. 44 // Check that reversing the URL works.
45 GURL saved_url(url); 45 GURL saved_url(url);
46 bool reversed = handler.ReverseURLRewrite(&url, 46 bool reversed = handler.ReverseURLRewrite(&url,
47 original_url, 47 original_url,
48 &browser_context); 48 &browser_context);
49 ASSERT_TRUE(reversed); 49 ASSERT_TRUE(reversed);
50 ASSERT_EQ("foo://foo", url.spec()); 50 ASSERT_EQ("foo://foo", url.spec());
51 51
52 // Check that reversing the URL only works with a matching |original_url|. 52 // Check that reversing the URL only works with a matching |original_url|.
53 url = saved_url; 53 url = saved_url;
54 original_url = GURL("bam://bam"); // Won't be matched by FooRewriter. 54 original_url = GURL("bam://bam"); // Won't be matched by FooRewriter.
55 reversed = handler.ReverseURLRewrite(&url, original_url, &browser_context); 55 reversed = handler.ReverseURLRewrite(&url, original_url, &browser_context);
56 ASSERT_FALSE(reversed); 56 ASSERT_FALSE(reversed);
57 ASSERT_EQ(saved_url, url); 57 ASSERT_EQ(saved_url, url);
58 } 58 }
59 59
60 TEST_F(BrowserURLHandlerTest, NullHandlerReverse) { 60 TEST_F(BrowserURLHandlerImplTest, NullHandlerReverse) {
61 TestBrowserContext browser_context; 61 TestBrowserContext browser_context;
62 BrowserURLHandler handler; 62 BrowserURLHandlerImpl handler;
63 63
64 GURL url("bar://foo"); 64 GURL url("bar://foo");
65 GURL original_url(url); 65 GURL original_url(url);
66 66
67 handler.AddHandlerPair(BrowserURLHandler::null_handler(), FooRewriter); 67 handler.AddHandlerPair(BrowserURLHandlerImpl::null_handler(), FooRewriter);
68 bool reversed = handler.ReverseURLRewrite(&url, 68 bool reversed = handler.ReverseURLRewrite(&url,
69 original_url, 69 original_url,
70 &browser_context); 70 &browser_context);
71 ASSERT_FALSE(reversed); 71 ASSERT_FALSE(reversed);
72 ASSERT_EQ(original_url, url); 72 ASSERT_EQ(original_url, url);
73 73
74 handler.AddHandlerPair(BrowserURLHandler::null_handler(), BarRewriter); 74 handler.AddHandlerPair(BrowserURLHandlerImpl::null_handler(), BarRewriter);
75 reversed = handler.ReverseURLRewrite(&url, original_url, &browser_context); 75 reversed = handler.ReverseURLRewrite(&url, original_url, &browser_context);
76 ASSERT_TRUE(reversed); 76 ASSERT_TRUE(reversed);
77 ASSERT_EQ("foo://foo", url.spec()); 77 ASSERT_EQ("foo://foo", url.spec());
78 } 78 }
OLDNEW
« no previous file with comments | « content/browser/browser_url_handler_impl.cc ('k') | content/browser/browser_url_handler_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698