content/browser/webui/webui_config_map_unittest.cc

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

#include "content/public/browser/webui_config_map.h"

#include "content/public/browser/web_ui_controller.h"
#include "content/public/browser/webui_config.h"
#include "content/public/test/scoped_web_ui_controller_factory_registration.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace content {

namespace {

constexpr const char kChromeFoo[] = "chrome://foo";
constexpr const char kChromeBar[] = "chrome://bar";
constexpr const char kChromeUntrustedFoo[] = "chrome-untrusted://foo";
constexpr const char kChromeUntrustedBar[] = "chrome-untrusted://bar";

// A valid BrowserContext is not needed for these tests.
BrowserContext* kBrowserContext = nullptr;

class TestConfig : public WebUIConfig {
 public:
  TestConfig(base::StringPiece scheme, base::StringPiece host)
      : WebUIConfig(scheme, host) {}
  ~TestConfig() override = default;

  std::unique_ptr<WebUIController> CreateWebUIController(
      WebUI* web_ui,
      const GURL& url) override {
    // Unused in these tests.
    return nullptr;
  }
};

}  // namespace

TEST(WebUIConfigTest, AddAndRemoveChromeUrl) {
  auto& map = WebUIConfigMap::GetInstance();
  size_t initial_size = map.GetSizeForTesting();

  auto chrome_config = std::make_unique<TestConfig>("chrome", "foo");
  auto* chrome_config_ptr = chrome_config.get();
  map.AddWebUIConfig(std::move(chrome_config));

  EXPECT_EQ(initial_size + 1, map.GetSizeForTesting());

  EXPECT_EQ(chrome_config_ptr,
            map.GetConfig(kBrowserContext, GURL(kChromeFoo)));
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, GURL(kChromeBar)));
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, GURL(kChromeUntrustedFoo)));
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, GURL(kChromeUntrustedBar)));

  EXPECT_EQ(nullptr, map.RemoveConfig(GURL(kChromeBar)));
  EXPECT_EQ(nullptr, map.RemoveConfig(GURL(kChromeUntrustedFoo)));
  EXPECT_EQ(nullptr, map.RemoveConfig(GURL(kChromeUntrustedBar)));

  auto removed_config = map.RemoveConfig(GURL(kChromeFoo));
  EXPECT_EQ(removed_config.get(), chrome_config_ptr);
  EXPECT_EQ(initial_size, map.GetSizeForTesting());
}

TEST(WebUIConfigTest, AddAndRemoteChromeUntrustedUrl) {
  auto& map = WebUIConfigMap::GetInstance();
  size_t initial_size = map.GetSizeForTesting();

  auto untrusted_config =
      std::make_unique<TestConfig>("chrome-untrusted", "foo");
  auto* untrusted_config_ptr = untrusted_config.get();
  map.AddUntrustedWebUIConfig(std::move(untrusted_config));

  EXPECT_EQ(initial_size + 1, map.GetSizeForTesting());

  EXPECT_EQ(untrusted_config_ptr,
            map.GetConfig(kBrowserContext, GURL(kChromeUntrustedFoo)));
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, GURL(kChromeUntrustedBar)));
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, GURL(kChromeFoo)));
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, GURL(kChromeBar)));

  EXPECT_EQ(nullptr, map.RemoveConfig(GURL(kChromeUntrustedBar)));
  EXPECT_EQ(nullptr, map.RemoveConfig(GURL(kChromeFoo)));
  EXPECT_EQ(nullptr, map.RemoveConfig(GURL(kChromeBar)));

  auto removed_config = map.RemoveConfig(GURL(kChromeUntrustedFoo));
  EXPECT_EQ(removed_config.get(), untrusted_config_ptr);
  EXPECT_EQ(initial_size, map.GetSizeForTesting());
}

// Regression test for https://crbug.com/1464456.
TEST(WebUIConfigTest, GetAndRemoveNonChromeUrls) {
  auto& map = WebUIConfigMap::GetInstance();

  ScopedWebUIConfigRegistration chrome_config(
      std::make_unique<TestConfig>("chrome", "foo"));
  ScopedWebUIConfigRegistration untrusted_config(
      std::make_unique<TestConfig>("chrome-untrusted", "foo"));

  // filesystem: URLs
  const GURL file_system_chrome("filesystem:chrome://foo/external/file.txt");
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, file_system_chrome));
  EXPECT_DEATH_IF_SUPPORTED(map.RemoveConfig(file_system_chrome), "");

  const GURL file_system_chrome_untrusted(
      "filesystem:chrome-untrusted://foo/external/file.txt");
  EXPECT_EQ(nullptr,
            map.GetConfig(kBrowserContext, file_system_chrome_untrusted));
  EXPECT_DEATH_IF_SUPPORTED(map.RemoveConfig(file_system_chrome_untrusted), "");

  // blob: URLs
  const GURL blob_chrome("blob:chrome://foo/GUID");
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, blob_chrome));
  EXPECT_DEATH_IF_SUPPORTED(map.RemoveConfig(blob_chrome), "");

  const GURL blob_chrome_untrusted("blob:chrome-untrusted://foo/GUID");
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, blob_chrome_untrusted));
  EXPECT_DEATH_IF_SUPPORTED(map.RemoveConfig(blob_chrome_untrusted), "");

  // HTTP/HTTPS URLs
  const GURL http("http://foo.com");
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, http));
  EXPECT_DEATH_IF_SUPPORTED(map.RemoveConfig(http), "");

  const GURL https("https://foo.com");
  EXPECT_EQ(nullptr, map.GetConfig(kBrowserContext, https));
  EXPECT_DEATH_IF_SUPPORTED(map.RemoveConfig(https), "");
}

}  // namespace content