edan_f0/tests/integration/payment/test_models.py

import datetime
from decimal import Decimal as D

from django.test import TestCase

from oscar.apps.payment.models import Bankcard, Source
from oscar.core.compat import get_user_model
from oscar.test import factories


class TestBankcard(TestCase):
    def test_obfuscates_number_before_saving(self):
        bankcard = Bankcard(number="1000011100000004")
        bankcard.prepare_for_save()
        self.assertEqual("XXXX-XXXX-XXXX-0004", bankcard.number)

    def test_determines_bankcard_type(self):
        bankcard = Bankcard(number="5500000000000004")
        self.assertEqual("Mastercard", bankcard.card_type)

    def test_provides_start_month_property(self):
        start = datetime.date(day=1, month=1, year=2010)
        bankcard = Bankcard(start_date=start)
        self.assertEqual("01/10", bankcard.start_month())

    def test_provides_end_month_property(self):
        end = datetime.date(day=1, month=1, year=2010)
        bankcard = Bankcard(expiry_date=end)
        self.assertEqual("01/10", bankcard.expiry_month())

    def test_bankcard_card_correct_save(self):
        # issue #1486
        user_klass = get_user_model()
        user = user_klass.objects.create_user("_", "a@a.com", "pwd")
        end = datetime.date(day=1, month=1, year=2010)
        bankcard = Bankcard.objects.create(
            user=user, number="5500000000000004", expiry_date=end
        )
        saved_bankcard = Bankcard.objects.get(id=bankcard.id)
        self.assertEqual("Mastercard", saved_bankcard.card_type)

    def test_cardtype_persists_after_save(self):
        user = factories.UserFactory()
        end = datetime.date(day=1, month=1, year=2010)
        bankcard = Bankcard(user=user, number="5500000000000004", expiry_date=end)
        self.assertEqual("Mastercard", bankcard.card_type)

        bankcard.save()
        self.assertEqual("Mastercard", bankcard.card_type)

        reloaded_bankcard = Bankcard.objects.get(id=bankcard.id)
        self.assertEqual("Mastercard", reloaded_bankcard.card_type)


class TestSource(TestCase):
    def test_calculates_initial_balance_correctly(self):
        source = Source(amount_allocated=D("100"))
        self.assertEqual(D("100"), source.balance)

    def test_calculates_balance_correctly(self):
        source = Source(
            amount_allocated=D("100"), amount_debited=D("80"), amount_refunded=D("20")
        )
        self.assertEqual(D("100") - D("80") + D("20"), source.balance)

    def test_calculates_amount_for_refund_correctly(self):
        source = Source(
            amount_allocated=D("100"), amount_debited=D("80"), amount_refunded=D("20")
        )
        self.assertEqual(D("80") - D("20"), source.amount_available_for_refund)


class TestAPaymentSource(TestCase):
    def setUp(self):
        order = factories.create_order()
        self.source = factories.SourceFactory(order=order)

    def test_allocation_doesnt_error(self):
        self.source.allocate(D("100.00"))

    def test_debit_doesnt_error(self):
        self.source.allocate(D("100.00"))
        self.source.debit(D("80.00"))

    def test_full_debit_doesnt_error(self):
        self.source.allocate(D("100.00"))
        self.source.debit()
        self.assertEqual(D("0.00"), self.source.balance)

    def test_refund_doesnt_error(self):
        self.source.allocate(D("100.00"))
        self.source.debit(D("80.00"))
        self.source.refund(D("50.00"))