chromium/components/services/app_service/public/cpp/shortcut/shortcut_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 "components/services/app_service/public/cpp/shortcut/shortcut.h"

#include <memory>
#include <optional>

#include "testing/gtest/include/gtest/gtest.h"

namespace apps {

class ShortcutTest : public testing::Test {};

TEST_F(ShortcutTest, CreateShortcut) {
  std::string host_app_id = "host_app_id";
  std::string local_id = "local_id";
  auto shortcut = std::make_unique<Shortcut>(host_app_id, local_id);
  shortcut->name = "name";
  shortcut->shortcut_source = ShortcutSource::kUser;
  shortcut->icon_key = IconKey();
  shortcut->icon_key->update_version = 100;
  shortcut->allow_removal = true;

  EXPECT_EQ(shortcut->shortcut_id, GenerateShortcutId(host_app_id, local_id));
  EXPECT_EQ(shortcut->name, "name");
  EXPECT_EQ(shortcut->shortcut_source, ShortcutSource::kUser);
  EXPECT_EQ(shortcut->host_app_id, host_app_id);
  EXPECT_EQ(shortcut->local_id, local_id);
  IconKey icon_key(0, 0);
  icon_key.update_version = 100;
  EXPECT_EQ(shortcut->icon_key, icon_key);
  EXPECT_EQ(shortcut->allow_removal, true);
}

TEST_F(ShortcutTest, CloneShortcut) {
  std::string host_app_id = "host_app_id";
  std::string local_id = "local_id";
  auto shortcut = std::make_unique<Shortcut>(host_app_id, local_id);
  ShortcutId shortcut_id = shortcut->shortcut_id;
  shortcut->name = "name";
  shortcut->shortcut_source = ShortcutSource::kUser;
  shortcut->icon_key = IconKey();
  shortcut->icon_key->update_version = 100;
  shortcut->allow_removal = true;

  auto cloned_shortcut = shortcut->Clone();

  EXPECT_EQ(cloned_shortcut->shortcut_id, shortcut_id);
  EXPECT_EQ(cloned_shortcut->name, "name");
  EXPECT_EQ(cloned_shortcut->shortcut_source, ShortcutSource::kUser);
  EXPECT_EQ(cloned_shortcut->host_app_id, host_app_id);
  EXPECT_EQ(cloned_shortcut->local_id, local_id);
  IconKey icon_key;
  icon_key.update_version = 100;
  EXPECT_EQ(cloned_shortcut->icon_key, icon_key);
  EXPECT_TRUE(cloned_shortcut->allow_removal);
}

TEST_F(ShortcutTest, CloneShortcutWithNoIconKey) {
  std::string host_app_id = "host_app_id";
  std::string local_id = "local_id";
  auto shortcut = std::make_unique<Shortcut>(host_app_id, local_id);
  ShortcutId shortcut_id = shortcut->shortcut_id;
  shortcut->name = "name";
  shortcut->shortcut_source = ShortcutSource::kUser;
  shortcut->allow_removal = true;

  auto cloned_shortcut = shortcut->Clone();

  EXPECT_EQ(cloned_shortcut->shortcut_id, shortcut_id);
  EXPECT_EQ(cloned_shortcut->name, "name");
  EXPECT_EQ(cloned_shortcut->shortcut_source, ShortcutSource::kUser);
  EXPECT_EQ(cloned_shortcut->host_app_id, host_app_id);
  EXPECT_EQ(cloned_shortcut->local_id, local_id);
  EXPECT_EQ(cloned_shortcut->icon_key, std::nullopt);
  EXPECT_TRUE(cloned_shortcut->allow_removal);
}

}  // namespace apps