// 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/ui/profiles/profile_customization_util.h"

#include <string>

#include "base/run_loop.h"
#include "base/strings/strcat.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "components/signin/public/identity_manager/identity_test_environment.h"
#include "google_apis/gaia/gaia_id.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

AccountInfo FillAccountInfo(const CoreAccountInfo& core_info,
                            const std::string& given_name) {
  return AccountInfo::Builder(core_info)
      .SetFullName(base::StrCat({given_name, " Doe"}))
      .SetGivenName(given_name)
      .SetHostedDomain(std::string())
      .SetAvatarUrl(base::StrCat({"https://picture.url/", given_name}))
      .SetLocale("en")
      .Build();
}

}  // namespace

class ProfileNameResolverTest : public testing::Test {
 protected:
  const std::string kTestGivenName = "Jane";
  const std::string kTestEmail = "jane@bar.baz";
  const GaiaId kTestGaiaId = GaiaId("123456");

  signin::IdentityTestEnvironment* identity_test_env() {
    return &identity_test_env_;
  }

 private:
  base::test::SingleThreadTaskEnvironment task_environment_;
  signin::IdentityTestEnvironment identity_test_env_;
};

TEST_F(ProfileNameResolverTest, RunWithProfileName) {
  CoreAccountInfo core_account_info =
      identity_test_env()->MakeAccountAvailable(kTestEmail);
  base::test::TestFuture<std::u16string> profile_name_future;
  base::test::TestFuture<std::u16string> profile_name_future_2;

  ProfileNameResolver resolver{identity_test_env()->identity_manager(),
                               core_account_info};

  // `RunWithProfileName` should not run the callback as no profile name is
  // available.
  resolver.RunWithProfileName(profile_name_future.GetCallback());
  base::RunLoop().RunUntilIdle();
  EXPECT_FALSE(profile_name_future.IsReady());

  // Simulate the account info being updated, should result in the callback
  // getting called.
  resolver.OnExtendedAccountInfoUpdated(
      FillAccountInfo(core_account_info, kTestGivenName));
  EXPECT_EQ(base::ASCIIToUTF16(kTestGivenName), profile_name_future.Get());

  // Calling `RunWithProfileName` again should make it get called right away,
  // synchronously.
  resolver.RunWithProfileName(profile_name_future_2.GetCallback());
  EXPECT_TRUE(profile_name_future_2.IsReady());
  EXPECT_EQ(base::ASCIIToUTF16(kTestGivenName), profile_name_future_2.Get());
}

TEST_F(ProfileNameResolverTest, RunWithProfileName_InfoAvailable) {
  CoreAccountInfo core_account_info =
      identity_test_env()->MakeAccountAvailable(kTestEmail);
  identity_test_env()->UpdateAccountInfoForAccount(
      FillAccountInfo(core_account_info, kTestGivenName));
  base::test::TestFuture<std::u16string> profile_name_future;

  ProfileNameResolver resolver{identity_test_env()->identity_manager(),
                               core_account_info};

  // The information is available, the callback should run right away.
  resolver.RunWithProfileName(profile_name_future.GetCallback());
  EXPECT_TRUE(profile_name_future.IsReady());
  EXPECT_EQ(base::ASCIIToUTF16(kTestGivenName), profile_name_future.Get());
}

TEST_F(ProfileNameResolverTest, RunWithProfileName_InfoUnvailable) {
  auto scoped_timeout_override =
      ProfileNameResolver::CreateScopedInfoFetchTimeoutOverrideForTesting(
          base::TimeDelta());

  base::test::TestFuture<std::u16string> profile_name_future;

  // The account is not known from the identity manager.
  CoreAccountInfo core_account_info;
  core_account_info.gaia = kTestGaiaId;
  core_account_info.account_id =
      CoreAccountId::FromGaiaId(core_account_info.gaia);
  core_account_info.email = kTestEmail;
  ProfileNameResolver resolver{identity_test_env()->identity_manager(),
                               core_account_info};
  resolver.RunWithProfileName(profile_name_future.GetCallback());

  // Simulate the account info being updated.
  resolver.OnExtendedAccountInfoUpdated(
      FillAccountInfo(core_account_info, kTestGivenName));
  EXPECT_TRUE(profile_name_future.IsReady());

  EXPECT_EQ(base::ASCIIToUTF16(kTestGivenName), profile_name_future.Get());
}

// Regression test for https://crbug.com/40072102
TEST_F(ProfileNameResolverTest, RunWithProfileName_InfoMissing) {
  auto scoped_timeout_override =
      ProfileNameResolver::CreateScopedInfoFetchTimeoutOverrideForTesting(
          base::TimeDelta());

  base::test::TestFuture<std::u16string> profile_name_future;

  // The account is not known from the identity manager.
  CoreAccountInfo core_account_info;
  core_account_info.gaia = kTestGaiaId;
  core_account_info.account_id =
      CoreAccountId::FromGaiaId(core_account_info.gaia);
  core_account_info.email = kTestEmail;
  ProfileNameResolver resolver{identity_test_env()->identity_manager(),
                               core_account_info};
  resolver.RunWithProfileName(profile_name_future.GetCallback());

  // Simulate the account info being updated with invalid info. Should
  // result in the callback not getting called, and waiting until the
  // timeout to get a fallback value as name.
  resolver.OnExtendedAccountInfoUpdated(AccountInfo());
  EXPECT_FALSE(profile_name_future.IsReady());

  EXPECT_EQ(base::ASCIIToUTF16(kTestEmail), profile_name_future.Get());
}