From e9d2284a077e6ca4fe3d88dced75d771465ca931 Mon Sep 17 00:00:00 2001
From: Nikolaus Schulz <microschulz@web.de>
Date: Tue, 30 Sep 2008 12:52:43 +0200
Subject: [PATCH] test suite: let make_message() optionally return a
 rfc822.Message

---
 test_archivemail.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/test_archivemail.py b/test_archivemail.py
index e152bc5..1fefff3 100755
--- a/test_archivemail.py
+++ b/test_archivemail.py
@@ -59,6 +59,8 @@ import tempfile
 import time
 import unittest
 import gzip
+import cStringIO
+import rfc822
 
 try:
     import archivemail
@@ -1197,7 +1199,7 @@ class TestArchiveMboxMode(TestCaseInTempdir):
 
 ########## helper routines ############
 
-def make_message(body=None, default_headers={}, hours_old=None):
+def make_message(body=None, default_headers={}, hours_old=None, wantobj=False):
     headers = copy.copy(default_headers)
     if not headers:
         headers = {}
@@ -1223,8 +1225,10 @@ def make_message(body=None, default_headers={}, hours_old=None):
         if headers[key] is not None:
             msg = msg + ("%s: %s\n" % (key, headers[key]))
     msg = msg + "\n\n" + body + "\n\n"
-    return msg
-
+    if not wantobj:
+        return msg
+    fp = cStringIO.StringIO(msg)
+    return rfc822.Message(fp)
 
 def append_file(source, dest):
     """appends the file named 'source' to the file named 'dest'"""