chrome/browser/extensions/extension_service_user_test_base.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 "chrome/browser/extensions/extension_service_user_test_base.h"

#include <memory>
#include <utility>

#include "build/chromeos_buildflags.h"
#include "chrome/browser/extensions/extension_service_test_base.h"
#include "content/public/test/browser_task_environment.h"
#include "google_apis/gaia/gaia_id.h"

#if BUILDFLAG(IS_CHROMEOS)
#include "chrome/browser/ash/login/users/fake_chrome_user_manager.h"
#include "components/account_id/account_id.h"
#include "components/user_manager/scoped_user_manager.h"
#include "components/user_manager/user_names.h"
#endif  // BUILDFLAG(IS_CHROMEOS)

namespace extensions {

ExtensionServiceUserTestBase::ExtensionServiceUserTestBase() = default;
ExtensionServiceUserTestBase::~ExtensionServiceUserTestBase() = default;
ExtensionServiceUserTestBase::ExtensionServiceUserTestBase(
    std::unique_ptr<content::BrowserTaskEnvironment> task_environment)
    : ExtensionServiceTestBase(std::move(task_environment)) {}

#if BUILDFLAG(IS_CHROMEOS)
void ExtensionServiceUserTestBase::SetUp() {
  ExtensionServiceTestBase::SetUp();
  scoped_user_manager_ = std::make_unique<user_manager::ScopedUserManager>(
      std::make_unique<ash::FakeChromeUserManager>());
  account_id_ = AccountId::FromUserEmailGaiaId("test-user@testdomain.com",
                                               GaiaId("1234567890"));
}

void ExtensionServiceUserTestBase::TearDown() {
  ExtensionServiceTestBase::TearDown();
  scoped_user_manager_.reset();
}

void ExtensionServiceUserTestBase::LoginChromeOSUser(
    const user_manager::User* user,
    const AccountId& account_id) {
  ASSERT_TRUE(user);
  GetFakeUserManager()->LoginUser(account_id,
                                  /*set_profile_created_flag=*/false);
  ASSERT_TRUE(GetFakeUserManager()->IsUserLoggedIn());
  ASSERT_TRUE(user == GetFakeUserManager()->GetActiveUser());
}
#endif  // BUILDFLAG(IS_CHROMEOS)

void ExtensionServiceUserTestBase::MaybeSetUpTestUser(bool is_guest) {
  SetGuestSessionOnProfile(is_guest);

  ASSERT_EQ(is_guest, profile()->IsGuestSession());

#if BUILDFLAG(IS_CHROMEOS)
  user_manager::User* user;
  AccountId account_id = account_id_;
  if (is_guest) {
    user = GetFakeUserManager()->AddGuestUser();
    account_id = user_manager::GuestAccountId();
  } else {
    user = GetFakeUserManager()->AddUser(account_id_);
  }
  ASSERT_NO_FATAL_FAILURE(LoginChromeOSUser(user, account_id));
#endif  // BUILDFLAG(IS_CHROMEOS)
}

}  // namespace extensions