1 // Copyright 2015 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #include "chrome/browser/pdf/pdf_extension_test_util.h"
6 
7 #include "content/public/test/browser_test_utils.h"
8 
9 namespace pdf_extension_test_util {
10 
EnsurePDFHasLoaded(content::WebContents * web_contents)11 bool EnsurePDFHasLoaded(content::WebContents* web_contents) {
12   bool load_success = false;
13   CHECK(content::ExecuteScriptAndExtractBool(
14       web_contents,
15       "window.addEventListener('message', event => {"
16       "  if (event.origin !="
17       "          'chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai' ||"
18       "      event.data.type != 'documentLoaded') {"
19       "    return;"
20       "  }"
21       "  window.domAutomationController.send("
22       "       event.data.load_state == 'success');"
23       "});"
24       "document.getElementsByTagName('embed')[0].postMessage("
25       "    {type: 'initialize'});",
26       &load_success));
27   return load_success;
28 }
29 
30 }  // namespace pdf_extension_test_util
31