Index: content/browser/frame_host/render_frame_host_manager_unittest.cc |
diff --git a/content/browser/frame_host/render_frame_host_manager_unittest.cc b/content/browser/frame_host/render_frame_host_manager_unittest.cc |
index 768b59fd4fb469e06597dae1a3bf97d4a722f2ae..bac4f6df8215295e4fe88f03dc88fe333798b34d 100644 |
--- a/content/browser/frame_host/render_frame_host_manager_unittest.cc |
+++ b/content/browser/frame_host/render_frame_host_manager_unittest.cc |
@@ -65,13 +65,21 @@ class RenderFrameHostManagerTestWebUIControllerFactory |
// WebUIFactory implementation. |
WebUIController* CreateWebUIControllerForURL(WebUI* web_ui, |
const GURL& url) const override { |
- if (!(should_create_webui_ && HasWebUIScheme(url))) |
- return NULL; |
- return new WebUIController(web_ui); |
+ // If WebUI creation is enabled for the test and this is a WebUI URL, |
+ // returns a new instance. |
+ if (should_create_webui_ && HasWebUIScheme(url)) |
+ return new WebUIController(web_ui); |
+ return nullptr; |
} |
WebUI::TypeID GetWebUIType(BrowserContext* browser_context, |
const GURL& url) const override { |
+ // If WebUI creation is enabled for the test and this is a WebUI URL, |
+ // returns a mock WebUI type. |
+ if (should_create_webui_ && HasWebUIScheme(url)) { |
+ return const_cast<RenderFrameHostManagerTestWebUIControllerFactory*>( |
+ this); |
+ } |
return WebUI::kNoWebUI; |
} |