edan_f0/tests/unit/payment/form_tests.py

import datetime

from django.test import TestCase
from django.forms import ValidationError

from oscar.apps.payment import forms, models


class TestBankcardNumberField(TestCase):

    def setUp(self):
        self.field = forms.BankcardNumberField()

    def test_strips_non_digits(self):
        self.assertEquals(
            '4111111111111111', self.field.clean('  4111 1111 1111 1111'))

    def test_rejects_numbers_which_dont_pass_luhn(self):
        with self.assertRaises(ValidationError):
            self.field.clean('1234123412341234')


class TestStartingMonthField(TestCase):

    def setUp(self):
        self.field = forms.BankcardStartingMonthField()

    def test_returns_a_date(self):
        start_date = self.field.clean(['01', '2010'])
        self.assertTrue(isinstance(start_date, datetime.date))

    def test_rejects_invalid_months(self):
        with self.assertRaises(ValidationError):
            self.field.clean(['00', '2010'])

    def test_rejects_invalid_years(self):
        with self.assertRaises(ValidationError):
            self.field.clean(['01', '201'])

    def test_rejects_months_in_the_future(self):
        today = datetime.date.today()
        with self.assertRaises(ValidationError):
            self.field.clean(['01', today.year + 1])

    def test_returns_the_first_day_of_month(self):
        start_date = self.field.clean(['01', '2010'])
        self.assertEquals(1, start_date.day)


class TestExpiryMonthField(TestCase):

    def setUp(self):
        self.field = forms.BankcardExpiryMonthField()

    def test_returns_a_date(self):
        today = datetime.date.today()
        end_date = self.field.clean(['01', today.year + 1])
        self.assertTrue(isinstance(end_date, datetime.date))

    def test_rejects_invalid_months(self):
        with self.assertRaises(ValidationError):
            self.field.clean(['00', '2010'])

    def test_rejects_invalid_years(self):
        with self.assertRaises(ValidationError):
            self.field.clean(['01', '201'])

    def test_rejects_months_in_the_past(self):
        today = datetime.date.today()
        with self.assertRaises(ValidationError):
            self.field.clean(['01', today.year - 1])

    def test_returns_last_day_of_month(self):
        today = datetime.date.today()
        end_date = self.field.clean(['01', today.year + 1])
        self.assertEquals(31, end_date.day)

    def test_defaults_to_current_month(self):
        today = datetime.date.today()
        self.assertEquals(["%.2d" % today.month, today.year],
                          self.field.initial)


class TestCCVField(TestCase):
    """CCV field"""

    def setUp(self):
        self.field = forms.BankcardCCVField()

    def test_is_required_by_default(self):
        with self.assertRaises(ValidationError):
            self.field.clean("")

    def test_only_permits_3_or_4_digit_numbers(self):
        invalid = ['00', '12a', '12345']
        for sample in invalid:
            with self.assertRaises(ValidationError):
                self.field.clean(sample)
        valid = ['123', '  123   ', '1235']
        for sample in valid:
            self.field.clean(sample)

    def test_has_meaningful_error_message(self):
        try:
            self.field.clean("asdf")
        except ValidationError, e:
            self.assertEquals("Please enter a 3 or 4 digit number",
                              e.messages[0])


class TestValidBankcardForm(TestCase):

    def setUp(self):
        today = datetime.date.today()
        data = {
            'number': '1000010000000007',
            'ccv': '123',
            'expiry_month_0': '01',
            'expiry_month_1': today.year + 1,
        }
        self.form = forms.BankcardForm(data)
        self.assertTrue(self.form.is_valid())

    def test_has_bankcard_property(self):
        self.assertTrue(isinstance(self.form.bankcard, models.Bankcard))

    def test_returns_bankcard_with_sensitive_data_intact(self):
        bankcard = self.form.bankcard
        self.assertFalse(bankcard.number.startswith('X'))
        self.assertEquals('123', bankcard.ccv)