untrusted_web_ui_browsertest_util.cc 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. // Copyright 2019 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. #include "ui/webui/untrusted_web_ui_browsertest_util.h"
  5. #include "content/public/browser/web_contents.h"
  6. #include "content/public/common/url_constants.h"
  7. #include "ui/webui/untrusted_web_ui_controller.h"
  8. namespace ui {
  9. namespace {
  10. class TestUntrustedWebUIController : public ui::UntrustedWebUIController {
  11. public:
  12. TestUntrustedWebUIController(
  13. content::WebUI* web_ui,
  14. const std::string& host,
  15. const content::TestUntrustedDataSourceHeaders& headers)
  16. : ui::UntrustedWebUIController(web_ui) {
  17. content::AddUntrustedDataSource(
  18. web_ui->GetWebContents()->GetBrowserContext(), host, headers);
  19. }
  20. ~TestUntrustedWebUIController() override = default;
  21. };
  22. } // namespace
  23. TestUntrustedWebUIConfig::TestUntrustedWebUIConfig(base::StringPiece host)
  24. : WebUIConfig(content::kChromeUIUntrustedScheme, host) {}
  25. TestUntrustedWebUIConfig::TestUntrustedWebUIConfig(
  26. base::StringPiece host,
  27. const content::TestUntrustedDataSourceHeaders& headers)
  28. : WebUIConfig(content::kChromeUIUntrustedScheme, host), headers_(headers) {}
  29. TestUntrustedWebUIConfig::~TestUntrustedWebUIConfig() = default;
  30. std::unique_ptr<content::WebUIController>
  31. TestUntrustedWebUIConfig::CreateWebUIController(content::WebUI* web_ui) {
  32. return std::make_unique<TestUntrustedWebUIController>(web_ui, host(),
  33. headers_);
  34. }
  35. } // namespace ui