// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <string>

#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/extensions/extension_browsertest.h"
#include "content/public/browser/navigation_controller.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/browser_test_utils.h"
#include "extensions/buildflags/buildflags.h"
#include "extensions/common/constants.h"
#include "extensions/common/extension.h"
#include "net/test/embedded_test_server/embedded_test_server.h"

static_assert(BUILDFLAG(ENABLE_EXTENSIONS_CORE));

namespace extensions {
namespace {

// Tests that the chrome-extension scheme disallows running Javascript URLs.
IN_PROC_BROWSER_TEST_F(ExtensionBrowserTest,
                       ChromeExtensionSchemeNotAllowJavascript) {
  ASSERT_TRUE(embedded_test_server()->Start());

  const Extension* extension =
      LoadExtension(test_data_dir_.AppendASCII("simple_with_file"));
  ASSERT_TRUE(extension);

  // Navigate to the extension's page.
  const GURL extension_file_url(extension->GetResourceURL("file.html"));
  ASSERT_TRUE(NavigateToURL(GetActiveWebContents(), extension_file_url));

  content::WebContents* web_contents = GetActiveWebContents();
  const std::u16string expected_title = u"foo";
  ASSERT_EQ(expected_title, web_contents->GetTitle());

  // Attempt to set the page title via Javascript. Don't try to block since
  // the javascript URL won't actually navigate anywhere.
  const GURL script_url("javascript:void(document.title='Bad Title')");
  content::NavigationController::LoadURLParams load_params(script_url);
  web_contents->GetController().LoadURLWithParams(load_params);

  // Force serialization with the renderer by executing a no-op script.
  EXPECT_EQ(true, content::EvalJs(web_contents, "true"));

  // Expect the title hasn't changed since the javascript URL was blocked
  // from executing.
  EXPECT_EQ(expected_title, web_contents->GetTitle());
}

}  // namespace
}  // namespace extensions