edan_f0/tests/unit/address/model_tests.py

# -*- coding: utf-8 -*-
from django.test import TestCase
from django.core import exceptions

from nose.tools import raises

from oscar.apps.order.models import ShippingAddress
from oscar.core.compat import get_user_model
from oscar.apps.address import models

from .factories import UserAddressFactory, CountryFactory, UserFactory


User = get_user_model()


class TestUserAddress(TestCase):

    def setUp(self):
        self.country = CountryFactory.build()
        self.user = UserFactory.build()

    def test_uses_title_firstname_and_lastname_in_salutation(self):
        a = UserAddressFactory.build(country=self.country,
                title='Dr',
                first_name='Barry',
                last_name='Barrington',
                user=self.user)
        self.assertEqual("Dr Barry Barrington", a.salutation)

    def test_strips_whitespace_from_salutation(self):
        a = UserAddressFactory.build(title='',
                first_name='',
                last_name='Barrington',
                country=self.country,
                user=self.user)
        self.assertEqual("Barrington", a.salutation)

    def test_has_name_property(self):
        a = UserAddressFactory.build(country=self.country,
                first_name='Barry',
                last_name='Barrington',
                user=self.user)
        self.assertEqual("Barry Barrington", a.name)

    def test_has_summary_property(self):
        c = CountryFactory.build(name='')
        a = UserAddressFactory.build(country=c,
                title='Dr',
                first_name='Barry',
                last_name='Barrington',
                line1='1 King Road',
                line4='London',
                postcode='SW1 9RE')
        self.assertEqual("Dr Barry Barrington, 1 King Road, London, SW1 9RE, ",
                          a.summary)

    def test_summary_includes_country(self):
        c = CountryFactory.build(name='UNITED KINGDOM')
        a = UserAddressFactory.build(country=c,
                title='Dr',
                first_name='Barry',
                last_name='Barrington',
                line1='1 King Road',
                line4='London',
                postcode='SW1 9RE',
                user=self.user)
        self.assertEqual(
            "Dr Barry Barrington, 1 King Road, London, SW1 9RE, UNITED KINGDOM",
            a.summary)

    def test_can_be_hashed(self):
        a = UserAddressFactory.build(country=self.country,
            user=self.user)
        hash = a.generate_hash()
        self.assertTrue(hash is not None)

    def test_can_be_hashed_including_non_ascii(self):
        a = UserAddressFactory.build(first_name=u"\u0141ukasz Smith",
                last_name=u'Smith',
                line1=u"75 Smith Road",
                postcode=u"n4 8ty",
                country=self.country,
                user=self.user)
        hash = a.generate_hash()
        self.assertTrue(hash is not None)

    def test_strips_whitespace_in_name_property(self):
        a = UserAddressFactory.build(first_name='', 
            last_name='Barrington',
            country=self.country,
            user=self.user)
        self.assertEqual("Barrington", a.name)

    def test_uses_city_as_an_alias_of_line4(self):
        a = UserAddressFactory.build(line4='London',
            country=self.country,
            user=self.user)
        self.assertEqual('London', a.city)

    def test_converts_postcode_to_uppercase_when_cleaning(self):
        address = UserAddressFactory.build(postcode="n4 8ty",
            country=self.country,
            user=self.user)
        address.clean()
        self.assertEqual("N4 8TY", address.postcode)

    def test_strips_whitespace_when_cleaning(self):
        a = UserAddressFactory.build(line1="  75 Smith Road  ", 
            postcode="  n4 8ty",
            country=self.country,
            user=self.user)
        a.clean()
        self.assertEqual("N4 8TY", a.postcode)
        self.assertEqual("75 Smith Road", a.line1)

    def test_active_address_fields_skips_whitespace_only_fields(self):
        a = UserAddressFactory.build(first_name='   ',
            last_name='Barrington',
            line1='  75 Smith Road  ',
            postcode='  n4 8ty',
            title='',
            country=self.country,
            user=self.user)
        active_fields = a.active_address_fields()
        self.assertEqual("Barrington", active_fields[0])

    def test_ignores_whitespace_when_hashing(self):
        a1 = UserAddressFactory.build(first_name='Terry',
            last_name='Barrington',
            country=self.country,
            user=self.user)
        a1.clean()
        a2 = UserAddressFactory.build(first_name='   Terry  ',
            last_name='     Barrington',
            country=self.country,
            user=self.user)
        a2.clean()
        self.assertEqual(a1.generate_hash(), a2.generate_hash())

    def test_populate_shipping_address_doesnt_set_id(self):
        a = UserAddressFactory.build(country=self.country,
            user=self.user)
        a.clean()
        sa = ShippingAddress()
        a.populate_alternative_model(sa)
        self.assertIsNone(sa.id)

    def test_populated_shipping_address_has_same_summary_user_address(self):
        a = UserAddressFactory.build(country=self.country,
            user=self.user)
        a.clean()
        sa = ShippingAddress()
        a.populate_alternative_model(sa)
        self.assertEqual(sa.summary, a.summary)

    def test_summary_is_property(self):
        a = UserAddressFactory.build(title='', 
            line4='',
            first_name=" Terry  ",
            last_name='Barrington',
            line1="  75 Smith Road  ",
            postcode="  n4 8ty",
            country=self.country,
            user=self.user)
        a.clean()
        self.assertEqual(
            "Terry Barrington, 75 Smith Road, N4 8TY, UNITED KINGDOM",
            a.summary)

VALID_POSTCODES = [
    ('GB', 'N1 9RT'),
    ('SK', '991 41'),
    ('CZ', '612 00'),
    ('CC', '6799'),
    ('CY', '8240'),
    ('MC', '98000'),
    ('SH', 'STHL 1ZZ'),
    ('JP', '150-2345'),
    ('PG', '314'),
    ('HN', '41202'),
    # It works for small cases as well
    ('GB', 'sw2 1rw'),
]


INVALID_POSTCODES = [
    ('GB', 'not-a-postcode'),
    ('DE', '123b4'),
]


def assert_valid_postcode(country_value, postcode_value):
    country = models.Country(iso_3166_1_a2=country_value)
    address = models.UserAddress(country=country, postcode=postcode_value)
    address.clean()


@raises(exceptions.ValidationError)
def assert_invalid_postcode(country_value, postcode_value):
    country = models.Country(iso_3166_1_a2=country_value)
    address = models.UserAddress(country=country, postcode=postcode_value)
    address.clean()


def test_postcode_is_validated_for_country():
    for country, postcode in VALID_POSTCODES:
        yield assert_valid_postcode, country, postcode


def test_postcode_is_only_valid():
    for country, postcode in INVALID_POSTCODES:
        yield assert_invalid_postcode, country, postcode