// 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 "base/test/metrics/histogram_tester.h"
#include "base/test/mock_callback.h"
#include "chrome/browser/ui/autofill/payments/payments_view_factory.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/views/autofill/payments/payments_window_user_consent_dialog_view.h"
#include "chrome/test/interaction/interactive_browser_test.h"
#include "components/autofill/core/browser/metrics/payments/payments_window_metrics.h"
#include "components/autofill/core/browser/ui/payments/payments_window_user_consent_dialog_controller_impl.h"
#include "content/public/test/browser_test.h"
#include "ui/views/window/dialog_client_view.h"

namespace autofill::payments {

constexpr char kSuppressedScreenshotError[] =
    "Screenshot can only run in pixel_tests.";
constexpr std::string_view
    kPaymentsWindowUserConsentDialogResultVcn3dsHistogramName =
        "Autofill.Vcn3ds.PaymentsWindowUserConsentDialogResult";
constexpr std::string_view
    kPaymentsWindowUserConsentDialogShownVcn3dsHistogramName =
        "Autofill.Vcn3ds.PaymentsWindowUserConsentDialogShown";

class PaymentsWindowUserConsentDialogBrowserTest
    : public InteractiveBrowserTest {
 public:
  PaymentsWindowUserConsentDialogBrowserTest() {
    controller_ =
        std::make_unique<PaymentsWindowUserConsentDialogControllerImpl>(
            accept_callback_.Get(), cancel_callback_.Get());
  }
  PaymentsWindowUserConsentDialogBrowserTest(
      const PaymentsWindowUserConsentDialogBrowserTest&) = delete;
  PaymentsWindowUserConsentDialogBrowserTest& operator=(
      const PaymentsWindowUserConsentDialogBrowserTest&) = delete;
  ~PaymentsWindowUserConsentDialogBrowserTest() override = default;

 protected:
  InteractiveBrowserTestApi::MultiStep TriggerDialogAndWaitForShow(
      ElementSpecifier element_specifier) {
    return Steps(
        TriggerDialog(),
        // b/332603033: Tab-modal dialogs on MacOS run in a different context.
        InAnyContext(WaitForShow(element_specifier)));
  }

  base::MockOnceClosure accept_callback_;
  base::MockOnceClosure cancel_callback_;
  std::unique_ptr<PaymentsWindowUserConsentDialogControllerImpl> controller_;
  base::HistogramTester histogram_tester_;

 private:
  InteractiveTestApi::StepBuilder TriggerDialog() {
    return Do([this]() {
      controller_->ShowDialog(base::BindOnce(
          &CreateAndShowPaymentsWindowUserConsentDialog,
          controller_->GetWeakPtr(),
          // The callback is run instantly, so `base::Unretained()` is safe here
          // as `browser()->GetTabStripModel()->GetActiveWebContents()` will
          // always be present when the callback is run.
          base::Unretained(
              browser()->GetTabStripModel()->GetActiveWebContents())));
    });
  }
};

// Ensures the UI can be shown, and verifies that it looks as expected.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(
    PaymentsWindowUserConsentDialogBrowserTest,
    DISABLED_InvokeUi_PaymentsWindowUserConsentDialogDisplays) {
  const std::string payments_user_consent_dialog_root_view =
      "Payments User Consent Dialog Root View";
  RunTestSequence(
      TriggerDialogAndWaitForShow(
          PaymentsWindowUserConsentDialogView::kTopViewId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          NameViewRelative(PaymentsWindowUserConsentDialogView::kTopViewId,
                           payments_user_consent_dialog_root_view,
                           [](views::View* dialog_view) {
                             return dialog_view->GetWidget()->GetRootView();
                           }),
          SetOnIncompatibleAction(OnIncompatibleAction::kIgnoreAndContinue,
                                  kSuppressedScreenshotError),
          Screenshot(payments_user_consent_dialog_root_view,
                     /*screenshot_name=*/"consent_popup",
                     /*baseline_cl=*/"5338589")));
}

// Ensures the UI can be shown, and verifies that the dialog shown histogram
// bucket is logged to.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_DialogShownHistogramBucketLogs) {
  RunTestSequence(
      TriggerDialogAndWaitForShow(
          PaymentsWindowUserConsentDialogView::kTopViewId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(Check([this]() {
        return histogram_tester_.GetBucketCount(
                   kPaymentsWindowUserConsentDialogShownVcn3dsHistogramName,
                   /*sample=*/true) == 1;
      })));
}

// Ensures the UI can be shown, and verifies that accepting the dialog runs the
// accept callback and hides the view.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_DialogAcceptance) {
  EXPECT_CALL(accept_callback_, Run);

  RunTestSequence(
      TriggerDialogAndWaitForShow(views::DialogClientView::kOkButtonElementId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          PressButton(views::DialogClientView::kOkButtonElementId),
          WaitForHide(PaymentsWindowUserConsentDialogView::kTopViewId)));
}

// Ensures the UI can be shown, and verifies that accepting the dialog logs to
// the dialog acceptance histogram bucket.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_DialogAcceptanceHistogramBucketLogs) {
  EXPECT_CALL(accept_callback_, Run);

  RunTestSequence(
      TriggerDialogAndWaitForShow(views::DialogClientView::kOkButtonElementId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          PressButton(views::DialogClientView::kOkButtonElementId),
          WaitForHide(PaymentsWindowUserConsentDialogView::kTopViewId),
          Check([this]() {
            return histogram_tester_.GetBucketCount(
                       /*name=*/
                       kPaymentsWindowUserConsentDialogResultVcn3dsHistogramName, /*sample=*/
                       autofill_metrics::PaymentsWindowUserConsentDialogResult::
                           kAcceptButtonClicked) == 1;
          })));
}

// Ensures the UI can be shown, and verifies that cancelling the dialog runs the
// cancel callback and hides the view.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_DialogCancelled) {
  EXPECT_CALL(cancel_callback_, Run);

  RunTestSequence(
      TriggerDialogAndWaitForShow(
          views::DialogClientView::kCancelButtonElementId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          PressButton(views::DialogClientView::kCancelButtonElementId),
          WaitForHide(PaymentsWindowUserConsentDialogView::kTopViewId)));
}

// Ensures the UI can be shown, and verifies that cancelling the dialog logs to
// the dialog cancelled histogram bucket.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_DialogCancelledHistogramBucketLogs) {
  EXPECT_CALL(cancel_callback_, Run);

  RunTestSequence(
      TriggerDialogAndWaitForShow(
          views::DialogClientView::kCancelButtonElementId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          PressButton(views::DialogClientView::kCancelButtonElementId),
          WaitForHide(PaymentsWindowUserConsentDialogView::kTopViewId),
          Check([this]() {
            return histogram_tester_.GetBucketCount(
                       /*name=*/
                       kPaymentsWindowUserConsentDialogResultVcn3dsHistogramName, /*sample=*/
                       autofill_metrics::PaymentsWindowUserConsentDialogResult::
                           kCancelButtonClicked) == 1;
          })));
}

// Ensures the UI can be shown, and verifies that pressing the escape key on the
// dialog hides the view.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_EscKeyPressed) {
  RunTestSequence(
      TriggerDialogAndWaitForShow(
          PaymentsWindowUserConsentDialogView::kTopViewId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          WithView(PaymentsWindowUserConsentDialogView::kTopViewId,
                   [](views::View* dialog_view) {
                     return dialog_view->GetWidget()->CloseWithReason(
                         views::Widget::ClosedReason::kEscKeyPressed);
                   }),
          WaitForHide(PaymentsWindowUserConsentDialogView::kTopViewId)));
}

// Ensures the UI can be shown, and verifies that pressing the escape key on the
// dialog logs to the escape key pressed histogram bucket.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_EscKeyPressedHistogramBucketLogs) {
  RunTestSequence(
      TriggerDialogAndWaitForShow(
          PaymentsWindowUserConsentDialogView::kTopViewId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          WithView(PaymentsWindowUserConsentDialogView::kTopViewId,
                   [](views::View* dialog_view) {
                     return dialog_view->GetWidget()->CloseWithReason(
                         views::Widget::ClosedReason::kEscKeyPressed);
                   }),
          WaitForHide(PaymentsWindowUserConsentDialogView::kTopViewId),
          Check([this]() {
            return histogram_tester_.GetBucketCount(
                       /*name=*/
                       kPaymentsWindowUserConsentDialogResultVcn3dsHistogramName, /*sample=*/
                       autofill_metrics::PaymentsWindowUserConsentDialogResult::
                           kEscapeKeyPressed) == 1;
          })));
}

// Ensures the UI can be shown, and verifies that closing the tab while the
// dialog is present does not crash.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_CanCloseTabWhileDialogShowing) {
  RunTestSequence(
      TriggerDialogAndWaitForShow(
          PaymentsWindowUserConsentDialogView::kTopViewId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(Do([this]() {
        browser()->GetTabStripModel()->GetActiveWebContents()->Close();
      })));
}

// Ensures the UI can be shown, and verifies that closing the tab while the
// dialog is present logs to the tab or browser closed histogram bucket.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(
    PaymentsWindowUserConsentDialogBrowserTest,
    DISABLED_InvokeUi_CloseTabWhileDialogShowingHistogramBucketLogs) {
  RunTestSequence(
      TriggerDialogAndWaitForShow(
          PaymentsWindowUserConsentDialogView::kTopViewId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          Do([this]() {
            browser()->GetTabStripModel()->GetActiveWebContents()->Close();
          }),
          Check([this]() {
            return histogram_tester_.GetBucketCount(
                       /*name=*/
                       kPaymentsWindowUserConsentDialogResultVcn3dsHistogramName, /*sample=*/
                       autofill_metrics::PaymentsWindowUserConsentDialogResult::
                           kTabOrBrowserClosed) == 1;
          })));
}

// Ensures the UI can be shown, and verifies that closing the browser while the
// dialog is present does not crash.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(PaymentsWindowUserConsentDialogBrowserTest,
                       DISABLED_InvokeUi_CanCloseBrowserWhileDialogShowing) {
  RunTestSequence(
      TriggerDialogAndWaitForShow(
          PaymentsWindowUserConsentDialogView::kTopViewId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(Do([this]() { browser()->GetWindow()->Close(); })));
}

// Ensures the UI can be shown, and verifies that closing the browser while the
// dialog is present logs to the tab or browser closed histogram bucket.
// TODO(crbug.com/441251456): Fix flakiness.
IN_PROC_BROWSER_TEST_F(
    PaymentsWindowUserConsentDialogBrowserTest,
    DISABLED_InvokeUi_CloseBrowserWhileDialogShowingHistogramBucketLogs) {
  RunTestSequence(
      TriggerDialogAndWaitForShow(
          PaymentsWindowUserConsentDialogView::kTopViewId),
      // TriggerDialogAndWaitForShow() changes the context, so the same context
      // must be used.
      InSameContext(
          Do([this]() { browser()->GetWindow()->Close(); }), Check([this]() {
            return histogram_tester_.GetBucketCount(
                       /*name=*/
                       kPaymentsWindowUserConsentDialogResultVcn3dsHistogramName, /*sample=*/
                       autofill_metrics::PaymentsWindowUserConsentDialogResult::
                           kTabOrBrowserClosed) == 1;
          })));
}

}  // namespace autofill::payments