Fix for single-process WebUI transition CHECK failure.

content/browser/frame_host/render_frame_host_manager_unittest.cc

 // Copyright 2013 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/command_line.h"
 #include "base/files/file_path.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/histogram_tester.h"
 #include "base/time/time.h"
 #include "content/browser/compositor/test/no_transport_image_transport_factory.h"
@@ skipping 1220 matching lines @@
   const GURL kUrl1("chrome://foo");
   contents()->NavigateAndCommit(kUrl1);
   EXPECT_TRUE(main_test_rfh()->frame_tree_node()->render_manager()->web_ui());
 
   // Navigate to a non-WebUI page.
   const GURL kUrl2("http://www.google.com");
   contents()->NavigateAndCommit(kUrl2);
   EXPECT_FALSE(main_test_rfh()->frame_tree_node()->render_manager()->web_ui());
 }
 
+// Test that WebUI can be navigated to in single-process mode.
+TEST_F(RenderFrameHostManagerTest,
+       CanNavigateBetweenNormalPagesAndWebUIInSingleProcess) {
+  const GURL kNormalUrl1("http://chromium.org");
+  const GURL kNormalUrl2("http://example.com");
+  const GURL kWebUIUrl("chrome://foo");
+
+  // We both set flag on RenderProcesHost and append the switch due to the fact
+  // that some code checks the switch directly instead of relying
+  // on RenderProcessHost
+  RenderProcessHost::SetRunRendererInProcess(true);
+  CommandLine::ForCurrentProcess()->AppendSwitch(switches::kSingleProcess);
+  set_should_create_webui(true);
+
+  RenderFrameHostManager* manager = static_cast<TestWebContents*>(
+      web_contents())->GetRenderManagerForTesting();
+
+  NavigateAndCommit(kNormalUrl1);
+  EXPECT_EQ(kNormalUrl1,
+            manager->current_frame_host()->GetSiteInstance()->GetSiteURL());
+  EXPECT_FALSE(manager->web_ui());
+  NavigateAndCommit(kWebUIUrl);
+  EXPECT_EQ(kWebUIUrl,
+            manager->current_frame_host()->GetSiteInstance()->GetSiteURL());
+  EXPECT_TRUE(manager->web_ui());
+  NavigateAndCommit(kNormalUrl2);
+  EXPECT_EQ(kNormalUrl2,
+            manager->current_frame_host()->GetSiteInstance()->GetSiteURL());
+  EXPECT_FALSE(manager->web_ui());
+}
+
 // Tests that we don't end up in an inconsistent state if a page does a back and
 // then reload. http://crbug.com/51680
 // Also tests that only user-gesture navigations can interrupt cross-process
 // navigations. http://crbug.com/75195
 TEST_F(RenderFrameHostManagerTest, PageDoesBackAndReload) {
   const GURL kUrl1("http://www.google.com/");
   const GURL kUrl2("http://www.evil-site.com/");
 
   // Navigate to a safe site, then an evil site.
   // This will switch RenderFrameHosts.  We cannot assert that the first and
@@ skipping 993 matching lines @@
   EXPECT_FALSE(contents2->GetMainFrame()->IsRenderFrameLive());
   contents2->NavigateAndCommit(kUrl3);
   EXPECT_TRUE(contents2->GetMainFrame()->IsRenderFrameLive());
   EXPECT_NE(nullptr,
             iframe->GetRenderFrameProxyHost(contents1->GetSiteInstance()));
   EXPECT_EQ(nullptr,
             iframe->GetRenderFrameProxyHost(contents2->GetSiteInstance()));
 }
 
 }  // namespace content