chrome/browser/ash/guest_os/guest_os_terminal_unittest.cc

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

#include "chrome/browser/ash/guest_os/guest_os_terminal.h"

#include "base/json/json_reader.h"
#include "base/values.h"
#include "chrome/browser/ash/crostini/crostini_pref_names.h"
#include "chrome/browser/ash/crostini/crostini_util.h"
#include "chrome/browser/ash/guest_os/guest_id.h"
#include "chrome/browser/ash/guest_os/guest_os_pref_names.h"
#include "chrome/browser/ash/guest_os/public/types.h"
#include "chrome/test/base/testing_profile.h"
#include "components/prefs/pref_service.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/skia/include/core/SkColor.h"

namespace guest_os {

using CrostiniTerminalTest = testing::Test;

TEST_F(CrostiniTerminalTest, GenerateTerminalURL) {
  content::BrowserTaskEnvironment task_environment;
  TestingProfile profile;
  EXPECT_EQ(
      GenerateTerminalURL(&profile, "", crostini::DefaultContainerId(), "", {}),
      "chrome-untrusted://terminal/html/terminal.html"
      "?command=vmshell"
      "&args[]=--vm_name%3Dtermina"
      "&args[]=--target_container%3Dpenguin"
      "&args[]=--owner_id%3Dtest");
  EXPECT_EQ(GenerateTerminalURL(
                &profile, "red",
                guest_os::GuestId(VmType::TERMINA, "test-vm", "test-container"),
                "/home/user", {"arg1"}),
            "chrome-untrusted://terminal/html/terminal.html"
            "?command=vmshell"
            "&settings_profile=red"
            "&args[]=--vm_name%3Dtest-vm"
            "&args[]=--target_container%3Dtest-container"
            "&args[]=--owner_id%3Dtest"
            "&args[]=--cwd%3D%2Fhome%2Fuser"
            "&args[]=--&args[]=arg1");
}

TEST_F(CrostiniTerminalTest, ShortcutIdForSSH) {
  EXPECT_EQ(ShortcutIdForSSH("test-profile"),
            R"({"profileId":"test-profile","shortcut":"ssh"})");
}

TEST_F(CrostiniTerminalTest, ShortcutIdFromContainerId) {
  content::BrowserTaskEnvironment task_environment;
  TestingProfile profile;
  guest_os::GuestId id(VmType::TERMINA, "test-vm", "test-container");
  std::string shortcut = ShortcutIdFromContainerId(&profile, id);
  EXPECT_EQ(shortcut, R"({"container_name":"test-container",)"
                      R"("shortcut":"terminal",)"
                      R"("vm_name":"test-vm",)"
                      R"("vm_type":0})");
  auto extras = ExtrasFromShortcutId(*base::JSONReader::ReadDict(
      shortcut, base::JSON_PARSE_CHROMIUM_EXTENSIONS));
  EXPECT_EQ(3u, extras.size());

  // Container with multi-profile should include settings_profile.
  auto pref = base::JSONReader::Read(R"({
    "/vsh/profile-ids": ["p1", "p2"],
    "/vsh/profiles/p1/vm-name": "test-vm",
    "/vsh/profiles/p1/container-name": "other-container",
    "/vsh/profiles/p1/terminal-profile": "red",
    "/vsh/profiles/p2/vm-name": "test-vm",
    "/vsh/profiles/p2/container-name": "test-container",
    "/vsh/profiles/p2/terminal-profile": "green"
  })",
                                     base::JSON_PARSE_CHROMIUM_EXTENSIONS);
  ASSERT_TRUE(pref.has_value());
  profile.GetPrefs()->Set(guest_os::prefs::kGuestOsTerminalSettings,
                          std::move(*pref));
  shortcut = ShortcutIdFromContainerId(&profile, id);
  EXPECT_EQ(shortcut, R"({"container_name":"test-container",)"
                      R"("settings_profile":"green",)"
                      R"("shortcut":"terminal",)"
                      R"("vm_name":"test-vm",)"
                      R"("vm_type":0})");
  extras = ExtrasFromShortcutId(*base::JSONReader::ReadDict(
      shortcut, base::JSON_PARSE_CHROMIUM_EXTENSIONS));
  EXPECT_EQ(4u, extras.size());
}

TEST_F(CrostiniTerminalTest, GetSSHConnections) {
  content::BrowserTaskEnvironment task_environment;
  TestingProfile profile;

  std::vector<std::pair<std::string, std::string>> expected;
  EXPECT_EQ(GetSSHConnections(&profile), expected);

  auto pref = base::JSONReader::Read(R"({
    "/nassh/profile-ids": ["p1", "p2", "p3"],
    "/nassh/profiles/p1/description": "d1",
    "/nassh/profiles/p2/description": "d2"
  })",
                                     base::JSON_PARSE_CHROMIUM_EXTENSIONS);
  ASSERT_TRUE(pref.has_value());
  profile.GetPrefs()->Set(guest_os::prefs::kGuestOsTerminalSettings,
                          std::move(*pref));

  expected = {{"p1", "d1"}, {"p2", "d2"}};
  EXPECT_EQ(GetSSHConnections(&profile), expected);
}

TEST_F(CrostiniTerminalTest, GetTerminalSettingBackgroundColor) {
  content::BrowserTaskEnvironment task_environment;
  TestingProfile profile;

  auto pref = base::JSONReader::Read(R"({
    "/hterm/profiles/default/background-color": "#101010",
    "/hterm/profiles/red/background-color": "#FF0000"
  })",
                                     base::JSON_PARSE_CHROMIUM_EXTENSIONS);
  ASSERT_TRUE(pref.has_value());
  profile.GetPrefs()->Set(guest_os::prefs::kGuestOsTerminalSettings,
                          std::move(*pref));

  // Use settings_profile param.
  EXPECT_EQ(GetTerminalSettingBackgroundColor(
                &profile,
                GURL("chrome-untrusted://terminal/html/"
                     "terminal.html?settings_profile=red"),
                SK_ColorGREEN),
            "#FF0000");

  // Use opener color.
  EXPECT_EQ(
      GetTerminalSettingBackgroundColor(
          &profile, GURL("chrome-untrusted://terminal/html/terminal.html"),
          SK_ColorGREEN),
      "#00ff00ff");

  // Use color from 'default' profile.
  EXPECT_EQ(
      GetTerminalSettingBackgroundColor(
          &profile, GURL("chrome-untrusted://terminal/html/terminal.html"),
          std::nullopt),
      "#101010");

  // Use default color.
  profile.GetPrefs()->SetDict(guest_os::prefs::kGuestOsTerminalSettings,
                              base::Value::Dict());
  EXPECT_EQ(
      GetTerminalSettingBackgroundColor(
          &profile, GURL("chrome-untrusted://terminal/html/terminal.html"),
          std::nullopt),
      "#202124");
}

}  // namespace guest_os