Block webpages from navigating to view-source URLs

content/public/test/content_browser_test_utils.h

 // Copyright (c) 2012 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.
 
 #ifndef CONTENT_PUBLIC_TEST_CONTENT_BROWSER_TEST_UTILS_H_
 #define CONTENT_PUBLIC_TEST_CONTENT_BROWSER_TEST_UTILS_H_
 
 #include "base/callback.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
+#include "content/public/browser/web_contents_delegate.h"
 #include "content/public/common/page_type.h"
 #include "ui/gfx/native_widget_types.h"
 #include "url/gurl.h"
 
 namespace base {
 class FilePath;
 }
 
 namespace gfx {
 class Rect;
 }
 
 // A collections of functions designed for use with content_shell based browser
 // tests.
 // Note: if a function here also works with browser_tests, it should be in
 // content\public\test\browser_test_utils.h
 
 namespace content {
 
 class MessageLoopRunner;
 class Shell;
+class WebContents;
 
 // Generate the file path for testing a particular test.
 // The file for the tests is all located in
 // content/test/data/dir/<file>
 // The returned path is FilePath format.
 base::FilePath GetTestFilePath(const char* dir, const char* file);
 
 // Generate the URL for testing a particular test.
 // HTML for the tests is all located in
 // test_root_directory/dir/<file>
@@ skipping 49 matching lines @@
 
  private:
   void ShellCreated(Shell* shell);
 
   Shell* shell_;
   scoped_refptr<MessageLoopRunner> runner_;
 
   DISALLOW_COPY_AND_ASSIGN(ShellAddedObserver);
 };
 
+// A WebContentsDelegate that catches messages sent to the console.
+class ConsoleObserverDelegate : public WebContentsDelegate {
+ public:
+  ConsoleObserverDelegate(WebContents* web_contents, const std::string& filter);
+
+  ~ConsoleObserverDelegate() override;

[Charlie Reis, 2016/05/24 22:54:17]

nit: No blank line above.

[meacer, 2016/05/31 23:57:54]

Done.

+
+  bool AddMessageToConsole(WebContents* source,

[Charlie Reis, 2016/05/24 22:54:17]

nit: // WebContentsDelegate overrides.

[meacer, 2016/05/31 23:57:54]

Done.

+                           int32_t level,
+                           const base::string16& message,
+                           int32_t line_no,
+                           const base::string16& source_id) override;
+
+  std::string message() { return message_; }

[Charlie Reis, 2016/05/24 22:54:17]

nit: This is now a public interface, so it needs comments on these methods. 
Something like "Returns the most recent message sent to the console."

[meacer, 2016/05/31 23:57:54]

Done.

+
+  void Wait();

[Charlie Reis, 2016/05/24 22:54:17]

// Wait for the next message sent to the console.

[meacer, 2016/05/31 23:57:54]

Done.

+
+ private:
+  WebContents* web_contents_;
+  std::string filter_;
+  std::string message_;
+
+  // The MessageLoopRunner used to spin the message loop.
+  scoped_refptr<MessageLoopRunner> message_loop_runner_;
+
+  DISALLOW_COPY_AND_ASSIGN(ConsoleObserverDelegate);
+};
+
 #if defined OS_MACOSX
 void SetWindowBounds(gfx::NativeWindow window, const gfx::Rect& bounds);
 #endif
 
 }  // namespace content
 
 #endif  // CONTENT_PUBLIC_TEST_CONTENT_BROWSER_TEST_UTILS_H_