Ticket #75: custom_providers_with_setuptools_75.2

diff -r 87ae6cd83bb3 tests/test_custom_provider_injection.py
--- /dev/null   Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test_custom_provider_injection.py   Mon Jul 14 21:59:31 2008 +0200
@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+'''Test the injection of custom TurboMail providers without setuptools.'''
+
+import logging
+import unittest
+
+from turbomail.api import Manager, Provider
+from turbomail.control import interface
+from turbomail.message import Message
+
+logging.disable(logging.WARNING)
+
+
+class DummyProvider(Provider):
+    def deliver(self, message):
+        return True
+
+
+class DummyManager(Manager):
+    def deliver(self, message):
+        return True
+
+
+class TestCustomProviderInjection(unittest.TestCase):
+    """Test the injection of custom TurboMail providers without setuptools."""
+    
+    def setUp(self):
+        self.manager = DummyManager()
+        self.provider = DummyProvider()
+        self.message = Message(
+                from_=("Author", "author@example.com"),
+                to=("Recipient", "recipient@example.com"),
+                subject="Test message subject.",
+                plain="This is a test message plain text body."
+            )
+        interface.config = {'mail.on': True}
+    
+    def tearDown(self):
+        interface.stop(force=True)
+        interface.config = {'mail.on': False}
+    
+    def test_inject_custom_manager_and_provider(self):
+        interface.start(manager=self.manager, provider=self.provider)
+        self.assertEqual(self.manager, interface.manager)
+        self.assertEqual(self.provider, interface.provider)
+        interface.send(self.message)
+
+
diff -r 87ae6cd83bb3 turbomail/control.py
--- a/turbomail/control.py      Sun Jul 13 19:37:04 2008 +0200
+++ b/turbomail/control.py      Mon Jul 14 21:59:31 2008 +0200
@@ -50,29 +50,33 @@
             return entrypoint.load()
         return None
     
-    def start(self):
+    def start(self, manager=None, provider=None):
         if not self.config.get("mail.on", False):
             return
         
         log.info("TurboMail extension starting up.")
         
-        def load(t, default):
-            extension = self.config.get("mail.%s" % t, default)
-            controller = self.__load_single_entry("turbomail.%ss" % t, extension)
-            if not controller:
-                self.config.update({"mail.on": False})
-                log.error("Unable to locate %s %s, TurboMail disabled." % (extension, t))
-                self.stop(force=True)
-                return
-            setattr(self, t, controller)
+        def load(t, default, custom_controller):
+            if custom_controller != None:
+                setattr(self, t, custom_controller)
+            else:
+                extension = self.config.get("mail.%s" % t, default)
+                entry_point = "turbomail.%ss" % t
+                controller = self.__load_single_entry(entry_point, extension)
+                if not controller:
+                    self.config.update({"mail.on": False})
+                    log.error("Unable to locate %s %s, TurboMail disabled." % (extension, t))
+                    self.stop(force=True)
+                    return
+                setattr(self, t, controller)
             if hasattr(getattr(self, t), 'load'):
                 setattr(self, t, getattr(self, t).load())
             if hasattr(getattr(self, t), 'start'):
                 getattr(self, t).start()
         
         # Load the requested manager and provider.
-        load('manager', 'immediate')
-        load('provider', 'debug')
+        load('manager', 'immediate', manager)
+        load('provider', 'debug', provider)
         
         # Load and initialize the various extensions.
         extensions = pkg_resources.iter_entry_points("turbomail.extensions")