edan_f0/tests/unit/address_tests.py

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

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


User = get_user_model()


class TestUserAddress(TestCase):

    def test_uses_title_firstname_and_lastname_in_salutation(self):
        a = models.UserAddress(
            title="Dr",
            first_name="Barry",
            last_name='Barrington')
        self.assertEquals("Dr Barry Barrington", a.salutation)

    def test_strips_whitespace_from_salutation(self):
        a = models.UserAddress(last_name='Barrington')
        self.assertEquals("Barrington", a.salutation)

    def test_has_name_property(self):
        a = models.UserAddress(
            title="Dr",
            first_name="Barry",
            last_name='Barrington')
        self.assertEquals("Barry Barrington", a.name)

    def test_has_summary_property(self):
        a = models.UserAddress(
            title="Dr",
            first_name="Barry",
            last_name='Barrington',
            line1="1 King Road",
            line4="London",
            postcode="SW1 9RE")
        self.assertEquals("Dr Barry Barrington, 1 King Road, London, SW1 9RE",
                          a.summary)

    def test_summary_includes_country(self):
        c = models.Country(
            pk=1, iso_3166_1_a2="GB", name="UNITED KINGDOM")
        a = models.UserAddress(
            title="Dr",
            first_name="Barry",
            last_name='Barrington',
            line1="1 King Road",
            line4="London",
            postcode="SW1 9RE",
            country=c)
        self.assertEquals(
            "Dr Barry Barrington, 1 King Road, London, SW1 9RE, UNITED KINGDOM",
            a.summary)

    def test_can_be_hashed(self):
        a = models.UserAddress(
            title="Dr",
            first_name="Barry",
            last_name='Barrington',
            line1="1 King Road",
            line4="London",
            postcode="SW1 9RE")
        hash = a.generate_hash()
        self.assertTrue(hash is not None)

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

    def test_strips_whitespace_in_name_property(self):
        a = models.UserAddress(
            last_name='Barrington')
        self.assertEquals("Barrington", a.name)

    def test_uses_city_as_an_alias_of_line4(self):
        a = models.UserAddress(
            last_name='Barrington',
            line1="75 Smith Road",
            line4="London",
            postcode="n4 8ty")
        self.assertEqual('London', a.city)