Simplify how TestChromeWebUIControllerFactory is registered now that multiple WebUIControllerFactor…

chrome/test/base/web_ui_browsertest.cc

Index: chrome/test/base/web_ui_browsertest.cc
===================================================================
--- chrome/test/base/web_ui_browsertest.cc	(revision 179158)
+++ chrome/test/base/web_ui_browsertest.cc	(working copy)
@@ -2,7 +2,7 @@
 // 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/web_ui_browsertest.h"
+#include "chrome/test/base/web_ui_browsertest.h"
 
 #include <string>
 #include <vector>
@@ -20,10 +20,10 @@
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/browser_navigator.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
-#include "chrome/browser/ui/webui/test_chrome_web_ui_controller_factory.h"
 #include "chrome/browser/ui/webui/web_ui_test_handler.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/url_constants.h"
+#include "chrome/test/base/test_chrome_web_ui_controller_factory.h"
 #include "chrome/test/base/test_tab_strip_model_observer.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/navigation_controller.h"
@@ -320,20 +320,16 @@
 }  // namespace
 
 void WebUIBrowserTest::SetUpOnMainThread() {
-  InProcessBrowserTest::SetUpOnMainThread();
-
   logging::SetLogMessageHandler(&LogHandler);
-}
 
-void WebUIBrowserTest::CleanUpOnMainThread() {
-  InProcessBrowserTest::CleanUpOnMainThread();
+  content::WebUIControllerFactory::UnregisterFactoryForTesting(
+      ChromeWebUIControllerFactory::GetInstance());

[scr, 2013/01/29 01:01:54]

See comment below - would it be easier to use if this were more like a stack and
we didn't have to unregister the previous one, just register/unregister ours?

[scr, 2013/01/29 01:30:10]

Per IM, the goal is to get to multiple non-overlapping factories registered, so
the answer is no - it shouldn't be a stack.  That said, it would be nice if the
main factory were implemented with some means to populate a hash rather than
walking each of its known types and individual factories could register what
they knew.

If we had that, then you could see each host -> WebUIMaker be more like a stack,
where you could override temporarily.

FWIW, I added this only in this factory originally because I didn't want to add
any complexity to the non-test runtime, but seeing how the runtime does that big
if if if block in its GetWebUIFactoryFunction, I think the eventual overhaul to
support this natively (ditching this specialty class altogether) would save
cycles in the long run and be easier to understand.

 
-  logging::SetLogMessageHandler(NULL);
-}
+  test_factory_.reset(new TestChromeWebUIControllerFactory);
 
-void WebUIBrowserTest::SetUpInProcessBrowserTestFixture() {
-  InProcessBrowserTest::SetUpInProcessBrowserTestFixture();
-  TestChromeWebUIControllerFactory::AddFactoryOverride(
+  content::WebUIControllerFactory::RegisterFactory(test_factory_.get());
+
+  test_factory_->AddFactoryOverride(
       GURL(kDummyURL).host(), mock_provider_.Pointer());
 
   ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &test_data_directory_));
@@ -352,10 +348,14 @@
   AddLibrary(FilePath(kWebUILibraryJS));
 }
 
-void WebUIBrowserTest::TearDownInProcessBrowserTestFixture() {
-  InProcessBrowserTest::TearDownInProcessBrowserTestFixture();
-  TestChromeWebUIControllerFactory::RemoveFactoryOverride(
-      GURL(kDummyURL).host());
+void WebUIBrowserTest::CleanUpOnMainThread() {
+  logging::SetLogMessageHandler(NULL);
+
+  test_factory_->RemoveFactoryOverride(GURL(kDummyURL).host());
+  content::WebUIControllerFactory::UnregisterFactoryForTesting(
+      test_factory_.get());

[scr, 2013/01/29 01:01:54]

Should the old one be put back?  If so, would some sort of stack registry be
easier - you could just register & unregister yours without caring/knowing what
was there before…

[scr, 2013/01/29 01:30:10]

Neg - read above.

+
+  test_factory_.reset();
 }
 
 void WebUIBrowserTest::SetWebUIInstance(content::WebUI* web_ui) {