chromium/chrome/browser/touch_to_fill/autofill/android/internal/java/src/org/chromium/chrome/browser/touch_to_fill/payments/TouchToFillPaymentMethodRenderTest.java

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

package org.chromium.chrome.browser.touch_to_fill.payments;

import static org.chromium.base.ThreadUtils.runOnUiThreadBlocking;
import static org.chromium.base.test.util.ApplicationTestUtils.finishActivity;
import static org.chromium.chrome.browser.autofill.AutofillTestHelper.createCreditCard;
import static org.chromium.chrome.browser.autofill.AutofillTestHelper.createCreditCardSuggestion;
import static org.chromium.chrome.browser.autofill.AutofillTestHelper.createVirtualCreditCard;
import static org.chromium.chrome.browser.night_mode.ChromeNightModeTestUtils.tearDownNightModeAfterChromeActivityDestroyed;
import static org.chromium.ui.base.LocalizationUtils.setRtlForTesting;

import android.view.View;
import android.view.ViewGroup;

import androidx.test.filters.MediumTest;

import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;
import org.mockito.quality.Strictness;

import org.chromium.base.ContextUtils;
import org.chromium.base.test.params.ParameterAnnotations;
import org.chromium.base.test.params.ParameterSet;
import org.chromium.base.test.params.ParameterizedRunner;
import org.chromium.base.test.util.CommandLineFlags;
import org.chromium.base.test.util.Feature;
import org.chromium.base.test.util.RequiresRestart;
import org.chromium.chrome.browser.autofill.AutofillTestHelper;
import org.chromium.chrome.browser.autofill.PersonalDataManager.CreditCard;
import org.chromium.chrome.browser.autofill.PersonalDataManager.Iban;
import org.chromium.chrome.browser.flags.ChromeSwitches;
import org.chromium.chrome.browser.night_mode.ChromeNightModeTestUtils;
import org.chromium.chrome.browser.touch_to_fill.common.BottomSheetFocusHelper;
import org.chromium.chrome.test.ChromeJUnit4RunnerDelegate;
import org.chromium.chrome.test.ChromeTabbedActivityTestRule;
import org.chromium.chrome.test.util.ChromeRenderTestRule;
import org.chromium.components.autofill.AutofillSuggestion;
import org.chromium.components.browser_ui.bottomsheet.BottomSheetController;
import org.chromium.components.browser_ui.bottomsheet.BottomSheetTestSupport;
import org.chromium.ui.test.util.RenderTestRule.Component;

import java.io.IOException;
import java.util.Arrays;
import java.util.List;

/**
 * These tests render screenshots of touch to fill for credit cards/IBANs sheet and compare them to
 * a gold standard.
 */
@RunWith(ParameterizedRunner.class)
@ParameterAnnotations.UseRunnerDelegate(ChromeJUnit4RunnerDelegate.class)
// TODO(crbug.com/344662597): Failing when batched, batch this again.
@CommandLineFlags.Add({ChromeSwitches.DISABLE_FIRST_RUN_EXPERIENCE})
public class TouchToFillPaymentMethodRenderTest {
    @ParameterAnnotations.ClassParameter
    private static List<ParameterSet> sClassParams =
            Arrays.asList(
                    new ParameterSet().value(false, false).name("Default"),
                    new ParameterSet().value(false, true).name("RTL"),
                    new ParameterSet().value(true, false).name("NightMode"));

    public ChromeTabbedActivityTestRule mActivityTestRule = new ChromeTabbedActivityTestRule();

    @Rule
    public final MockitoRule mMockitoRule = MockitoJUnit.rule().strictness(Strictness.STRICT_STUBS);

    @Rule
    public final ChromeRenderTestRule mRenderTestRule =
            ChromeRenderTestRule.Builder.withPublicCorpus()
                    .setRevision(14)
                    .setBugComponent(Component.UI_BROWSER_AUTOFILL)
                    .build();

    @Mock private TouchToFillPaymentMethodComponent.Delegate mDelegateMock;
    @Mock private BottomSheetFocusHelper mBottomSheetFocusHelper;

    private static final CreditCard VISA =
            createCreditCard(
                    "Visa",
                    "4111111111111111",
                    "05",
                    "2100",
                    true,
                    "Visa",
                    "• • • • 1111",
                    R.drawable.visa_metadata_card,
                    "visa");
    private static final CreditCard MASTERCARD =
            createCreditCard(
                    "MasterCard",
                    "5555555555554444",
                    "08",
                    "2100",
                    true,
                    "Mastercard",
                    "• • • • 4444",
                    R.drawable.mc_metadata_card,
                    "mastercard");
    private static final CreditCard SERVER_MASTERCARD =
            createCreditCard(
                    "MasterCard",
                    "5454545454545454",
                    "11",
                    "2100",
                    false,
                    "MasterCard-GPay",
                    "• • • • 5454",
                    R.drawable.mc_metadata_card,
                    "mastercard");
    private static final CreditCard DISCOVER =
            createCreditCard(
                    "Discover",
                    "6011111111111117",
                    "09",
                    "2100",
                    true,
                    "Discover",
                    "• • • • 1117",
                    R.drawable.discover_metadata_card,
                    "discover");
    private static final CreditCard AMERICAN_EXPRESS =
            createCreditCard(
                    "American Express",
                    "378282246310005",
                    "10",
                    "2100",
                    true,
                    "American Express",
                    "• • • • 0005",
                    R.drawable.amex_metadata_card,
                    "american express");
    private static final CreditCard MASTERCARD_VIRTUAL_CARD =
            createVirtualCreditCard(
                    /* name= */ "MasterCard-GPay",
                    /* number= */ "5454545454545454",
                    /* month= */ "11",
                    /* year= */ "2100",
                    /* network= */ "Mastercard",
                    /* iconId= */ R.drawable.mc_metadata_card,
                    /* cardNameForAutofillDisplay= */ "MasterCard-GPay",
                    /* obfuscatedLastFourDigits= */ "• • • • 5454");
    private static final Iban LOCAL_IBAN =
            Iban.createLocal(
                    /* guid= */ "000000111111",
                    /* label= */ "CH56 •••• •••• •••• •800 9",
                    /* nickname= */ "My brother's IBAN",
                    /* value= */ "CH5604835012345678009");

    private static final Iban LOCAL_IBAN_NO_NICKNAME =
            Iban.createLocal(
                    /* guid= */ "000000222222",
                    /* label= */ "FR76 •••• •••• •••• •••• •••0 189",
                    /* nickname= */ "",
                    /* value= */ "FR7630006000011234567890189");
    private static final AutofillSuggestion VISA_SUGGESTION =
            createCreditCardSuggestion(
                    VISA.getCardNameForAutofillDisplay(),
                    VISA.getObfuscatedLastFourDigits(),
                    VISA.getFormattedExpirationDate(ContextUtils.getApplicationContext()),
                    /* applyDeactivatedStyle= */ false);
    private static final AutofillSuggestion MASTERCARD_SUGGESTION =
            createCreditCardSuggestion(
                    MASTERCARD.getCardNameForAutofillDisplay(),
                    MASTERCARD.getObfuscatedLastFourDigits(),
                    MASTERCARD.getFormattedExpirationDate(ContextUtils.getApplicationContext()),
                    /* applyDeactivatedStyle= */ false);
    private static final AutofillSuggestion SERVER_MASTERCARD_SUGGESTION =
            createCreditCardSuggestion(
                    SERVER_MASTERCARD.getCardNameForAutofillDisplay(),
                    SERVER_MASTERCARD.getObfuscatedLastFourDigits(),
                    SERVER_MASTERCARD.getFormattedExpirationDate(
                            ContextUtils.getApplicationContext()),
                    /* applyDeactivatedStyle= */ false);
    private static final AutofillSuggestion DISCOVER_SUGGESTION =
            createCreditCardSuggestion(
                    DISCOVER.getCardNameForAutofillDisplay(),
                    DISCOVER.getObfuscatedLastFourDigits(),
                    DISCOVER.getFormattedExpirationDate(ContextUtils.getApplicationContext()),
                    /* applyDeactivatedStyle= */ false);
    private static final AutofillSuggestion AMERICAN_EXPRESS_SUGGESTION =
            createCreditCardSuggestion(
                    AMERICAN_EXPRESS.getCardNameForAutofillDisplay(),
                    AMERICAN_EXPRESS.getObfuscatedLastFourDigits(),
                    AMERICAN_EXPRESS.getFormattedExpirationDate(
                            ContextUtils.getApplicationContext()),
                    /* applyDeactivatedStyle= */ false);
    private static final AutofillSuggestion ACCEPTABLE_MASTERCARD_VIRTUAL_CARD_SUGGESTION =
            createCreditCardSuggestion(
                    MASTERCARD_VIRTUAL_CARD.getCardNameForAutofillDisplay(),
                    MASTERCARD_VIRTUAL_CARD.getObfuscatedLastFourDigits(),
                    /* subLabel= */ "Virtual card",
                    /* applyDeactivatedStyle= */ false);
    private static final AutofillSuggestion NON_ACCEPTABLE_MASTERCARD_VIRTUAL_CARD_SUGGESTION =
            createCreditCardSuggestion(
                    MASTERCARD_VIRTUAL_CARD.getCardNameForAutofillDisplay(),
                    MASTERCARD_VIRTUAL_CARD.getObfuscatedLastFourDigits(),
                    /* subLabel= */ "Merchant doesn't accept this virtual card",
                    /* applyDeactivatedStyle= */ true);

    private BottomSheetController mBottomSheetController;
    private TouchToFillPaymentMethodCoordinator mCoordinator;

    public TouchToFillPaymentMethodRenderTest(boolean nightModeEnabled, boolean useRtlLayout) {
        setRtlForTesting(useRtlLayout);
        ChromeNightModeTestUtils.setUpNightModeForChromeActivity(nightModeEnabled);
        mRenderTestRule.setNightModeEnabled(nightModeEnabled);
        mRenderTestRule.setVariantPrefix(useRtlLayout ? "RTL" : "LTR");
    }

    @Before
    public void setUp() throws InterruptedException {
        MockitoAnnotations.initMocks(this);
        mActivityTestRule.startMainActivityOnBlankPage();
        mActivityTestRule.waitForActivityCompletelyLoaded();
        mBottomSheetController =
                mActivityTestRule
                        .getActivity()
                        .getRootUiCoordinatorForTesting()
                        .getBottomSheetController();
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator = new TouchToFillPaymentMethodCoordinator();
                    mCoordinator.initialize(
                            mActivityTestRule.getActivity(),
                            AutofillTestHelper.getPersonalDataManagerForLastUsedProfile(),
                            mBottomSheetController,
                            mDelegateMock,
                            mBottomSheetFocusHelper);
                });
    }

    @After
    public void tearDown() {
        setRtlForTesting(false);
        try {
            finishActivity(mActivityTestRule.getActivity());
        } catch (Exception e) {
            // Activity was already closed (e.g. due to last test tearing down the suite).
        }
        runOnUiThreadBlocking(() -> tearDownNightModeAfterChromeActivityDestroyed());
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    @RequiresRestart("crbug.com/344665938")
    public void testShowsOneCard() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA),
                            List.of(VISA_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(bottomSheetView, "touch_to_fill_credit_card_sheet_one_card");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsOneCardHalfState() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA),
                            List.of(VISA_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        ViewGroup bottomSheetParentView =
                (ViewGroup)
                        mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet).getParent();
        mRenderTestRule.render(
                bottomSheetParentView, "touch_to_fill_credit_card_sheet_one_card_half_state");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsTwoCards() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA, MASTERCARD),
                            List.of(VISA_SUGGESTION, MASTERCARD_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(bottomSheetView, "touch_to_fill_credit_card_sheet_two_cards");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsTwoCardsHalfState() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA, MASTERCARD),
                            List.of(VISA_SUGGESTION, MASTERCARD_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        ViewGroup bottomSheetParentView =
                (ViewGroup)
                        mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet).getParent();
        mRenderTestRule.render(
                bottomSheetParentView, "touch_to_fill_credit_card_sheet_two_cards_half_state");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsThreeCards() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA, MASTERCARD, DISCOVER),
                            List.of(VISA_SUGGESTION, MASTERCARD_SUGGESTION, DISCOVER_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(bottomSheetView, "touch_to_fill_credit_card_sheet_three_cards");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsThreeCardsHalfState() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA, MASTERCARD, DISCOVER),
                            List.of(VISA_SUGGESTION, MASTERCARD_SUGGESTION, DISCOVER_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        ViewGroup bottomSheetParentView =
                (ViewGroup)
                        mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet).getParent();
        mRenderTestRule.render(
                bottomSheetParentView, "touch_to_fill_credit_card_sheet_three_cards_half_state");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsFourCards() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA, MASTERCARD, DISCOVER, AMERICAN_EXPRESS),
                            List.of(
                                    VISA_SUGGESTION,
                                    MASTERCARD_SUGGESTION,
                                    DISCOVER_SUGGESTION,
                                    AMERICAN_EXPRESS_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(bottomSheetView, "touch_to_fill_credit_card_sheet_four_cards");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsFourCardsHalfState() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA, MASTERCARD, DISCOVER, AMERICAN_EXPRESS),
                            List.of(
                                    VISA_SUGGESTION,
                                    MASTERCARD_SUGGESTION,
                                    DISCOVER_SUGGESTION,
                                    AMERICAN_EXPRESS_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        ViewGroup bottomSheetParentView =
                (ViewGroup)
                        mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet).getParent();
        mRenderTestRule.render(
                bottomSheetParentView, "touch_to_fill_credit_card_sheet_four_cards_half_state");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsLocalAndServerAndVirtualCards() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA, MASTERCARD_VIRTUAL_CARD, SERVER_MASTERCARD),
                            List.of(
                                    VISA_SUGGESTION,
                                    ACCEPTABLE_MASTERCARD_VIRTUAL_CARD_SUGGESTION,
                                    SERVER_MASTERCARD_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(
                bottomSheetView,
                "touch_to_fill_credit_card_sheet_shows_local_and_server_and_virtual_cards");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsLocalAndServerAndNonAcceptableVirtualCards() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA, MASTERCARD_VIRTUAL_CARD, SERVER_MASTERCARD),
                            List.of(
                                    VISA_SUGGESTION,
                                    NON_ACCEPTABLE_MASTERCARD_VIRTUAL_CARD_SUGGESTION,
                                    SERVER_MASTERCARD_SUGGESTION),
                            /* shouldShowScanCreditCard= */ true);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(
                bottomSheetView,
                "touch_to_fill_credit_card_sheet_shows_local_and_server_and_non_acceptable_virtual_cards");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    @RequiresRestart("crbug.com/344665938")
    public void testScanNewCardButtonIsHidden() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(
                            List.of(VISA),
                            List.of(VISA_SUGGESTION),
                            /* shouldShowScanCreditCard= */ false);
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(
                bottomSheetView, "touch_to_fill_credit_card_sheet_scan_credit_card_hidden");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsOneIban() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(List.of(LOCAL_IBAN));
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(bottomSheetView, "touch_to_fill_iban_sheet_one_iban");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsOneIbanHalfState() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(List.of(LOCAL_IBAN));
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        ViewGroup bottomSheetParentView =
                (ViewGroup)
                        mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet).getParent();
        mRenderTestRule.render(
                bottomSheetParentView, "touch_to_fill_iban_sheet_one_iban_half_state");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsTwoIbans() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(List.of(LOCAL_IBAN, LOCAL_IBAN_NO_NICKNAME));
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        View bottomSheetView = mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet);
        mRenderTestRule.render(bottomSheetView, "touch_to_fill_iban_sheet_two_ibans");
    }

    @Test
    @MediumTest
    @Feature({"RenderTest"})
    public void testShowsTwoIbansHalfState() throws IOException {
        runOnUiThreadBlocking(
                () -> {
                    mCoordinator.showSheet(List.of(LOCAL_IBAN, LOCAL_IBAN_NO_NICKNAME));
                });
        BottomSheetTestSupport.waitForOpen(mBottomSheetController);

        ViewGroup bottomSheetParentView =
                (ViewGroup)
                        mActivityTestRule.getActivity().findViewById(R.id.bottom_sheet).getParent();
        mRenderTestRule.render(
                bottomSheetParentView, "touch_to_fill_iban_sheet_two_ibans_half_state");
    }
}