iceraven-browser/app/src/test/java/org/mozilla/fenix/settings/creditcards/CreditCardEditorViewTest.kt

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

package org.mozilla.fenix.settings.creditcards

import android.view.LayoutInflater
import android.view.View
import io.mockk.every
import io.mockk.mockk
import io.mockk.spyk
import io.mockk.verify
import kotlinx.coroutines.test.runTest
import mozilla.components.concept.storage.CreditCard
import mozilla.components.concept.storage.CreditCardNumber
import mozilla.components.concept.storage.NewCreditCardFields
import mozilla.components.concept.storage.UpdatableCreditCardFields
import mozilla.components.service.sync.autofill.AutofillCreditCardsAddressesStorage
import mozilla.components.service.sync.autofill.AutofillCrypto
import mozilla.components.support.ktx.android.content.getColorFromAttr
import mozilla.components.support.test.robolectric.testContext
import mozilla.components.support.utils.CreditCardNetworkType
import org.junit.Assert.assertEquals
import org.junit.Assert.assertFalse
import org.junit.Assert.assertNotNull
import org.junit.Assert.assertNull
import org.junit.Assert.assertTrue
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
import org.mozilla.fenix.R
import org.mozilla.fenix.databinding.FragmentCreditCardEditorBinding
import org.mozilla.fenix.ext.toEditable
import org.mozilla.fenix.helpers.FenixRobolectricTestRunner
import org.mozilla.fenix.settings.creditcards.CreditCardEditorFragment.Companion.NUMBER_OF_YEARS_TO_SHOW
import org.mozilla.fenix.settings.creditcards.interactor.CreditCardEditorInteractor
import org.mozilla.fenix.settings.creditcards.view.CreditCardEditorView
import java.util.Calendar

@RunWith(FenixRobolectricTestRunner::class)
class CreditCardEditorViewTest {

    private lateinit var view: View
    private lateinit var interactor: CreditCardEditorInteractor
    private lateinit var creditCardEditorView: CreditCardEditorView
    private lateinit var storage: AutofillCreditCardsAddressesStorage
    private lateinit var crypto: AutofillCrypto
    private lateinit var fragmentCreditCardEditorBinding: FragmentCreditCardEditorBinding

    private val cardNumber = "4111111111111111"
    private val creditCard = CreditCard(
        guid = "id",
        billingName = "Banana Apple",
        encryptedCardNumber = CreditCardNumber.Encrypted(cardNumber),
        cardNumberLast4 = "1111",
        expiryMonth = 5,
        expiryYear = 2030,
        cardType = CreditCardNetworkType.VISA.cardName,
        timeCreated = 1L,
        timeLastUsed = 1L,
        timeLastModified = 1L,
        timesUsed = 1L,
    )

    @Before
    fun setup() {
        view = LayoutInflater.from(testContext).inflate(R.layout.fragment_credit_card_editor, null)
        fragmentCreditCardEditorBinding = FragmentCreditCardEditorBinding.bind(view)
        interactor = mockk(relaxed = true)
        storage = mockk(relaxed = true)
        crypto = mockk(relaxed = true)

        every { storage.getCreditCardCrypto() } returns crypto
        every { crypto.decrypt(any(), any()) } returns CreditCardNumber.Plaintext(cardNumber)

        creditCardEditorView = spyk(CreditCardEditorView(fragmentCreditCardEditorBinding, interactor))
    }

    @Test
    fun `GIVEN the initial credit card editor state THEN credit card form inputs are in initial state`() {
        creditCardEditorView.bind(getInitialCreditCardEditorState())

        val calendar = Calendar.getInstance()
        val startYear = calendar.get(Calendar.YEAR)
        val endYear = startYear + NUMBER_OF_YEARS_TO_SHOW - 1

        assertEquals("", fragmentCreditCardEditorBinding.cardNumberInput.text.toString())
        assertEquals("", fragmentCreditCardEditorBinding.nameOnCardInput.text.toString())

        with(fragmentCreditCardEditorBinding.expiryMonthDropDown) {
            assertEquals(12, count)
            assertEquals("January (01)", selectedItem.toString())
            assertEquals("December (12)", getItemAtPosition(count - 1).toString())
        }

        with(fragmentCreditCardEditorBinding.expiryYearDropDown) {
            assertEquals(10, count)
            assertEquals(startYear.toString(), selectedItem.toString())
            assertEquals(endYear.toString(), getItemAtPosition(count - 1).toString())
        }

        assertEquals(View.GONE, fragmentCreditCardEditorBinding.deleteButton.visibility)
    }

    @Test
    fun `GIVEN a credit card THEN credit card form inputs are displaying the provided credit card information`() = runTest {
        creditCardEditorView.bind(creditCard.toCreditCardEditorState(storage))

        assertEquals(cardNumber, fragmentCreditCardEditorBinding.cardNumberInput.text.toString())
        assertEquals(creditCard.billingName, fragmentCreditCardEditorBinding.nameOnCardInput.text.toString())

        with(fragmentCreditCardEditorBinding.expiryMonthDropDown) {
            assertEquals(12, count)
            assertEquals("May (05)", selectedItem.toString())
        }

        with(fragmentCreditCardEditorBinding.expiryYearDropDown) {
            val endYear = creditCard.expiryYear + NUMBER_OF_YEARS_TO_SHOW - 1

            assertEquals(10, count)
            assertEquals(creditCard.expiryYear.toString(), selectedItem.toString())
            assertEquals(endYear.toString(), getItemAtPosition(count - 1).toString())
        }
    }

    @Test
    fun `GIVEN a credit card WHEN the delete card button is clicked THEN interactor is called`() = runTest {
        creditCardEditorView.bind(creditCard.toCreditCardEditorState(storage))

        assertEquals(View.VISIBLE, fragmentCreditCardEditorBinding.deleteButton.visibility)

        fragmentCreditCardEditorBinding.deleteButton.performClick()

        verify { interactor.onDeleteCardButtonClicked(creditCard.guid) }
    }

    @Test
    fun `WHEN the cancel button is clicked THEN interactor is called`() {
        creditCardEditorView.bind(getInitialCreditCardEditorState())

        fragmentCreditCardEditorBinding.cancelButton.performClick()

        verify { interactor.onCancelButtonClicked() }
    }

    @Test
    fun `GIVEN invalid credit card number WHEN the save button is clicked THEN interactor is not called`() {
        creditCardEditorView.bind(getInitialCreditCardEditorState())

        val calendar = Calendar.getInstance()

        var billingName = "Banana Apple"
        val cardNumber = "2221000000000000"
        val expiryMonth = 5
        val expiryYear = calendar.get(Calendar.YEAR)

        fragmentCreditCardEditorBinding.cardNumberInput.text = cardNumber.toEditable()
        fragmentCreditCardEditorBinding.nameOnCardInput.text = billingName.toEditable()
        fragmentCreditCardEditorBinding.expiryMonthDropDown.setSelection(expiryMonth - 1)

        fragmentCreditCardEditorBinding.saveButton.performClick()

        verify {
            creditCardEditorView.validateForm()
        }

        assertFalse(creditCardEditorView.validateForm())
        assertNotNull(fragmentCreditCardEditorBinding.cardNumberLayout.error)
        assertEquals(
            fragmentCreditCardEditorBinding.cardNumberLayout.errorCurrentTextColors,
            fragmentCreditCardEditorBinding.root.context.getColorFromAttr(R.attr.textWarning),
        )

        verify(exactly = 0) {
            interactor.onSaveCreditCard(
                NewCreditCardFields(
                    billingName = billingName,
                    plaintextCardNumber = CreditCardNumber.Plaintext(cardNumber),
                    cardNumberLast4 = "0000",
                    expiryMonth = expiryMonth.toLong(),
                    expiryYear = expiryYear.toLong(),
                    cardType = CreditCardNetworkType.MASTERCARD.cardName,
                ),
            )
        }

        billingName = ""
        fragmentCreditCardEditorBinding.nameOnCardInput.text = billingName.toEditable()

        fragmentCreditCardEditorBinding.saveButton.performClick()

        assertFalse(creditCardEditorView.validateForm())
        assertNotNull(fragmentCreditCardEditorBinding.cardNumberLayout.error)
        assertEquals(
            fragmentCreditCardEditorBinding.cardNumberLayout.errorCurrentTextColors,
            fragmentCreditCardEditorBinding.root.context.getColorFromAttr(R.attr.textWarning),
        )

        verify(exactly = 0) {
            interactor.onSaveCreditCard(
                NewCreditCardFields(
                    billingName = billingName,
                    plaintextCardNumber = CreditCardNumber.Plaintext(cardNumber),
                    cardNumberLast4 = "0000",
                    expiryMonth = expiryMonth.toLong(),
                    expiryYear = expiryYear.toLong(),
                    cardType = CreditCardNetworkType.MASTERCARD.cardName,
                ),
            )
        }
    }

    @Test
    fun `GIVEN invalid credit card values WHEN valid values are entered and the save button is clicked THEN error messages are cleared`() {
        creditCardEditorView.bind(getInitialCreditCardEditorState())

        var billingName = ""
        var cardNumber = "1234567891234567"
        val expiryMonth = 5

        fragmentCreditCardEditorBinding.cardNumberInput.text = cardNumber.toEditable()
        fragmentCreditCardEditorBinding.nameOnCardInput.text = billingName.toEditable()
        fragmentCreditCardEditorBinding.expiryMonthDropDown.setSelection(expiryMonth - 1)

        fragmentCreditCardEditorBinding.saveButton.performClick()

        verify {
            creditCardEditorView.validateForm()
        }

        billingName = "Banana Apple"
        cardNumber = "2720994326581252"
        fragmentCreditCardEditorBinding.nameOnCardInput.text = billingName.toEditable()
        fragmentCreditCardEditorBinding.cardNumberInput.text = cardNumber.toEditable()

        fragmentCreditCardEditorBinding.saveButton.performClick()

        verify {
            creditCardEditorView.validateForm()
        }

        assertTrue(creditCardEditorView.validateForm())
        assertNull(fragmentCreditCardEditorBinding.cardNumberLayout.error)
        assertNull(fragmentCreditCardEditorBinding.nameOnCardLayout.error)
    }

    @Test
    fun `GIVEN invalid name on card WHEN the save button is clicked THEN interactor is not called`() {
        creditCardEditorView.bind(getInitialCreditCardEditorState())

        val calendar = Calendar.getInstance()

        val billingName = "       "
        val cardNumber = "2221000000000000"
        val expiryMonth = 5
        val expiryYear = calendar.get(Calendar.YEAR)

        fragmentCreditCardEditorBinding.cardNumberInput.text = cardNumber.toEditable()
        fragmentCreditCardEditorBinding.nameOnCardInput.text = billingName.toEditable()
        fragmentCreditCardEditorBinding.expiryMonthDropDown.setSelection(expiryMonth - 1)

        fragmentCreditCardEditorBinding.saveButton.performClick()

        verify {
            creditCardEditorView.validateForm()
        }

        assertFalse(creditCardEditorView.validateForm())
        assertNotNull(fragmentCreditCardEditorBinding.nameOnCardLayout.error)
        assertEquals(
            fragmentCreditCardEditorBinding.nameOnCardLayout.errorCurrentTextColors,
            fragmentCreditCardEditorBinding.root.context.getColorFromAttr(R.attr.textWarning),
        )

        verify(exactly = 0) {
            interactor.onSaveCreditCard(
                NewCreditCardFields(
                    billingName = billingName,
                    plaintextCardNumber = CreditCardNumber.Plaintext(cardNumber),
                    cardNumberLast4 = "0000",
                    expiryMonth = expiryMonth.toLong(),
                    expiryYear = expiryYear.toLong(),
                    cardType = CreditCardNetworkType.MASTERCARD.cardName,
                ),
            )
        }
    }

    @Test
    fun `GIVEN valid credit card number WHEN the save button is clicked THEN interactor is called`() {
        creditCardEditorView.bind(getInitialCreditCardEditorState())

        val calendar = Calendar.getInstance()

        val billingName = "Banana Apple"
        val cardNumber = "2720994326581252"
        val expiryMonth = 5
        val expiryYear = calendar.get(Calendar.YEAR)

        fragmentCreditCardEditorBinding.cardNumberInput.text = cardNumber.toEditable()
        fragmentCreditCardEditorBinding.nameOnCardInput.text = billingName.toEditable()
        fragmentCreditCardEditorBinding.expiryMonthDropDown.setSelection(expiryMonth - 1)

        fragmentCreditCardEditorBinding.saveButton.performClick()

        verify {
            creditCardEditorView.validateForm()
        }

        assertTrue(creditCardEditorView.validateForm())

        verify {
            interactor.onSaveCreditCard(
                NewCreditCardFields(
                    billingName = billingName,
                    plaintextCardNumber = CreditCardNumber.Plaintext(cardNumber),
                    cardNumberLast4 = "1252",
                    expiryMonth = expiryMonth.toLong(),
                    expiryYear = expiryYear.toLong(),
                    cardType = CreditCardNetworkType.MASTERCARD.cardName,
                ),
            )
        }
    }

    @Test
    fun `GIVEN a valid credit card WHEN the save button is clicked THEN interactor is called`() = runTest {
        creditCardEditorView.bind(creditCard.toCreditCardEditorState(storage))

        fragmentCreditCardEditorBinding.saveButton.performClick()

        verify {
            interactor.onUpdateCreditCard(
                guid = creditCard.guid,
                creditCardFields = UpdatableCreditCardFields(
                    billingName = creditCard.billingName,
                    cardNumber = CreditCardNumber.Plaintext(cardNumber),
                    cardNumberLast4 = creditCard.cardNumberLast4,
                    expiryMonth = creditCard.expiryMonth,
                    expiryYear = creditCard.expiryYear,
                    cardType = creditCard.cardType,
                ),
            )
        }
    }
}