ash/assistant/util/resource_util_unittest.cc

// Copyright 2020 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 "ash/assistant/util/resource_util.h"

#include <map>
#include <string>

#include "ash/assistant/util/test_support/macros.h"
#include "ash/public/cpp/vector_icons/vector_icons.h"
#include "net/base/url_util.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/gfx/canvas.h"
#include "url/gurl.h"

namespace ash {
namespace assistant {
namespace util {

namespace {

// Helpers ---------------------------------------------------------------------

std::string GetParamValue(const GURL& url, const std::string& param_key) {
  if (!url.has_query())
    return std::string();

  auto ToString = [&url](const url::Component& component) {
    if (component.len <= 0)
      return std::string();
    return std::string(url.query(), component.begin, component.len);
  };

  url::Component query(0, url.query().length()), key, value;
  while (url::ExtractQueryKeyValue(url.query().c_str(), &query, &key, &value)) {
    if (ToString(key) == param_key)
      return ToString(value);
  }

  return std::string();
}

}  // namespace

// Tests -----------------------------------------------------------------------

using ResourceUtilTest = testing::Test;

TEST_F(ResourceUtilTest, AppendOrReplaceColorParam) {
  GURL icon_link("googleassistant://resource?type=icon");

  // Test append.
  icon_link = AppendOrReplaceColorParam(icon_link, SK_ColorBLACK);
  EXPECT_EQ(GetParamValue(icon_link, "color"), "ff000000");

  // Test replace.
  icon_link = AppendOrReplaceColorParam(icon_link, SK_ColorWHITE);
  EXPECT_EQ(GetParamValue(icon_link, "color"), "ffffffff");
}

TEST_F(ResourceUtilTest, CreateIconResourceLink) {
  // Test w/ color.
  EXPECT_EQ(CreateIconResourceLink(IconName::kAssistant, SK_ColorBLACK),
            GURL("googleassistant://"
                 "resource?type=icon&name=assistant&color=ff000000"));
  // Test w/o color.
  EXPECT_EQ(CreateIconResourceLink(IconName::kAssistant),
            GURL("googleassistant://resource?type=icon&name=assistant"));
}

TEST_F(ResourceUtilTest, CreateVectorIcon) {
  std::string url;
  gfx::ImageSkia actual, expected;

  // Test w/ name.
  url = "googleassistant://resource?type=icon&name=assistant";
  actual = CreateVectorIcon(GURL(url));
  expected = gfx::CreateVectorIcon(gfx::IconDescription(ash::kAssistantIcon));
  ASSERT_PIXELS_EQ(actual, expected);

  // Test w/ name and size.
  url = "googleassistant://resource?type=icon&name=assistant";
  actual = CreateVectorIcon(GURL(url), /*size=*/24);
  expected = gfx::CreateVectorIcon(
      gfx::IconDescription(ash::kAssistantIcon, /*size=*/24));
  ASSERT_PIXELS_EQ(actual, expected);

  // Test w/ name, size, and color.
  url = "googleassistant://resource?type=icon&name=assistant&color=ff000000";
  actual = CreateVectorIcon(GURL(url), /*size=*/24);
  expected = gfx::CreateVectorIcon(gfx::IconDescription(
      ash::kAssistantIcon, /*size=*/24, /*color=*/SK_ColorBLACK));
  ASSERT_PIXELS_EQ(actual, expected);

  // Test w/o icon resource link.
  actual = CreateVectorIcon(GURL("googleassistant://resource"));
  EXPECT_TRUE(actual.isNull());

  // Test w/o resource link.
  actual = CreateVectorIcon(GURL("https://g.co/"));
  EXPECT_TRUE(actual.isNull());

  // Test w/ empty.
  actual = CreateVectorIcon(GURL());
  EXPECT_TRUE(actual.isNull());
}

TEST_F(ResourceUtilTest, GetResourceLinkType) {
  const std::map<std::string, ResourceLinkType> test_cases = {
      // SUPPORTED:
      {"googleassistant://resource?type=icon", ResourceLinkType::kIcon},

      // UNSUPPORTED:
      {"googleassistant://resource", ResourceLinkType::kUnsupported},
      {"GOOGLEASSISTANT://RESOURCE?TYPE=ICON", ResourceLinkType::kUnsupported},
      {"https://g.co/", ResourceLinkType::kUnsupported},
      {"", ResourceLinkType::kUnsupported},
  };
  for (const auto& test_case : test_cases)
    EXPECT_EQ(GetResourceLinkType(GURL(test_case.first)), test_case.second);
}

TEST_F(ResourceUtilTest, IsResourceLinkType) {
  const std::map<std::string, ResourceLinkType> test_cases = {
      // SUPPORTED:
      {"googleassistant://resource?type=icon", ResourceLinkType::kIcon},

      // UNSUPPORTED:
      {"googleassistant://resource", ResourceLinkType::kUnsupported},
      {"GOOGLEASSISTANT://RESOURCE?TYPE=ICON", ResourceLinkType::kUnsupported},
      {"https://g.co/", ResourceLinkType::kUnsupported},
      {"", ResourceLinkType::kUnsupported},
  };
  for (const auto& test_case : test_cases)
    EXPECT_TRUE(IsResourceLinkType(GURL(test_case.first), test_case.second));
}

TEST_F(ResourceUtilTest, IsResourceLinkUrl) {
  const std::map<std::string, bool> test_cases = {
      // RESOURCE LINK:
      {"googleassistant://resource", true},
      {"googleassistant://resource?type=icon", true},

      // NOT RESOURCE LINK:
      {"GOOGLEASSISTANT://RESOURCE", false},
      {"https://g.co/", false},
      {"", false},
  };
  for (const auto& test_case : test_cases)
    EXPECT_EQ(IsResourceLinkUrl(GURL(test_case.first)), test_case.second);
}

}  // namespace util
}  // namespace assistant
}  // namespace ash