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 6e94708a84ff61ecaf120f89a690cf38924cde51..2e2a38bd3b7f39d63e283b2d0bd4d39fcf249901 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) 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() OVERRIDE { |
+ RenderViewHostImplTestHarness::SetUp(); |
+ old_browser_client_ = GetContentClient()->browser(); |
+ GetContentClient()->set_browser_for_testing(&test_browser_client_); |
+ } |
+ |
+ virtual void TearDown() OVERRIDE { |
+ RenderViewHostImplTestHarness::TearDown(); |
+ GetContentClient()->set_browser_for_testing(old_browser_client_); |
mmenke
2013/02/19 17:26:15
This should be first, so cleanup order is the reve
pauljensen
2013/02/20 15:13:27
Done.
|
+ } |
+ |
+ 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. |