Move WebUIFactory to chrome/, try 2.

content/browser/site_instance_unittest.cc

 // 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 "base/stl_util-inl.h"
 #include "base/string16.h"
 #include "chrome/browser/renderer_host/browser_render_process_host.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/render_messages.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/testing_profile.h"
 #include "content/browser/browsing_instance.h"
-#include "testing/gtest/include/gtest/gtest.h"
-#include "content/browser/browsing_instance.h"
 #include "content/browser/child_process_security_policy.h"
+#include "content/browser/content_browser_client.h"
 #include "content/browser/renderer_host/render_view_host.h"
+#include "content/browser/renderer_host/test_render_view_host.h"
 #include "content/browser/site_instance.h"
-#include "content/browser/renderer_host/test_render_view_host.h"
 #include "content/browser/tab_contents/navigation_entry.h"
 #include "content/browser/tab_contents/tab_contents.h"
-
-class SiteInstanceTest : public testing::Test {
- private:
-  MessageLoopForUI message_loop_;
-};
+#include "content/browser/webui/empty_web_ui_factory.h"
+#include "content/common/content_client.h"
+#include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
+// TODO(estade): this shouldn't need to be chrome:, but it does (or else GURL
+// doesn't think that the webui URLs have a host). Figure out where this is
+// coming from and fix it.
+const char kWebUIScheme[] = "chrome";
+
+class SiteInstanceTestWebUIFactory : public content::EmptyWebUIFactory {
+ public:
+  virtual bool UseWebUIForURL(Profile* profile, const GURL& url) const {
+    return HasWebUIScheme(url);
+  }
+  virtual bool HasWebUIScheme(const GURL& url) const {
+    return url.SchemeIs(kWebUIScheme);
+  }
+};
+
+class SiteInstanceTestBrowserClient : public content::ContentBrowserClient {
+ public:
+  virtual content::WebUIFactory* GetWebUIFactory() {
+    return &factory_;
+  }
+
+ private:
+  SiteInstanceTestWebUIFactory factory_;
+};
+
+class SiteInstanceTest : public testing::Test {
+ public:
+  virtual void SetUp() {
+    content::GetContentClient()->set_browser(&browser_client_);
+  }
+
+ private:
+  MessageLoopForUI message_loop_;
+
+  SiteInstanceTestBrowserClient browser_client_;
+};
+
 class TestBrowsingInstance : public BrowsingInstance {
  public:
   TestBrowsingInstance(Profile* profile, int* deleteCounter)
       : BrowsingInstance(profile),
         use_process_per_site(false),
         deleteCounter_(deleteCounter) {
   }
 
   // Overrides BrowsingInstance::ShouldUseProcessPerSite so that we can test
   // both alternatives without using command-line switches.
   bool ShouldUseProcessPerSite(const GURL& url) {
     return use_process_per_site;
   }
 
   // Set by individual tests.
   bool use_process_per_site;
 
  private:
   ~TestBrowsingInstance() {
     (*deleteCounter_)++;
   }
 
   int* deleteCounter_;
 };
 
-
 class TestSiteInstance : public SiteInstance {
  public:
   static TestSiteInstance* CreateTestSiteInstance(Profile* profile,
                                                   int* siteDeleteCounter,
                                                   int* browsingDeleteCounter) {
     TestBrowsingInstance* browsing_instance =
-      new TestBrowsingInstance(profile, browsingDeleteCounter);
+        new TestBrowsingInstance(profile, browsingDeleteCounter);
     return new TestSiteInstance(browsing_instance, siteDeleteCounter);
   }
 
  private:
   TestSiteInstance(BrowsingInstance* browsing_instance, int* deleteCounter)
     : SiteInstance(browsing_instance), deleteCounter_(deleteCounter) {}
   ~TestSiteInstance() {
     (*deleteCounter_)++;
   }
 
@@ skipping 362 matching lines @@
 
   scoped_refptr<SiteInstance> extension2_instance(
       CreateSiteInstance(&rph_factory, GURL("chrome-extension://baz/bar")));
 
   scoped_ptr<RenderProcessHost> extension_host(
       extension1_instance->GetProcess());
   EXPECT_EQ(extension1_instance->GetProcess(),
             extension2_instance->GetProcess());
 
   // Create some WebUI instances and make sure they share a process.
-  scoped_refptr<SiteInstance> dom1_instance(
-      CreateSiteInstance(&rph_factory, GURL("chrome://newtab")));
-  policy->GrantWebUIBindings(dom1_instance->GetProcess()->id());
+  scoped_refptr<SiteInstance> webui1_instance(
+      CreateSiteInstance(&rph_factory,
+                         GURL(kWebUIScheme + std::string("://newtab"))));
+  policy->GrantWebUIBindings(webui1_instance->GetProcess()->id());
 
-  scoped_refptr<SiteInstance> dom2_instance(
-      CreateSiteInstance(&rph_factory, GURL("chrome://history")));
+  scoped_refptr<SiteInstance> webui2_instance(
+      CreateSiteInstance(&rph_factory,
+                         GURL(kWebUIScheme + std::string("://history"))));
 
-  scoped_ptr<RenderProcessHost> dom_host(dom1_instance->GetProcess());
-  EXPECT_EQ(dom1_instance->GetProcess(), dom2_instance->GetProcess());
+  scoped_ptr<RenderProcessHost> dom_host(webui1_instance->GetProcess());
+  EXPECT_EQ(webui1_instance->GetProcess(), webui2_instance->GetProcess());
 
   // Make sure none of differing privilege processes are mixed.
-  EXPECT_NE(extension1_instance->GetProcess(), dom1_instance->GetProcess());
+  EXPECT_NE(extension1_instance->GetProcess(), webui1_instance->GetProcess());
 
   for (size_t i = 0; i < chrome::kMaxRendererProcessCount; ++i) {
     EXPECT_NE(extension1_instance->GetProcess(), hosts[i]);
-    EXPECT_NE(dom1_instance->GetProcess(), hosts[i]);
+    EXPECT_NE(webui1_instance->GetProcess(), hosts[i]);
   }
 
   STLDeleteContainerPointers(hosts.begin(), hosts.end());
 }
 
 // Test to ensure that profiles that derive from each other share site
 // information.
 TEST_F(SiteInstanceTest, GetSiteInstanceMap) {
   int deleteCounter = 0;
 
@@ skipping 51 matching lines @@
       GURL("chrome-extension://baz/bar")));
   EXPECT_EQ(s5a, s5b);
 
   // The derived profiles with the different sites.
   scoped_refptr<SiteInstance> s6a(instance1->GetSiteInstanceForURL(
       GURL("chrome-extension://baz/bar")));
   scoped_refptr<SiteInstance> s6b(instance3->GetSiteInstanceForURL(
       GURL("chrome-extension://foo/boo")));
   EXPECT_NE(s6a, s6b);
 }