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

#include "components/signin/public/identity_manager/identity_test_environment.h"

#include "base/functional/bind.h"
#include "base/test/task_environment.h"
#include "components/signin/public/base/consent_level.h"
#include "components/signin/public/base/oauth_consumer_id.h"
#include "components/signin/public/identity_manager/access_token_info.h"
#include "components/signin/public/identity_manager/accounts_in_cookie_jar_info.h"
#include "components/signin/public/identity_manager/test_identity_manager_observer.h"
#include "google_apis/gaia/gaia_constants.h"
#include "services/network/test/test_url_loader_factory.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace signin {

class IdentityTestEnvironmentTest : public testing::Test {
 public:
  IdentityTestEnvironmentTest()
      : task_environment_(
            base::test::TaskEnvironment::MainThreadType::DEFAULT,
            base::test::TaskEnvironment::ThreadPoolExecutionMode::QUEUED) {}

  IdentityTestEnvironmentTest(const IdentityTestEnvironmentTest&) = delete;
  IdentityTestEnvironmentTest& operator=(const IdentityTestEnvironmentTest&) =
      delete;

  ~IdentityTestEnvironmentTest() override { task_environment_.RunUntilIdle(); }

 private:
  base::test::TaskEnvironment task_environment_;
};

TEST_F(IdentityTestEnvironmentTest,
       IdentityTestEnvironmentCancelsPendingRequestsOnDestruction) {
  std::unique_ptr<IdentityTestEnvironment> identity_test_environment =
      std::make_unique<IdentityTestEnvironment>();

  identity_test_environment->MakePrimaryAccountAvailable(
      "primary@example.com", signin::ConsentLevel::kSignin);
  AccessTokenFetcher::TokenCallback callback = base::BindOnce(
      [](GoogleServiceAuthError error, AccessTokenInfo access_token_info) {});

  std::unique_ptr<AccessTokenFetcher> fetcher =
      identity_test_environment->identity_manager()
          ->CreateAccessTokenFetcherForAccount(
              identity_test_environment->identity_manager()
                  ->GetPrimaryAccountId(ConsentLevel::kSignin),
              OAuthConsumerId::kSync, std::move(callback),
              AccessTokenFetcher::Mode::kImmediate);

  // Deleting the IdentityTestEnvironment should cancel any pending
  // task in order to avoid use-after-free crashes. The destructor of
  // the test will spin the runloop which would run
  // IdentityTestEnvironment pending tasks if not canceled.
  identity_test_environment.reset();
  fetcher.reset();
}

// TODO(https://crbug.com/40066949): Delete this test once `ConsentLevel::kSync`
// is deleted.
TEST_F(IdentityTestEnvironmentTest,
       IdentityTestEnvironmentSetPrimaryAccountWithSyncConsent) {
  std::unique_ptr<IdentityTestEnvironment> identity_test_environment =
      std::make_unique<IdentityTestEnvironment>();
  IdentityManager* identity_manager =
      identity_test_environment->identity_manager();
  std::string primary_account_email = "primary@example.com";

  EXPECT_FALSE(identity_manager->HasPrimaryAccount(ConsentLevel::kSync));
  AccountInfo account_info =
      identity_test_environment->MakeAccountAvailable(primary_account_email);
  identity_test_environment->SetPrimaryAccount(primary_account_email,
                                               ConsentLevel::kSync);
  EXPECT_TRUE(identity_manager->HasPrimaryAccount(ConsentLevel::kSync));
  EXPECT_TRUE(identity_manager->HasPrimaryAccount(ConsentLevel::kSignin));
}

TEST_F(IdentityTestEnvironmentTest,
       IdentityTestEnvironmentSetPrimaryAccountWithoutSyncConsent) {
  std::unique_ptr<IdentityTestEnvironment> identity_test_environment =
      std::make_unique<IdentityTestEnvironment>();
  IdentityManager* identity_manager =
      identity_test_environment->identity_manager();
  std::string primary_account_email = "primary@example.com";

  EXPECT_FALSE(identity_manager->HasPrimaryAccount(ConsentLevel::kSignin));
  AccountInfo account_info =
      identity_test_environment->MakeAccountAvailable(primary_account_email);
  identity_test_environment->SetPrimaryAccount(primary_account_email,
                                               ConsentLevel::kSignin);
  EXPECT_TRUE(identity_manager->HasPrimaryAccount(ConsentLevel::kSignin));

  // TODO(https://crbug.com/40066949): Remove once `ConsentLevel::kSync` is
  // deleted.
  EXPECT_FALSE(identity_manager->HasPrimaryAccount(ConsentLevel::kSync));
}

// TODO(https://crbug.com/40066949): Delete this test once `ConsentLevel::kSync`
// is deleted.
TEST_F(IdentityTestEnvironmentTest,
       IdentityTestEnvironmentMakePrimaryAccountAvailableWithSyncConsent) {
  std::unique_ptr<IdentityTestEnvironment> identity_test_environment =
      std::make_unique<IdentityTestEnvironment>();
  IdentityManager* identity_manager =
      identity_test_environment->identity_manager();
  std::string primary_account_email = "primary@example.com";

  EXPECT_FALSE(identity_manager->HasPrimaryAccount(ConsentLevel::kSync));
  identity_test_environment->MakePrimaryAccountAvailable(primary_account_email,
                                                         ConsentLevel::kSync);
  EXPECT_TRUE(identity_manager->HasPrimaryAccount(ConsentLevel::kSync));
  EXPECT_TRUE(identity_manager->HasPrimaryAccount(ConsentLevel::kSignin));
}

TEST_F(IdentityTestEnvironmentTest,
       IdentityTestEnvironmentMakePrimaryAccountAvailableWithoutSyncConsent) {
  std::unique_ptr<IdentityTestEnvironment> identity_test_environment =
      std::make_unique<IdentityTestEnvironment>();
  IdentityManager* identity_manager =
      identity_test_environment->identity_manager();
  std::string primary_account_email = "primary@example.com";

  EXPECT_FALSE(identity_manager->HasPrimaryAccount(ConsentLevel::kSignin));
  identity_test_environment->MakePrimaryAccountAvailable(primary_account_email,
                                                         ConsentLevel::kSignin);
  EXPECT_TRUE(identity_manager->HasPrimaryAccount(ConsentLevel::kSignin));

  // TODO(https://crbug.com/40066949): Remove once `ConsentLevel::kSync` is
  // deleted.
  EXPECT_FALSE(identity_manager->HasPrimaryAccount(ConsentLevel::kSync));
}

TEST_F(IdentityTestEnvironmentTest, TriggerListAccount) {
  std::unique_ptr<IdentityTestEnvironment> identity_test_environment =
      std::make_unique<IdentityTestEnvironment>();

  const std::string kPrimaryEmail = "primary@gmail.com";
  identity_test_environment->MakeAccountAvailable(
      kPrimaryEmail, {.primary_account_consent_level = ConsentLevel::kSignin,
                      .set_cookie = true});
  ASSERT_EQ(identity_test_environment->identity_manager()
                ->GetAccountsInCookieJar()
                .GetPotentiallyInvalidSignedInAccounts()
                .size(),
            1u);

  {
    TestIdentityManagerObserver observer(
        identity_test_environment->identity_manager());

    // Changes should be reflected when triggering list account.
    identity_test_environment->TriggerListAccount();

    const AccountsInCookieJarInfo& observed_cookie_jar =
        observer.AccountsInfoFromAccountsInCookieUpdatedCallback();
    auto signed_in_accounts =
        observed_cookie_jar.GetPotentiallyInvalidSignedInAccounts();
    ASSERT_EQ(signed_in_accounts.size(), 1u);
    EXPECT_EQ(signed_in_accounts[0].email, kPrimaryEmail);
  }

  // Add a secondary account
  const std::string secondary_email = "secondary@gmail.com";
  identity_test_environment->MakeAccountAvailable(secondary_email,
                                                  {.set_cookie = true});

  {
    TestIdentityManagerObserver observer(
        identity_test_environment->identity_manager());

    // Changes should be reflected when triggering list account.
    identity_test_environment->TriggerListAccount();

    const AccountsInCookieJarInfo& observed_cookie_jar =
        observer.AccountsInfoFromAccountsInCookieUpdatedCallback();
    auto signed_in_accounts =
        observed_cookie_jar.GetPotentiallyInvalidSignedInAccounts();
    ASSERT_EQ(signed_in_accounts.size(), 2u);
    EXPECT_EQ(signed_in_accounts[0].email, kPrimaryEmail);
    EXPECT_EQ(signed_in_accounts[1].email, secondary_email);
  }
}

}  // namespace signin