root / trunk / tests / test_custom_transport_injection.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
'''Test the injection of custom TurboMail transports without setuptools.'''

import logging
import unittest

from turbomail.api import Manager, Transport
from turbomail.control import interface
from turbomail.message import Message

logging.disable(logging.WARNING)


class DummyTransport(Transport):
    def deliver(self, message):
        return True


class DummyManager(Manager):
    def deliver(self, message):
        return True


class TestCustomTransportInjection(unittest.TestCase):
    """Test the injection of custom TurboMail transports without setuptools."""
    
    def setUp(self):
        self.manager = DummyManager()
        self.transport = DummyTransport()
        self.message = Message(author=("Author", "author@example.com"),
                               to=("Recipient", "recipient@example.com"),
                               subject="Test message subject.",
                               plain="This is a test message.")
        interface.config = {'mail.on': True}
    
    def tearDown(self):
        interface.stop(force=True)
        interface.config = {'mail.on': False}
    
    def test_provide_dict_with_additional_managers_and_transports_which_overrides_setuptools(self):
        interface.config.update({'mail.manager': 'immediate',
                                 'mail.transport': 'foo',})
        interface.start(extra_classes=dict(immediate=self.manager, foo=self.transport))
        self.assertEqual(self.manager, interface.manager)
        self.assertEqual(self.transport, interface.transport)
        interface.send(self.message)
    
    def test_dont_fail_if_no_extra_classes_were_provided(self):
        interface.start()