Catch CSP violations in DOM distiller browser test

chrome/browser/dom_distiller/tab_utils_browsertest.cc

 // Copyright 2014 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 <string.h>
 
 #include "base/command_line.h"
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "chrome/browser/dom_distiller/dom_distiller_service_factory.h"
@@ skipping 14 matching lines @@
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/common/isolated_world_ids.h"
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/test_utils.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace dom_distiller {
 
 namespace {
+
 const char* kSimpleArticlePath = "/dom_distiller/simple_article.html";
+
+static bool hit_javascript_error = false;
+
+bool JavascriptErrorDetectingLogHandler(int severity,
+                                        const char* file,
+                                        int line,
+                                        size_t message_start,
+                                        const std::string& str) {
+  if (file == NULL || std::string("CONSOLE") != file)
+    return false;
+
+  // Catch CSP violation as well.
+  bool contains_error = str.find(" violates ") != std::string::npos;
+  if (severity == logging::LOG_ERROR ||
+      (severity == logging::LOG_INFO && contains_error)) {
+    hit_javascript_error = true;
+  }
+
+  return false;
+}
+
 }  // namespace
 
 class DomDistillerTabUtilsBrowserTest : public InProcessBrowserTest {
  public:
+  DomDistillerTabUtilsBrowserTest() {
+    logging::SetLogMessageHandler(&JavascriptErrorDetectingLogHandler);
+    hit_javascript_error = false;
+  }
+
+  ~DomDistillerTabUtilsBrowserTest() override {
+    logging::SetLogMessageHandler(nullptr);
+  }
+
   void SetUpOnMainThread() override {
     if (!DistillerJavaScriptWorldIdIsSet()) {
       SetDistillerJavaScriptWorldId(content::ISOLATED_WORLD_ID_CONTENT_END);
     }
     InProcessBrowserTest::SetUpOnMainThread();
   }
 
   void SetUpCommandLine(base::CommandLine* command_line) override {
     command_line->AppendSwitch(switches::kEnableDomDistiller);
   }
+
+  void TearDown() override {
+    EXPECT_FALSE(hit_javascript_error);
+  }
 };
 
 // WebContentsMainFrameHelper is used to detect if a distilled page has
 // finished loading. This is done by checking how many times the title has
 // been set rather than using "DidFinishLoad" directly due to the content
 // being set by JavaScript.
 class WebContentsMainFrameHelper : public content::WebContentsObserver {
  public:
   WebContentsMainFrameHelper(content::WebContents* web_contents,
                              const base::Closure& callback)
@@ skipping 113 matching lines @@
                                     "document.title")->GetAsString(&page_title);
   EXPECT_EQ("Test Page Title", page_title);
 
   content::WebContentsDestroyedWatcher destroyed_watcher(
       destination_web_contents);
   browser()->tab_strip_model()->CloseWebContentsAt(1, 0);
   destroyed_watcher.Wait();
 }
 
 }  // namespace dom_distiller