root / trunk / tests / test_tm2_compatibility.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
'''Test that most interfaces of TurboMail 2.x are working in TurboMail 3.x'''

import unittest
import warnings

import turbomail
from turbomail import Message, MailNotEnabledException
from turbomail.managers.immediate import ImmediateManager
from turbomail.transports.debug import DebugTransportFactory

class TestTurboMail2xCompatibility(unittest.TestCase):
    
    def setUp(self):
        warnings.filterwarnings('ignore', 'Property ".+" is deprecated.*', 
                                category=DeprecationWarning)
        warnings.filterwarnings('ignore', 'Please specify the author using.*', 
                                category=DeprecationWarning)
    
    def tearDown(self):
        turbomail.control.interface.stop(force=True)
        turbomail.control.interface.config = {'mail.on': False}
        warnings.resetwarnings()
    
    def test_message_instantiation_with_positional_parameters(self):
        message = Message('from@example.com', 'to@example.com', 'Test')
        message.plain = 'Hello world!'
        msg_string = str(message)
        self.failUnless('From: from@example.com' in msg_string)
        self.failUnless('To: to@example.com' in msg_string)
        self.failUnless('Subject: Test' in msg_string)
    
    def test_message_instantiation_with_keyword_parameters(self):
        message = Message(sender='from@example.com', recipient='to@example.com',
                          subject='Test')
        message.plain = 'Hello world!'
        msg_string = str(message)
        self.failUnless('From: from@example.com' in msg_string)
        self.failUnless('To: to@example.com' in msg_string)
        self.failUnless('Subject: Test' in msg_string)
    
    def test_message_instantiation_with_tuples(self):
        message = Message(('Foo Bar', 'from@example.com'), 
                          ('To', 'to@example.com'), 'Test')
        message.plain = 'Hello world!'
        msg_string = str(message)
        self.failUnless('From: Foo Bar <from@example.com>' in msg_string)
        self.failUnless('To: To <to@example.com>' in msg_string)
        self.failUnless('Subject: Test' in msg_string)
    
    def test_message_enqueue_not_enabled(self):
        message = Message('from@example.com', 'to@example.com', 'Test')
        message.plain = 'Hello world!'
        self.assertRaises(MailNotEnabledException, turbomail.enqueue, message)
    
    def test_message_enqueue(self):
        turbomail.control.interface.config = {'mail.on': True}
        fake_setuptools =  {'immediate': ImmediateManager,
                            'debug': DebugTransportFactory}
        turbomail.control.interface.start(extra_classes=fake_setuptools)
        
        message = Message('from@example.com', 'to@example.com', 'Test')
        message.plain = 'Hello world!'
        warnings.filterwarnings('ignore', '.*enqueue.*', category=DeprecationWarning)
        turbomail.enqueue(message)
    
    def test_message_plain_text_as_callable(self):
        message = Message('from@example.com', 'to@example.com', 'Test')
        message.plain = lambda: 'Hello world!'
        msg_string = str(message)
        self.failUnless('Hello world!' in msg_string)
    
    def test_message_properties(self):
        message = Message('from@example.com', 'to@example.com', 'Test')
        message.plain = 'Hello world!'
        
        property_names = ['bcc', 'cc', 'date', 'disposition', 'encoding',
                          'headers', 'organization', 'plain', 'priority',
                          'recipient', 'replyto', 'rich', 'sender', 'smtpfrom',
                          'subject']
        for name in property_names:
            getattr(message, name)
        
        args = dict(zip(property_names, ['foo@example.com'] * len(property_names)))
        Message(**args)
    
    def test_use_deprecated_smtp_from(self):
        message = Message(smtpfrom='smtpfrom@example.com')
        self.assertEqual('smtpfrom@example.com', str(message.smtp_from))
        self.assertEqual(message.smtpfrom, message.smtp_from)
        message.smtpfrom = 'smtp_from@example.com'
        self.assertEqual('smtp_from@example.com', str(message.smtp_from))
        self.assertEqual(message.smtpfrom, message.smtp_from)
    
    def test_use_deprecated_reply_to(self):
        message = Message(replyto='replyto@example.com')
        self.assertEqual('replyto@example.com', str(message.reply_to))
        self.assertEqual(message.replyto, message.reply_to)
        message.replyto = 'reply_to@example.com'
        self.assertEqual('reply_to@example.com', str(message.reply_to))
        self.assertEqual(message.replyto, message.reply_to)
    
    def test_use_deprecated_recipient(self):
        message = Message(recipient='recipient@example.com')
        self.assertEqual('recipient@example.com', str(message.to))
        self.assertEqual(message.recipient, message.to)
        message.recipient = 'to@example.com'
        self.assertEqual('to@example.com', str(message.to))
        self.assertEqual(message.recipient, message.to)
    
    def test_use_sender_for_author_if_no_author_given(self):
        message = Message(sender='sender@example.com', to='to@example.com', 
                          subject='Test')
        self.assertEqual('sender@example.com', str(message.sender))
        message.plain = 'Hello world!'
        msg_string = str(message)
        self.failUnless('From: sender@example.com' in msg_string)
        self.failUnless('To: to@example.com' in msg_string)
        self.failUnless('Subject: Test' in msg_string)
        self.failIf('Sender: ' in msg_string)
        self.assertEqual('sender@example.com', str(message.sender))

# TODO: Check that all items do the right thing!