Block webpages from navigating to view-source URLs

content/browser/web_contents/web_contents_impl_browsertest.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/macros.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "content/browser/frame_host/navigation_entry_impl.h"
 #include "content/browser/renderer_host/render_widget_host_impl.h"
@@ skipping 752 matching lines @@
   shell()->web_contents()->SetPageScale(1.5);
   EXPECT_CALL(observer, OnPageScaleFactorChanged(::testing::FloatEq(1.5)));
   observer.WaitForPageScaleUpdate();
 
   // Navigate to reset the page scale factor.
   shell()->LoadURL(embedded_test_server()->GetURL("/title2.html"));
   EXPECT_CALL(observer, OnPageScaleFactorChanged(::testing::_));
   observer.WaitForPageScaleUpdate();
 }
 
+// Test view source mode for a webui page.
+IN_PROC_BROWSER_TEST_F(WebContentsImplBrowserTest, WebUIViewSource) {

[Charlie Reis, 2016/05/16 22:22:18]

Seems like a nice test.  Was this behavior at risk in one of your attempts, or
is this more of a general sanity check that we can do browser-initiated
view-source for WebUI?

[meacer, 2016/05/19 01:13:54]

I didn't run into this case in my attempts, but I got the idea from
WebContentsImplTest.ContentInitiatedViewSource test. I fixed the other test now,
but it seemed fragile because it manually constructed the URL to be navigated
to.

+  const char kUrl[] = "view-source:chrome://chrome/settings";
+  const GURL kGURL(kUrl);
+  NavigateToURL(shell(), kGURL);
+  EXPECT_EQ(base::ASCIIToUTF16(kUrl), shell()->web_contents()->GetTitle());
+}
+
 IN_PROC_BROWSER_TEST_F(WebContentsImplBrowserTest, NewNamedWindow) {
   ASSERT_TRUE(embedded_test_server()->Start());
 
   GURL url = embedded_test_server()->GetURL("/click-noreferrer-links.html");
   EXPECT_TRUE(NavigateToURL(shell(), url));
 
   {
     ShellAddedObserver new_shell_observer;
 
     // Open a new, named window.
@@ skipping 222 matching lines @@
       content::ExecuteScript(root->current_frame_host(), alert_location));
   dialog_manager.Wait();
   EXPECT_EQ(GURL("http://a.com/title1.html"),
             GURL(dialog_manager.last_message()).ReplaceComponents(clear_port));
 
   wc->SetDelegate(nullptr);
   wc->SetJavaScriptDialogManagerForTesting(nullptr);
 }
 
 }  // namespace content