// Copyright 2021 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/autofill/autofill_snackbar_controller_impl.h"

#include <optional>

#include "base/memory/raw_ptr.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/mock_callback.h"
#include "chrome/browser/autofill/mock_manual_filling_view.h"
#include "chrome/browser/keyboard_accessory/android/manual_filling_controller_impl.h"
#include "chrome/browser/keyboard_accessory/test_utils/android/mock_address_accessory_controller.h"
#include "chrome/browser/keyboard_accessory/test_utils/android/mock_password_accessory_controller.h"
#include "chrome/browser/keyboard_accessory/test_utils/android/mock_payment_method_accessory_controller.h"
#include "chrome/browser/ui/autofill/autofill_snackbar_view.h"
#include "chrome/test/base/chrome_render_view_host_test_harness.h"
#include "components/autofill/core/browser/data_model/payments/credit_card.h"
#include "components/strings/grit/components_strings.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/l10n/l10n_util.h"

using testing::NiceMock;

namespace autofill {

class AutofillSnackbarControllerImplTest
    : public ChromeRenderViewHostTestHarness {
 public:
  AutofillSnackbarControllerImplTest() = default;

  void SetUp() override {
    ChromeRenderViewHostTestHarness::SetUp();
    ManualFillingControllerImpl::CreateForWebContentsForTesting(
        web_contents(), mock_pwd_controller_.AsWeakPtr(),
        mock_address_controller_.AsWeakPtr(),
        mock_payment_method_controller_.AsWeakPtr(),
        std::make_unique<NiceMock<MockManualFillingView>>());
  }

  AutofillSnackbarControllerImpl* controller() {
    if (!controller_) {
      controller_ = new AutofillSnackbarControllerImpl(web_contents());
    }
    return controller_;
  }

 private:
  raw_ptr<AutofillSnackbarControllerImpl> controller_ = nullptr;
  NiceMock<MockPasswordAccessoryController> mock_pwd_controller_;
  NiceMock<MockAddressAccessoryController> mock_address_controller_;
  NiceMock<MockPaymentMethodAccessoryController>
      mock_payment_method_controller_;

  base::WeakPtrFactory<AutofillSnackbarControllerImplTest> weak_ptr_factory_{
      this};
};

TEST_F(AutofillSnackbarControllerImplTest, Metrics_VirtualCard) {
  base::HistogramTester histogram_tester;
  controller()->Show(AutofillSnackbarType::kVirtualCard, base::DoNothing());
  // Verify that the count for Shown is incremented and ActionClicked hasn't
  // changed.
  histogram_tester.ExpectUniqueSample("Autofill.Snackbar.VirtualCard.Shown", 1,
                                      1);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.VirtualCard.ActionClicked", 1, 0);
  controller()->OnDismissed();

  base::MockCallback<base::OnceClosure> on_action_clicked_callback;
  controller()->Show(AutofillSnackbarType::kVirtualCard,
                     on_action_clicked_callback.Get());
  EXPECT_CALL(on_action_clicked_callback, Run);
  controller()->OnActionClicked();
  // Verify that the count for both Shown and ActionClicked is incremented.
  histogram_tester.ExpectUniqueSample("Autofill.Snackbar.VirtualCard.Shown", 1,
                                      2);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.VirtualCard.ActionClicked", 1, 1);
}

TEST_F(AutofillSnackbarControllerImplTest,
       Metrics_ShowVirtualCardWhenAlreadyShowing) {
  base::HistogramTester histogram_tester;
  controller()->Show(AutofillSnackbarType::kVirtualCard, base::DoNothing());
  // Verify that the count for Shown is incremented and ActionClicked hasn't
  // changed.
  histogram_tester.ExpectUniqueSample("Autofill.Snackbar.VirtualCard.Shown", 1,
                                      1);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.VirtualCard.ActionClicked", 1, 0);

  // Attempt to show another dialog without dismissing the previous one.
  controller()->Show(AutofillSnackbarType::kVirtualCard, base::DoNothing());

  // Verify that the count for both Shown is not incremented.
  histogram_tester.ExpectUniqueSample("Autofill.Snackbar.VirtualCard.Shown", 1,
                                      1);
}

TEST_F(AutofillSnackbarControllerImplTest, Metrics_ShowMandatoryReauth) {
  base::HistogramTester histogram_tester;
  controller()->Show(AutofillSnackbarType::kMandatoryReauth, base::DoNothing());
  // Verify that the count for Shown is incremented and ActionClicked hasn't
  // changed.
  histogram_tester.ExpectUniqueSample("Autofill.Snackbar.MandatoryReauth.Shown",
                                      1, 1);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.MandatoryReauth.ActionClicked", 1, 0);
  controller()->OnDismissed();

  // TODO(crbug.com/40570965): Figure out how to mock
  // ShowAutofillCreditCardSettings to test ActionClicked metric.
}

TEST_F(AutofillSnackbarControllerImplTest, Metrics_SaveCardSuccess) {
  base::HistogramTester histogram_tester;

  controller()->Show(AutofillSnackbarType::kSaveCardSuccess, base::DoNothing());

  histogram_tester.ExpectUniqueSample("Autofill.Snackbar.SaveCardSuccess.Shown",
                                      true, 1);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.SaveCardSuccess.ActionClicked", true, 0);

  controller()->OnActionClicked();

  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.SaveCardSuccess.ActionClicked", true, 1);
}

TEST_F(AutofillSnackbarControllerImplTest, Metrics_VirtualCardEnrollSuccess) {
  base::HistogramTester histogram_tester;

  controller()->Show(AutofillSnackbarType::kVirtualCardEnrollSuccess,
                     base::DoNothing());

  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.VirtualCardEnrollSuccess.Shown", true, 1);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.VirtualCardEnrollSuccess.ActionClicked", true, 0);

  controller()->OnActionClicked();

  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.VirtualCardEnrollSuccess.ActionClicked", true, 1);
}

TEST_F(AutofillSnackbarControllerImplTest, Metrics_UndoPlusAddressEmailSwap) {
  base::HistogramTester histogram_tester;

  controller()->Show(AutofillSnackbarType::kPlusAddressEmailOverride,
                     base::DoNothing());

  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.PlusAddressEmailOverride.Shown", true, 1);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.PlusAddressEmailOverride.ActionClicked", true, 0);

  controller()->OnActionClicked();

  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.PlusAddressEmailOverride.ActionClicked", true, 1);
}

TEST_F(AutofillSnackbarControllerImplTest, Metrics_CardInfoRetrieval) {
  base::HistogramTester histogram_tester;

  CreditCard card;
  card.set_is_bnpl_card(false);
  card.set_record_type(CreditCard::RecordType::kMaskedServerCard);

  controller()->ShowPaymentsSnackbar(AutofillSnackbarType::kCardInfoRetrieval,
                                     card, base::DoNothing());

  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.CardInfoRetrievalEnrolled.Shown", true, 1);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.CardInfoRetrievalEnrolled.ActionClicked", true, 0);

  controller()->OnActionClicked();

  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.CardInfoRetrievalEnrolled.ActionClicked", true, 1);
}

TEST_F(AutofillSnackbarControllerImplTest, Metrics_Bnpl) {
  base::HistogramTester histogram_tester;

  CreditCard card;
  card.set_is_bnpl_card(true);
  card.SetNickname(u"Affirm");

  controller()->ShowPaymentsSnackbar(AutofillSnackbarType::kBnpl, card,
                                     base::DoNothing());

  histogram_tester.ExpectUniqueSample("Autofill.Snackbar.BnplVirtualCard.Shown",
                                      true, 1);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.BnplVirtualCard.ActionClicked", true, 0);

  controller()->OnActionClicked();

  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.BnplVirtualCard.ActionClicked", true, 1);
}

TEST_F(AutofillSnackbarControllerImplTest,
       SaveCardSuccessMessageAndActionButtonText) {
  controller()->Show(AutofillSnackbarType::kSaveCardSuccess, base::DoNothing());

  EXPECT_EQ(controller()->GetMessageText(),
            l10n_util::GetStringUTF16(
                IDS_AUTOFILL_SAVE_CARD_CONFIRMATION_SUCCESS_DESCRIPTION_TEXT));
  EXPECT_EQ(
      controller()->GetActionButtonText(),
      l10n_util::GetStringUTF16(
          IDS_AUTOFILL_SAVE_CARD_AND_VIRTUAL_CARD_ENROLL_CONFIRMATION_BUTTON_TEXT));
}

TEST_F(AutofillSnackbarControllerImplTest,
       VirtualCardEnrollSuccessMessageAndActionButtonText) {
  controller()->Show(AutofillSnackbarType::kVirtualCardEnrollSuccess,
                     base::DoNothing());

  EXPECT_EQ(
      controller()->GetMessageText(),
      l10n_util::GetStringUTF16(
          IDS_AUTOFILL_VIRTUAL_CARD_ENROLL_CONFIRMATION_SUCCESS_DESCRIPTION_TEXT));
  EXPECT_EQ(
      controller()->GetActionButtonText(),
      l10n_util::GetStringUTF16(
          IDS_AUTOFILL_SAVE_CARD_AND_VIRTUAL_CARD_ENROLL_CONFIRMATION_BUTTON_TEXT));
}

TEST_F(AutofillSnackbarControllerImplTest, Metrics_SaveServerIbanSuccess) {
  base::HistogramTester histogram_tester;
  controller()->Show(AutofillSnackbarType::kSaveServerIbanSuccess,
                     base::DoNothing());
  // Verify that the count for Shown is incremented and ActionClicked hasn't
  // changed.
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.SaveServerIbanSuccess.Shown", 1, 1);
  histogram_tester.ExpectTotalCount(
      "Autofill.Snackbar.SaveServerIbanSuccess.ActionClicked", 0);
  controller()->OnDismissed();

  controller()->Show(AutofillSnackbarType::kSaveServerIbanSuccess,
                     base::DoNothing());
  controller()->OnActionClicked();

  // Verify that the count for both Shown and ActionClicked is incremented.
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.SaveServerIbanSuccess.Shown", 1, 2);
  histogram_tester.ExpectUniqueSample(
      "Autofill.Snackbar.SaveServerIbanSuccess.ActionClicked", true, 1);
}

TEST_F(AutofillSnackbarControllerImplTest,
       SaveServerIbanSuccessMessageAndActionButtonText) {
  controller()->Show(AutofillSnackbarType::kSaveServerIbanSuccess,
                     base::DoNothing());

  EXPECT_EQ(controller()->GetMessageText(),
            l10n_util::GetStringUTF16(
                IDS_AUTOFILL_SAVE_SERVER_IBAN_SUCCESS_SNACKBAR_MESSAGE_TEXT));
  EXPECT_EQ(controller()->GetActionButtonText(),
            l10n_util::GetStringUTF16(
                IDS_AUTOFILL_SAVE_SERVER_IBAN_SUCCESS_SNACKBAR_BUTTON_TEXT));
}

TEST_F(AutofillSnackbarControllerImplTest, OnShowDefaultDurationSet) {
  controller()->Show(AutofillSnackbarType::kSaveCardSuccess, base::DoNothing());
  EXPECT_EQ(controller()->GetDuration(),
            AutofillSnackbarControllerImpl::kDefaultSnackbarDuration);
}

TEST_F(AutofillSnackbarControllerImplTest,
       OnShowWithDurationCustomDurationSet) {
  base::TimeDelta duration = base::Seconds(3);
  controller()->ShowWithDurationAndCallback(
      AutofillSnackbarType::kSaveCardSuccess, duration, base::DoNothing(),
      std::nullopt);
  EXPECT_EQ(controller()->GetDuration(), duration);
}

TEST_F(AutofillSnackbarControllerImplTest, OnDismissCallbackCalled) {
  base::MockCallback<base::OnceClosure> on_dismiss_callback;

  controller()->ShowWithDurationAndCallback(
      AutofillSnackbarType::kSaveCardSuccess,
      AutofillSnackbarControllerImpl::kDefaultSnackbarDuration,
      base::DoNothing(), on_dismiss_callback.Get());

  EXPECT_CALL(on_dismiss_callback, Run);
  controller()->OnDismissed();
}

TEST_F(AutofillSnackbarControllerImplTest, OnDismissTwiceCallbackCalledOnce) {
  base::MockCallback<base::OnceClosure> on_dismiss_callback;

  controller()->ShowWithDurationAndCallback(
      AutofillSnackbarType::kSaveCardSuccess,
      AutofillSnackbarControllerImpl::kDefaultSnackbarDuration,
      base::DoNothing(), on_dismiss_callback.Get());

  EXPECT_CALL(on_dismiss_callback, Run);

  controller()->OnDismissed();
  controller()->Show(AutofillSnackbarType::kSaveCardSuccess, base::DoNothing());
  controller()->OnDismissed();
}

TEST_F(AutofillSnackbarControllerImplTest,
       ShowPaymentsSnackbar_BnplCard_SetsCorrectMessageAndAction) {
  CreditCard card;
  card.set_is_bnpl_card(true);
  card.SetNickname(u"Affirm");

  controller()->ShowPaymentsSnackbar(AutofillSnackbarType::kBnpl, card,
                                     base::DoNothing());

  EXPECT_EQ(
      controller()->GetMessageText(),
      l10n_util::GetStringFUTF16(
          IDS_AUTOFILL_BNPL_FILLED_CARD_SNACKBAR_MESSAGE_TEXT, u"Affirm"));
  EXPECT_EQ(controller()->GetActionButtonText(),
            l10n_util::GetStringUTF16(
                IDS_AUTOFILL_VIRTUAL_CARD_NUMBER_SNACKBAR_ACTION_TEXT));
}

TEST_F(AutofillSnackbarControllerImplTest,
       ShowPaymentsSnackbar_VirtualCard_SetsCorrectMessageAndAction) {
  CreditCard card;
  card.set_is_bnpl_card(false);
  card.set_record_type(CreditCard::RecordType::kVirtualCard);

  controller()->ShowPaymentsSnackbar(AutofillSnackbarType::kVirtualCard, card,
                                     base::DoNothing());

  EXPECT_EQ(controller()->GetMessageText(),
            l10n_util::GetStringUTF16(
                IDS_AUTOFILL_VIRTUAL_CARD_NUMBER_SNACKBAR_MESSAGE_TEXT));
  EXPECT_EQ(controller()->GetActionButtonText(),
            l10n_util::GetStringUTF16(
                IDS_AUTOFILL_VIRTUAL_CARD_NUMBER_SNACKBAR_ACTION_TEXT));
}

TEST_F(AutofillSnackbarControllerImplTest,
       ShowPaymentsSnackbar_CardInfoRetrieval_SetsCorrectMessageAndAction) {
  CreditCard card;
  card.set_is_bnpl_card(false);
  card.set_record_type(CreditCard::RecordType::kMaskedServerCard);

  controller()->ShowPaymentsSnackbar(AutofillSnackbarType::kCardInfoRetrieval,
                                     card, base::DoNothing());

  EXPECT_EQ(controller()->GetMessageText(),
            l10n_util::GetStringUTF16(
                IDS_AUTOFILL_CARD_INFO_RETRIEVAL_SNACKBAR_MESSAGE_TEXT));
  EXPECT_EQ(controller()->GetActionButtonText(),
            l10n_util::GetStringUTF16(
                IDS_AUTOFILL_VIRTUAL_CARD_NUMBER_SNACKBAR_ACTION_TEXT));
}

}  // namespace autofill