Ticket #78: testmode_provider_78

diff -r 91840406d81b tests/test_debug_provider.py
--- /dev/null   Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test_debug_provider.py      Mon Jul 14 21:59:43 2008 +0200
@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+'''Test that the debug provider stores all "sent" mails.'''
+
+import unittest
+
+import turbomail
+from turbomail.control import interface
+from turbomail import Message
+from turbomail.managers.immediate import ImmediateManager
+from turbomail.providers.debug import DebugProviderFactory
+
+class TestDebugProviderStoresAllMail(unittest.TestCase):
+    
+    def setUp(self):
+        interface.config = {'mail.on': True}
+        manager = ImmediateManager()
+        interface.start(manager=manager, 
+                        provider=DebugProviderFactory())
+        self.msg = Message(sender='foo@example.com', recipient='to@example.com',
+                           subject='Test', plain='Foo')
+    
+    def tearDown(self):
+        interface.stop(force=True)
+        interface.config = {'mail.on': False}
+    
+    def test_fetch_sent_messages(self):
+        msg_string = str(self.msg)
+        turbomail.send(self.msg)
+        
+        stored_mails = interface.manager.provider.get_sent_mails()
+        self.assertEqual(1, len(stored_mails))
+        self.assertEqual(msg_string, str(stored_mails[0]))
+
diff -r 91840406d81b turbomail/control.py
--- a/turbomail/control.py      Mon Jul 14 21:59:34 2008 +0200
+++ b/turbomail/control.py      Mon Jul 14 21:59:43 2008 +0200
@@ -123,7 +123,6 @@
     def send(self, message):
         if not self.manager:
             raise MailNotEnabledError
-
         return self.manager.deliver(message)
         
 
diff -r 91840406d81b turbomail/providers/debug.py
--- a/turbomail/providers/debug.py      Mon Jul 14 21:59:34 2008 +0200
+++ b/turbomail/providers/debug.py      Mon Jul 14 21:59:43 2008 +0200
@@ -1,16 +1,14 @@
 # encoding: utf-8
-
 """TurboMail extension API."""
 
 __version__ = "$Revision: 84 $"
 
+import copy
 import logging
 log = logging.getLogger("turbomail.provider")
 deliverylog = logging.getLogger("turbomail.delivery")
 
-import turbomail
 from turbomail.api import ProviderFactory, Provider
-from turbomail.exceptions import ProviderExhaustedException
 
 __all__ = ['load']
 
@@ -23,6 +21,7 @@
     def __init__(self):
         super(DebugProvider, self).__init__()
         log.debug("Debug provider setup.")
+        self._sent_mails = []
     
     def __del__(self):
         if log: log.debug("Debug provider teardown.")
@@ -30,16 +29,20 @@
     def deliver(self, message):
         log.info("Attempting delivery of message %s." % message.id)
         deliverylog.info("%s DELIVER" % message.id)
-        
-        for i in str(message).split("\n"):
+        msg_string = str(message)
+        self._sent_mails.append(msg_string)
+        for i in msg_string.split("\n"):
             deliverylog.debug("%s BODY %s" % (message.id, str(i)))
         
         deliverylog.info("%s SENT" % message.id)
         return True
+    
+    def get_sent_mails(self):
+        return copy.copy(self._sent_mails)
 
 
 class DebugProviderFactory(ProviderFactory):
     name = "Debug"
     version = "1.0"
     url = "http://www.python-turbomail.org/wiki/DebugProvider"
-    provider = DebugProvider
\ No newline at end of file
+    provider = DebugProvider