Allow overriding WebUIFactory when test flag is set.

chrome/browser/ui/webui/test_chrome_web_ui_factory.cc

Index: chrome/browser/ui/webui/test_chrome_web_ui_factory.cc
diff --git a/chrome/browser/ui/webui/test_chrome_web_ui_factory.cc b/chrome/browser/ui/webui/test_chrome_web_ui_factory.cc
new file mode 100644
index 0000000000000000000000000000000000000000..199e965e71a1dd8fd418951803171e23f6595a3d
--- /dev/null
+++ b/chrome/browser/ui/webui/test_chrome_web_ui_factory.cc
@@ -0,0 +1,55 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/ui/webui/test_chrome_web_ui_factory.h"
+
+#include "chrome/browser/ui/webui/chrome_web_ui_factory.h"

[Lei Zhang, 2011/05/25 21:21:18]

nit: You don't need to include this and web_ui.h again. You already include them
in test_chrome_web_ui_factory.h.

+#include "content/browser/tab_contents/tab_contents.h"
+#include "content/browser/webui/web_ui.h"
+
+TestChromeWebUIFactory::WebUIProvider::~WebUIProvider() {
+}
+
+TestChromeWebUIFactory::TestChromeWebUIFactory() {
+}
+
+TestChromeWebUIFactory::~TestChromeWebUIFactory() {
+}
+
+void TestChromeWebUIFactory::AddFactoryOverride(const std::string& host,
+                                                WebUIProvider* provider) {
+  DCHECK_EQ(0U, GetInstance()->factory_overrides_.count(host));
+  GetInstance()->factory_overrides_[host] = provider;
+}
+
+void TestChromeWebUIFactory::RemoveFactoryOverride(const std::string& host) {
+  DCHECK_EQ(1U, GetInstance()->factory_overrides_.count(host));
+  GetInstance()->factory_overrides_.erase(host);
+}
+
+WebUI::TypeID TestChromeWebUIFactory::GetWebUIType(
+    Profile* profile, const GURL& url) const {
+  WebUIProvider* provider = GetWebUIProvider(profile, url);
+  return provider ? reinterpret_cast<WebUI::TypeID>(provider) :
+      ChromeWebUIFactory::GetWebUIType(profile, url);
+}
+
+WebUI* TestChromeWebUIFactory::CreateWebUIForURL(TabContents* tab_contents,
+                                                 const GURL& url) const {
+  WebUIProvider* provider = GetWebUIProvider(tab_contents->profile(), url);
+  return provider ? provider->NewWebUI(tab_contents, url) :
+      ChromeWebUIFactory::CreateWebUIForURL(tab_contents, url);
+}
+
+TestChromeWebUIFactory* TestChromeWebUIFactory::GetInstance() {
+  return static_cast<TestChromeWebUIFactory*>(
+      ChromeWebUIFactory::GetInstance());
+}
+
+TestChromeWebUIFactory::WebUIProvider* TestChromeWebUIFactory::GetWebUIProvider(
+    Profile* profile, const GURL& url) const {
+  FactoryOverridesMap::const_iterator found =
+      factory_overrides_.find(url.host());
+  return (found == factory_overrides_.end()) ? NULL : found->second;
+}