| Index: content/browser/web_contents/render_view_host_manager_unittest.cc
|
| ===================================================================
|
| --- content/browser/web_contents/render_view_host_manager_unittest.cc (revision 175970)
|
| +++ content/browser/web_contents/render_view_host_manager_unittest.cc (working copy)
|
| @@ -10,12 +10,12 @@
|
| #include "content/browser/web_contents/navigation_entry_impl.h"
|
| #include "content/browser/web_contents/render_view_host_manager.h"
|
| #include "content/browser/web_contents/test_web_contents.h"
|
| +#include "content/browser/webui/web_ui_controller_factory_registry.h"
|
| #include "content/common/view_messages.h"
|
| #include "content/public/browser/notification_details.h"
|
| #include "content/public/browser/notification_source.h"
|
| #include "content/public/browser/notification_types.h"
|
| #include "content/public/browser/web_ui_controller.h"
|
| -#include "content/public/browser/web_ui_controller_factory.h"
|
| #include "content/public/common/bindings_policy.h"
|
| #include "content/public/common/javascript_message_type.h"
|
| #include "content/public/common/page_transition_types.h"
|
| @@ -25,7 +25,6 @@
|
| #include "content/public/test/test_notification_tracker.h"
|
| #include "content/test/test_content_browser_client.h"
|
| #include "content/test/test_content_client.h"
|
| -#include "googleurl/src/url_util.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| #include "webkit/glue/glue_serialize.h"
|
|
|
| @@ -47,7 +46,7 @@
|
| // WebUIFactory implementation.
|
| virtual WebUIController* CreateWebUIControllerForURL(
|
| WebUI* web_ui, const GURL& url) const OVERRIDE {
|
| - if (!(should_create_webui_ && GetContentClient()->HasWebUIScheme(url)))
|
| + if (!(should_create_webui_ && HasWebUIScheme(url)))
|
| return NULL;
|
| return new WebUIController(web_ui);
|
| }
|
| @@ -59,19 +58,19 @@
|
|
|
| virtual bool UseWebUIForURL(BrowserContext* browser_context,
|
| const GURL& url) const OVERRIDE {
|
| - return GetContentClient()->HasWebUIScheme(url);
|
| + return HasWebUIScheme(url);
|
| }
|
|
|
| virtual bool UseWebUIBindingsForURL(BrowserContext* browser_context,
|
| const GURL& url) const OVERRIDE {
|
| - return GetContentClient()->HasWebUIScheme(url);
|
| + return HasWebUIScheme(url);
|
| }
|
|
|
| virtual bool IsURLAcceptableForWebUI(
|
| BrowserContext* browser_context,
|
| const GURL& url,
|
| bool data_urls_allowed) const OVERRIDE {
|
| - return GetContentClient()->HasWebUIScheme(url);
|
| + return HasWebUIScheme(url);
|
| }
|
|
|
| private:
|
| @@ -80,38 +79,6 @@
|
| DISALLOW_COPY_AND_ASSIGN(RenderViewHostManagerTestWebUIControllerFactory);
|
| };
|
|
|
| -class RenderViewHostManagerTestClient : public TestContentClient {
|
| - public:
|
| - RenderViewHostManagerTestClient() {
|
| - }
|
| -
|
| - virtual bool HasWebUIScheme(const GURL& url) const OVERRIDE {
|
| - return url.SchemeIs(chrome::kChromeUIScheme);
|
| - }
|
| -};
|
| -
|
| -class RenderViewHostManagerTestBrowserClient
|
| - : public TestContentBrowserClient {
|
| - public:
|
| - RenderViewHostManagerTestBrowserClient() {}
|
| - virtual ~RenderViewHostManagerTestBrowserClient() {}
|
| -
|
| - void set_should_create_webui(bool should_create_webui) {
|
| - factory_.set_should_create_webui(should_create_webui);
|
| - }
|
| -
|
| - // TestContentBrowserClient implementation.
|
| - virtual WebUIControllerFactory*
|
| - GetWebUIControllerFactory() OVERRIDE {
|
| - return &factory_;
|
| - }
|
| -
|
| - private:
|
| - RenderViewHostManagerTestWebUIControllerFactory factory_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(RenderViewHostManagerTestBrowserClient);
|
| -};
|
| -
|
| } // namespace
|
|
|
| class RenderViewHostManagerTest
|
| @@ -119,21 +86,16 @@
|
| public:
|
| virtual void SetUp() OVERRIDE {
|
| RenderViewHostImplTestHarness::SetUp();
|
| - old_client_ = GetContentClient();
|
| - old_browser_client_ = GetContentClient()->browser();
|
| - SetContentClient(&client_);
|
| - GetContentClient()->set_browser_for_testing(&browser_client_);
|
| - url_util::AddStandardScheme(chrome::kChromeUIScheme);
|
| + WebUIControllerFactory::RegisterFactory(&factory_);
|
| }
|
|
|
| virtual void TearDown() OVERRIDE {
|
| RenderViewHostImplTestHarness::TearDown();
|
| - GetContentClient()->set_browser_for_testing(old_browser_client_);
|
| - SetContentClient(old_client_);
|
| + WebUIControllerFactoryRegistry::UnregisterFactoryForTesting(&factory_);
|
| }
|
|
|
| void set_should_create_webui(bool should_create_webui) {
|
| - browser_client_.set_should_create_webui(should_create_webui);
|
| + factory_.set_should_create_webui(should_create_webui);
|
| }
|
|
|
| void NavigateActiveAndCommit(const GURL& url) {
|
| @@ -166,10 +128,7 @@
|
| }
|
|
|
| private:
|
| - RenderViewHostManagerTestClient client_;
|
| - RenderViewHostManagerTestBrowserClient browser_client_;
|
| - ContentClient* old_client_;
|
| - ContentBrowserClient* old_browser_client_;
|
| + RenderViewHostManagerTestWebUIControllerFactory factory_;
|
| };
|
|
|
| // Tests that when you navigate from a chrome:// url to another page, and
|
|
|