Chromium Code Reviews| Index: content/browser/renderer_host/render_view_host_unittest.cc |
| diff --git a/content/browser/renderer_host/render_view_host_unittest.cc b/content/browser/renderer_host/render_view_host_unittest.cc |
| index 1feaa74c514e25c4f78dab03fcd325c1d51aabb7..da056f6eab42f7bad028762c9f43137d723c6d3e 100644 |
| --- a/content/browser/renderer_host/render_view_host_unittest.cc |
| +++ b/content/browser/renderer_host/render_view_host_unittest.cc |
| @@ -12,14 +12,49 @@ |
| #include "content/public/browser/navigation_entry.h" |
| #include "content/public/common/bindings_policy.h" |
| #include "content/public/common/page_transition_types.h" |
| +#include "content/public/common/url_constants.h" |
| #include "content/public/test/mock_render_process_host.h" |
| +#include "content/test/test_content_browser_client.h" |
| #include "net/base/net_util.h" |
| #include "third_party/WebKit/Source/WebKit/chromium/public/WebDragOperation.h" |
| #include "webkit/glue/webdropdata.h" |
| namespace content { |
| +class RenderViewHostTestBrowserClient : public TestContentBrowserClient { |
| + public: |
| + RenderViewHostTestBrowserClient() {} |
| + virtual ~RenderViewHostTestBrowserClient() {} |
| + |
| + virtual bool IsHandledURL(const GURL& url) { |
|
mmenke
2013/01/22 16:50:37
OVERRIDE
|
| + return url.scheme() == chrome::kFileScheme; |
| + } |
| + |
| + private: |
| + DISALLOW_COPY_AND_ASSIGN(RenderViewHostTestBrowserClient); |
| +}; |
| + |
| class RenderViewHostTest : public RenderViewHostImplTestHarness { |
| + public: |
| + RenderViewHostTest() : old_browser_client_(NULL) {} |
| + virtual ~RenderViewHostTest() {} |
| + |
| + virtual void SetUp() { |
|
mmenke
2013/01/22 16:50:37
OVERRIDE
|
| + RenderViewHostImplTestHarness::SetUp(); |
| + old_browser_client_ = GetContentClient()->browser(); |
| + GetContentClient()->set_browser_for_testing(&test_browser_client_); |
| + } |
| + |
| + virtual void TearDown() { |
|
mmenke
2013/01/22 16:50:37
OVERRIDE
|
| + RenderViewHostImplTestHarness::TearDown(); |
| + GetContentClient()->set_browser_for_testing(old_browser_client_); |
| + } |
| + |
| + private: |
| + RenderViewHostTestBrowserClient test_browser_client_; |
| + ContentBrowserClient* old_browser_client_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(RenderViewHostTest); |
| }; |
| // All about URLs reported by the renderer should get rewritten to about:blank. |