// Copyright 2015 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 "chrome/browser/pdf/pdf_extension_test_util.h"

#include "content/public/test/browser_test_utils.h"
#include "grit/component_extension_resources.h"
#include "ui/base/resource/resource_bundle.h"

namespace pdf_extension_test_util {

bool EnsurePDFHasLoaded(content::WebContents* web_contents) {
  std::string scripting_api_js =
      ResourceBundle::GetSharedInstance()
          .GetRawDataResource(IDR_PDF_PDF_SCRIPTING_API_JS)
          .as_string();
  CHECK(content::ExecuteScript(web_contents, scripting_api_js));

  bool load_success = false;
  CHECK(content::ExecuteScriptAndExtractBool(
      web_contents,
      "var scriptingAPI = new PDFScriptingAPI(window, "
      "    document.getElementsByTagName('embed')[0]);"
      "scriptingAPI.setLoadCallback(function(success) {"
      "  window.domAutomationController.send(success);"
      "});",
      &load_success));
  return load_success;
}

}  // namespace pdf_extension_test_util