chromeos/crosapi/cpp/gurl_os_handler_unittest.cc

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

#include "chromeos/crosapi/cpp/gurl_os_handler_utils.h"

#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"
#include "url/url_util.h"

namespace crosapi {

namespace gurl_os_handler_utils {

TEST(GurlOsHandlerUtilsTest, GetAshUrlFromLacrosUrl) {
  // To allow the "chrome" scheme, we need to add it to the registry.
  url::ScopedSchemeRegistryForTests scoped_registry;
  url::AddStandardScheme("chrome", url::SCHEME_WITH_HOST);
  CHECK(GURL("chrome://version").has_host());

  EXPECT_EQ(GetAshUrlFromLacrosUrl(GURL("http://abc:123/def/ghi?jkl#mno")),
            GURL("http://abc:123/def/ghi?jkl#mno"));

  EXPECT_EQ(GetAshUrlFromLacrosUrl(GURL("chrome://abc/def/ghi?jkl#mno")),
            GURL("chrome://abc/def/ghi?jkl#mno"));

  EXPECT_EQ(GetAshUrlFromLacrosUrl(GURL("os://abc/def/ghi?jkl#mno")),
            GURL("chrome://abc/def/ghi?jkl#mno"));

  // os://settings is mapped to chrome://os-settings.
  EXPECT_EQ(GetAshUrlFromLacrosUrl(GURL("os://settings/def/ghi?jkl#mno")),
            GURL("chrome://os-settings/def/ghi?jkl#mno"));
}

TEST(GurlOsHandlerUtilsTest, SanitizeAshUrl) {
  // To allow the "chrome" scheme, we need to add it to the registry.
  url::ScopedSchemeRegistryForTests scoped_registry;
  url::AddStandardScheme("chrome", url::SCHEME_WITH_HOST);
  CHECK(GURL("chrome://version").has_host());

  // Invalid examples.
  EXPECT_EQ(SanitizeAshUrl(GURL("os://version")), GURL());  // Unknown scheme.
  EXPECT_EQ(SanitizeAshUrl(GURL("chrome://")), GURL());     // No host.

  // Valid examples.
  EXPECT_EQ(SanitizeAshUrl(GURL("chrome://version")), GURL("chrome://version"));
  EXPECT_EQ(SanitizeAshUrl(GURL("chrome://abc/def/ghi?jkl")),
            GURL("chrome://abc/def/ghi?jkl"));  // Query preserved.
  EXPECT_EQ(SanitizeAshUrl(GURL("chrome://abc:123/def/ghi?jkl#mno")),
            GURL("chrome://abc/def/ghi?jkl"));  // Port and ref removed.
  EXPECT_EQ(SanitizeAshUrl(GURL("https://abc:123/def/ghi?jkl#mno")),
            GURL("https://abc/def/ghi?jkl"));  // Here too.
}

TEST(GurlOsHandlerUtilsTest, IsAshUrlInList) {
  // To allow the "chrome" scheme, we need to add it to the registry.
  url::ScopedSchemeRegistryForTests scoped_registry;
  url::AddStandardScheme("chrome", url::SCHEME_WITH_HOST);
  CHECK(GURL("chrome://version").has_host());

  std::vector<GURL> list_of_urls = {GURL("chrome://foo")};

  EXPECT_FALSE(IsAshUrlInList(GURL("os://version"), list_of_urls));
  EXPECT_FALSE(IsAshUrlInList(GURL("chrome://version"), list_of_urls));

  EXPECT_FALSE(IsAshUrlInList(GURL("os://foo"), list_of_urls));
  EXPECT_FALSE(IsAshUrlInList(GURL("chrome://fooz"), list_of_urls));

  EXPECT_TRUE(IsAshUrlInList(GURL("chrome://foo"), list_of_urls));
  EXPECT_TRUE(IsAshUrlInList(GURL("chrome://foo?baz"), list_of_urls));
  EXPECT_TRUE(IsAshUrlInList(GURL("chrome://foo#baz"), list_of_urls));
  EXPECT_TRUE(IsAshUrlInList(GURL("chrome://foo?bar#baz"), list_of_urls));
  EXPECT_TRUE(IsAshUrlInList(GURL("chrome://foo/bar"), list_of_urls));
  EXPECT_TRUE(IsAshUrlInList(GURL("chrome://foo/bar?baz"), list_of_urls));
  EXPECT_TRUE(IsAshUrlInList(GURL("chrome://foo/bar#baz"), list_of_urls));
  EXPECT_TRUE(IsAshUrlInList(GURL("chrome://foo/bar?baz#bax"), list_of_urls));
}

TEST(GurlOsHandlerUtilsTest, HasOsScheme) {
  EXPECT_TRUE(HasOsScheme(GURL("os://version?foo#bar")));
  EXPECT_TRUE(HasOsScheme(GURL("os://flags")));
  EXPECT_TRUE(HasOsScheme(GURL("OS://flags")));     // case insensitive.
  EXPECT_FALSE(HasOsScheme(GURL("os:/flags")));     // Proper '://' required.
  EXPECT_FALSE(HasOsScheme(GURL("os://")));         // There needs to be a host.
  EXPECT_FALSE(HasOsScheme(GURL("oo://version")));  // scheme need matching.
  EXPECT_FALSE(HasOsScheme(GURL("")));              // No crash.
  EXPECT_FALSE(HasOsScheme(GURL("::/bar")));        // No crash.
  EXPECT_FALSE(HasOsScheme(GURL("osos::/bar")));    // In string correct.
}

TEST(GurlOsHandlerUtilsTest, IsOsScheme) {
  EXPECT_TRUE(IsOsScheme("os"));
  EXPECT_TRUE(IsOsScheme("Os"));
  EXPECT_FALSE(IsOsScheme("oss"));
  EXPECT_FALSE(IsOsScheme(""));  // Should not crash.
}

}  // namespace gurl_os_handler_utils

}  // namespace crosapi