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 <memory>
#include <utility>

#include "build/chromeos_buildflags.h"
#include "chrome/browser/extensions/extension_service_test_base.h"
#include "chrome/browser/extensions/extension_service_user_test_base.h"
#include "chrome/test/base/testing_profile.h"
#include "content/public/test/browser_task_environment.h"

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_ASH)
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", "1234567890");
}

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

void ExtensionServiceUserTestBase::LoginChromeOSAshUser(
    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_ASH)

void ExtensionServiceUserTestBase::MaybeSetUpTestUser(bool is_guest) {
  testing_profile()->SetGuestSession(is_guest);

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

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

}  // namespace extensions