chrome/browser/component_updater/translate_kit_component_installer_unittest.cc

// Copyright 2024 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/component_updater/translate_kit_component_installer.h"

#include <memory>
#include <utility>

#include "base/files/file_path.h"
#include "base/files/scoped_temp_dir.h"
#include "base/functional/callback.h"
#include "base/notreached.h"
#include "base/run_loop.h"
#include "base/test/bind.h"
#include "base/test/task_environment.h"
#include "base/time/time.h"
#include "base/version.h"
#include "chrome/browser/on_device_translation/pref_names.h"
#include "components/component_updater/mock_component_updater_service.h"
#include "components/crx_file/id_util.h"
#include "components/prefs/pref_service.h"
#include "components/sync_preferences/testing_pref_service_syncable.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace component_updater {
namespace {

using ::testing::_;

// The SHA256 of the SubjectPublicKeyInfo used to sign the component.
// The component id is: lbimbicckdokpoicboneldipejkhjgdg
constexpr uint8_t kTranslateKitPublicKeySHA256[32] = {
    0xb1, 0x8c, 0x18, 0x22, 0xa3, 0xea, 0xfe, 0x82, 0x1e, 0xd4, 0xb3,
    0x8f, 0x49, 0xa7, 0x96, 0x36, 0x55, 0xf3, 0xbc, 0x0d, 0xa5, 0x67,
    0x48, 0x09, 0xcd, 0x7b, 0xa9, 0x5f, 0xd8, 0x7f, 0x53, 0xb4};

constexpr char kFakeTranslateKitVersion[] = "0.0.1";

}  // namespace

class TranslateKitComponentMockComponentUpdateService
    : public component_updater::MockComponentUpdateService {
 public:
  TranslateKitComponentMockComponentUpdateService() = default;

  // Not Copyable.
  TranslateKitComponentMockComponentUpdateService(
      const TranslateKitComponentMockComponentUpdateService&) = delete;
  TranslateKitComponentMockComponentUpdateService& operator=(
      const TranslateKitComponentMockComponentUpdateService&) = delete;
};

class RegisterTranslateKitComponentTest : public ::testing::Test {
 public:
  RegisterTranslateKitComponentTest() = default;

  void SetUp() override {
    ASSERT_TRUE(fake_install_dir_.CreateUniqueTempDir());
    SetVersion(kFakeTranslateKitVersion);
    on_device_translation::RegisterLocalStatePrefs(pref_service_.registry());
  }

  // Not Copyable.
  RegisterTranslateKitComponentTest(const RegisterTranslateKitComponentTest&) =
      delete;
  RegisterTranslateKitComponentTest& operator=(
      const RegisterTranslateKitComponentTest&) = delete;

 protected:
  content::BrowserTaskEnvironment& env() { return env_; }
  PrefService* pref_service() { return &pref_service_; }
  const base::FilePath& install_dir() const {
    return fake_install_dir_.GetPath();
  }
  const base::Version& version() const { return fake_version_; }
  const base::Value::Dict& manifest() const { return fake_manifest_; }

  void SetVersion(std::string_view version_str) {
    fake_version_ = base::Version(version_str);
    fake_manifest_.Set("version", version_str);
  }

 private:
  content::BrowserTaskEnvironment env_;
  sync_preferences::TestingPrefServiceSyncable pref_service_;
  base::ScopedTempDir fake_install_dir_;
  base::Version fake_version_;
  base::Value::Dict fake_manifest_;
};

TEST_F(RegisterTranslateKitComponentTest, ComponentRegistrationNoForceInstall) {
  auto service =
      std::make_unique<TranslateKitComponentMockComponentUpdateService>();

  RegisterTranslateKitComponent(
      service.get(), pref_service(),
      /*force_install=*/false,
      /*registered_callback=*/base::BindOnce([]() { NOTREACHED(); }));
  env().RunUntilIdle();
  EXPECT_FALSE(
      pref_service()->GetBoolean(prefs::kTranslateKitPreviouslyRegistered));
}

TEST_F(RegisterTranslateKitComponentTest,
       ComponentRegistrationNoForceInstallPreviouslyRegistered) {
  auto service =
      std::make_unique<TranslateKitComponentMockComponentUpdateService>();

  pref_service()->SetBoolean(prefs::kTranslateKitPreviouslyRegistered, true);
  EXPECT_CALL(*service, RegisterComponent(_));
  EXPECT_CALL(*service, GetComponentIDs());
  base::RunLoop run_loop;
  RegisterTranslateKitComponent(service.get(), pref_service(),
                                /*force_install=*/false,
                                /*registered_callback=*/run_loop.QuitClosure());
  run_loop.Run();
  EXPECT_TRUE(
      pref_service()->GetBoolean(prefs::kTranslateKitPreviouslyRegistered));
}

TEST_F(RegisterTranslateKitComponentTest, ComponentRegistrationForceInstall) {
  auto service =
      std::make_unique<TranslateKitComponentMockComponentUpdateService>();

  EXPECT_CALL(*service, RegisterComponent(_));
  EXPECT_CALL(*service, GetComponentIDs());
  base::RunLoop run_loop;
  RegisterTranslateKitComponent(service.get(), pref_service(),
                                /*force_install=*/true,
                                /*registered_callback=*/run_loop.QuitClosure());
  run_loop.Run();
  EXPECT_TRUE(
      pref_service()->GetBoolean(prefs::kTranslateKitPreviouslyRegistered));
}

TEST_F(RegisterTranslateKitComponentTest,
       ComponentRegistrationForceInstallAlreadyRegistered) {
  auto service =
      std::make_unique<TranslateKitComponentMockComponentUpdateService>();

  EXPECT_CALL(*service, GetComponentIDs())
      .WillOnce(testing::Return(
          std::vector<std::string>({crx_file::id_util::GenerateIdFromHash(
              kTranslateKitPublicKeySHA256)})));
  base::RunLoop run_loop;
  RegisterTranslateKitComponent(
      service.get(), pref_service(),
      /*force_install=*/true,
      /*registered_callback=*/base::BindOnce([]() { NOTREACHED(); }));
  env().RunUntilIdle();
  EXPECT_FALSE(
      pref_service()->GetBoolean(prefs::kTranslateKitPreviouslyRegistered));
}

TEST_F(RegisterTranslateKitComponentTest, VerifyInstallationDefaultEmpty) {
  TranslateKitComponentInstallerPolicy policy(pref_service());

  // An empty directory lacks all required files.
  EXPECT_FALSE(policy.VerifyInstallation(manifest(), install_dir()));
}

}  // namespace component_updater