// Copyright 2019 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/test/payments/personal_data_manager_test_util.h"

#include "base/run_loop.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/autofill/personal_data_manager_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "components/autofill/core/browser/data_manager/addresses/address_data_manager.h"
#include "components/autofill/core/browser/data_manager/payments/payments_data_manager.h"
#include "components/autofill/core/browser/data_manager/payments/payments_data_manager_test_api.h"
#include "components/autofill/core/browser/data_manager/personal_data_manager.h"
#include "components/autofill/core/browser/data_manager/personal_data_manager_observer.h"
#include "components/autofill/core/browser/data_manager/personal_data_manager_test_utils.h"
#include "components/autofill/core/browser/data_model/addresses/autofill_profile.h"
#include "components/autofill/core/browser/data_model/payments/credit_card.h"
#include "components/autofill/core/browser/test_utils/autofill_test_utils.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace payments::test {

void AddAutofillProfile(content::BrowserContext* browser_context,
                        const autofill::AutofillProfile& autofill_profile) {
  autofill::PersonalDataManager* personal_data_manager =
      autofill::PersonalDataManagerFactory::GetForBrowserContext(
          browser_context);
  size_t profile_count =
      personal_data_manager->address_data_manager().GetProfiles().size();
  autofill::PersonalDataChangedWaiter waiter(*personal_data_manager);
  personal_data_manager->address_data_manager().AddProfile(autofill_profile);
  std::move(waiter).Wait();
  EXPECT_EQ(profile_count + 1,
            personal_data_manager->address_data_manager().GetProfiles().size());
}

void AddCreditCard(content::BrowserContext* browser_context,
                   const autofill::CreditCard& card) {
  autofill::PersonalDataManager* personal_data_manager =
      autofill::PersonalDataManagerFactory::GetForBrowserContext(
          browser_context);
  size_t card_count =
      personal_data_manager->payments_data_manager().GetCreditCards().size();
  autofill::PersonalDataChangedWaiter waiter(*personal_data_manager);
  if (card.record_type() == autofill::CreditCard::RecordType::kLocalCard) {
    personal_data_manager->payments_data_manager().AddCreditCard(card);
  } else {
    test_api(personal_data_manager->payments_data_manager())
        .AddServerCreditCard(card);
  }
  std::move(waiter).Wait();
  EXPECT_EQ(
      card_count + 1,
      personal_data_manager->payments_data_manager().GetCreditCards().size());
}

}  // namespace payments::test