ui/webui/untrusted_web_ui_browsertest_util.cc

// Copyright 2019 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 "ui/webui/untrusted_web_ui_browsertest_util.h"

#include "content/public/browser/web_contents.h"
#include "content/public/common/url_constants.h"

namespace ui {

TestUntrustedWebUIControllerFactory::TestUntrustedWebUIControllerFactory() =
    default;

TestUntrustedWebUIControllerFactory::~TestUntrustedWebUIControllerFactory() =
    default;

const ui::UntrustedWebUIControllerFactory::WebUIConfigMap&
TestUntrustedWebUIControllerFactory::GetWebUIConfigMap() {
  return configs_;
}

TestUntrustedWebUIConfig::TestUntrustedWebUIConfig(base::StringPiece host)
    : WebUIConfig(content::kChromeUIUntrustedScheme, host) {}

TestUntrustedWebUIConfig::TestUntrustedWebUIConfig(
    base::StringPiece host,
    const content::TestUntrustedDataSourceHeaders& headers)
    : WebUIConfig(content::kChromeUIUntrustedScheme, host), headers_(headers) {}

TestUntrustedWebUIConfig::~TestUntrustedWebUIConfig() = default;

std::unique_ptr<content::WebUIController>
TestUntrustedWebUIConfig::CreateWebUIController(content::WebUI* web_ui) {
  return std::make_unique<TestUntrustedWebUIController>(web_ui, host(),
                                                        headers_);
}

TestUntrustedWebUIController::TestUntrustedWebUIController(
    content::WebUI* web_ui,
    const std::string& host,
    const content::TestUntrustedDataSourceHeaders& headers)
    : ui::UntrustedWebUIController(web_ui) {
  content::AddUntrustedDataSource(web_ui->GetWebContents()->GetBrowserContext(),
                                  host, headers);
}

TestUntrustedWebUIController::~TestUntrustedWebUIController() = default;

}  // namespace ui