root / trunk / tests / test_address_list.py

#!/usr/bin/env python
# encoding: utf-8

"""Test the TurboMail Message class."""


import unittest
from turbomail.util import Address, AddressList
try:
    from email.mime.text import MIMEText
except ImportError:
    # Python < 2.5 has no email.mime module
    from email.MIMEText import MIMEText

import logging
logging.disable(logging.WARNING)


class TestAddress(unittest.TestCase):
    def test_punycode(self):
        addr = Address("Foo", u"foo@exámple.test")
        self.assertEqual("Foo <foo@xn--exmple-qta.test>", str(addr))
    
    def test_initialization_with_addresslist_which_contains_only_email(self):
        emailaddress = 'foo@example.com'
        address = Address(AddressList(emailaddress))
        self.assertEqual(emailaddress, str(address))
    
    def test_initialization_with_addresslist_which_contains_tuple(self):
        name = 'Foo'
        emailaddress = 'foo@example.com'
        address = Address(AddressList((name, emailaddress)))
        self.assertEqual('%s <%s>' % (name, emailaddress), str(address))
    
    def test_initialization_with_tuple(self):
        name = 'Foo'
        emailaddress = 'foo@example.com'
        address = Address((name, emailaddress))
        self.assertEqual('%s <%s>' % (name, emailaddress), str(address))
    
    def test_initialization_with_string(self):
        emailaddress = 'foo@example.com'
        address = Address(emailaddress)
        self.assertEqual('%s' % emailaddress, str(address))
        


class TestAddressList(unittest.TestCase):
    """Test the AddressList helper class."""
    
    addresses = AddressList.protected('_addresses')
    
    def setUp(self):
        self._addresses = AddressList()
    
    # Helper function to avoid repetition.
    def check(self, addresses, string):
        self.assertEqual(self.addresses, addresses)
        self.assertEqual(str(self.addresses), string)
    
    def test_assignment(self):
        self.assertEqual(self.addresses, [])
        
        # A single address.
        addresses = "user@example.com"
        self.addresses = addresses
        self.check([addresses], addresses)
        
        # A list of addresses.
        addresses = ["user1@example.com", "user2@example.com"]
        self.addresses = addresses
        self.check(addresses, ",\n ".join(addresses))
        
        # A single named address.
        addresses = ("Test User", "user@example.com")
        self.addresses = addresses
        self.check([addresses], "%s <%s>" % addresses)
        
        # A list of named addresses.
        addresses = [("Test User 1", "user1@example.com"), ("Test User 2", "user2@example.com")]
        self.addresses = addresses
        self.check(addresses, ",\n ".join(["%s <%s>" % i for i in addresses]))
        
        # Clean up our changes.
        del self.addresses
    
    def test_validation(self):
        self.addresses.append("user.name+test@example.com.info")
        self.addresses.append("eviluser@example.com\nTo: spammeduser@example.com\nFrom: spammeduser@example.com")
        self.assertEqual(str(self.addresses[-1]), "eviluser@example.com")
        
        # Double-at symbols automatically strip from the second at on.
        self.addresses.append("bad@user@example.com")
        self.assertEqual(str(self.addresses[-1]), "bad@user")
        
        self.assertRaises(ValueError, lambda: self.addresses.append("baduser.example.com"))
        self.assertRaises(ValueError, lambda: self.addresses.append("bad-u+s+er@example.com"))
        
        # Domains without TLDs are now supported.
        self.addresses.append("user@company")
        self.assertEqual(str(self.addresses[-1]), "user@company")
        
        del self.addresses