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)11bool 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