def test_clean_ccv_invalid(self):
"""
Test clean ccv code
"""
form = PaymentForm()
form.cleaned_data = {'ccv': 'invalid cvv'}
self.assertRaises(ValidationError, form.clean_ccv)
def test_clean_ccv(self):
"""
Test clean ccv code
"""
expected_value = '123'
form = PaymentForm()
form.cleaned_data = {'ccv': expected_value}
value = form.clean_ccv()
self.assertEqual(expected_value, value)
def test_clean_method(self):
"""
Test that clean method validates expire date
"""
form = PaymentForm()
form.cleaned_data = {}
with patch('salest.payments.forms.PaymentForm.validate_expire_date')\
as valid_date:
form.clean()
valid_date.assert_called_once_with()
def test_validate_expire_date_valid(self):
"""
Test that form can validate expire date
"""
expire_month = datetime.datetime.now().month
expire_year = datetime.datetime.now().year + 1
form = PaymentForm()
form.cleaned_data = {'expire_month': expire_month,
'expire_year': expire_year}
result = form.validate_expire_date()
self.assertTrue(result)
def test_clean_card_number(self):
"""
Test clean card number
"""
expected_value = '4111111111111111'
form = PaymentForm()
form.cleaned_data = {'card_number': '4111-1111-1111-1111'}
with patch('salest.payments.forms.validate_card_number') as valid_card:
value = form.clean_card_number()
valid_card.assert_called_once_with(expected_value)
self.assertEqual(expected_value, value)
def test_validate_expire_date_invalid(self):
"""
Test that form rise exception when expire date is invalid
"""
expire_month = datetime.datetime.now().month - 1
expire_year = datetime.datetime.now().year
form = PaymentForm()
form.cleaned_data = {'expire_month': expire_month,
'expire_year': expire_year}
result = form.validate_expire_date()
self.assertFalse(result)
