edan_f0/tests/unit/payment/bankcard_tests.py

import nose.tools

from oscar.apps.payment import bankcards


fixture_data = {
    bankcards.VISA: ('4111111111111111',),
    bankcards.MASTERCARD: ('5500000000000004',),
    bankcards.DISCOVER: ('6011000000000004',),
    bankcards.AMEX: ('340000000000009',),
    None: ('1000010000000007',)  # Magic number
}


def compare(number, type):
    nose.tools.eq_(bankcards.bankcard_type(number), type)


def test_bankcard_type_sniffing():

    for bankcard_type, numbers in fixture_data.items():
        for number in numbers:
            yield compare, number, bankcard_type


valid_numbers = [
    '4111111111111111',
    '5500000000000004',
    '6011000000000004',
    '340000000000009']

invalid_numbers = [
    '4111111111111110',
    '5500000000000009',
    '6011000000000000',
    '340000000000005']


def is_valid(n):
    assert bankcards.luhn(n) is True


def is_not_valid(n):
    assert bankcards.luhn(n) is False


def test_luhn_check():
    for number in valid_numbers:
        yield is_valid, number

    for number in invalid_numbers:
        yield is_not_valid, number