Changeset 104 for trunk

Timestamp:

10/04/08 11:17:12 (3 months ago)

Author:

fs

Message:

debug provider now stores all mails so you can check then in unit tests (#78)

Location:

trunk

Files:

• tests/test_debug_provider.py (added)
• turbomail/providers/debug.py (modified) (3 diffs)

trunk/turbomail/providers/debug.py

@@ -1 +1 @@
 # encoding: utf-8
-
-"""TurboMail extension API."""
 
 __version__ = "$Revision$"
 
+import copy
 import logging
+
+from turbomail.api import ProviderFactory, Provider
+
+
 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']
@@ -24 +23 @@
         super(DebugProvider, self).__init__()
         log.debug("Debug provider setup.")
+        self._sent_mails = []
     
     def __del__(self):
-        if log: log.debug("Debug provider teardown.")
+        if log:
+            log.debug("Debug provider teardown.")
     
     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)
 
 
@@ -44 +49 @@
     url = "http://www.python-turbomail.org/wiki/DebugProvider"
     provider = DebugProvider
+