// 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 "components/signin/public/identity_manager/accounts_in_cookie_jar_info.h"

#include "google_apis/gaia/gaia_auth_util.h"
#include "google_apis/gaia/gaia_id.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

using ::testing::ElementsAre;

class AccountsInCookieJarInfoTest : public testing::Test {
 public:
  AccountsInCookieJarInfoTest() {
    valid_account1_ = CreateListedAccount("a@example.com", GaiaId("gaia_a"));
    valid_account1_ = CreateListedAccount("b@example.com", GaiaId("gaia_b"));

    invalid_account1_ = CreateListedAccount("c@example.com", GaiaId("gaia_c"));
    invalid_account1_.valid = false;
    invalid_account2_ = CreateListedAccount("d@example.com", GaiaId("gaia_d"));
    invalid_account2_.valid = false;

    signed_out_account1_ =
        CreateListedAccount("e@example.com", GaiaId("gaia_e"));
    signed_out_account1_.signed_out = true;
    signed_out_account2_ =
        CreateListedAccount("f@example.com", GaiaId("gaia_f"));
    signed_out_account2_.signed_out = true;
  }

 protected:
  gaia::ListedAccount valid_account1_;
  gaia::ListedAccount valid_account2_;
  gaia::ListedAccount invalid_account1_;
  gaia::ListedAccount invalid_account2_;
  gaia::ListedAccount signed_out_account1_;
  gaia::ListedAccount signed_out_account2_;

 private:
  static gaia::ListedAccount CreateListedAccount(const std::string& email,
                                                 const GaiaId& gaia_id) {
    gaia::ListedAccount result;
    result.id = CoreAccountId::FromGaiaId(gaia_id);
    result.email = email;
    result.gaia_id = gaia_id;
    result.raw_email = email;
    return result;
  }
};

TEST_F(AccountsInCookieJarInfoTest, AccountsFiltered) {
  signin::AccountsInCookieJarInfo cookies(
      false, {signed_out_account1_, valid_account1_, invalid_account1_,
              valid_account2_, signed_out_account2_, invalid_account2_});
  EXPECT_THAT(
      cookies.GetAllAccounts(),
      ElementsAre(signed_out_account1_, valid_account1_, invalid_account1_,
                  valid_account2_, signed_out_account2_, invalid_account2_));
  EXPECT_THAT(cookies.GetValidSignedInAccounts(),
              ElementsAre(valid_account1_, valid_account2_));
  EXPECT_THAT(cookies.GetPotentiallyInvalidSignedInAccounts(),
              ElementsAre(valid_account1_, invalid_account1_, valid_account2_,
                          invalid_account2_));
  EXPECT_THAT(cookies.GetSignedOutAccounts(),
              ElementsAre(signed_out_account1_, signed_out_account2_));
}