1 files changed, 15 insertions, 0 deletions
diff --git a/Mailman/tests/test_documentation.py b/Mailman/tests/test_documentation.py
index d8578bd05..c6ba0a1d6 100644
--- a/Mailman/tests/test_documentation.py
+++ b/Mailman/tests/test_documentation.py
@@ -22,8 +22,11 @@ import pdb
 import doctest
 import unittest
 
+from email import message_from_string
+
 import Mailman
 
+from Mailman.Message import Message
 from Mailman.app.styles import style_manager
 from Mailman.configuration import config
 from Mailman.database import flush
@@ -33,6 +36,17 @@ COMMASPACE = ', '
 
 
 
+def specialized_message_from_string(string):
+    return message_from_string(string, Message)
+
+
+def setup(testobj):
+    """Set up some things for convenience."""
+    testobj.globs['config'] = config
+    testobj.globs['message_from_string'] = specialized_message_from_string
+
+
+
 def cleaning_teardown(testobj):
     """Clear all persistent data at the end of a doctest."""
     # Clear the database of all rows.
@@ -69,6 +83,7 @@ def test_suite():
                 'docs/' + filename,
                 package=Mailman,
                 optionflags=flags,
+                setUp=setup,
                 tearDown=cleaning_teardown)
             suite.addTest(test)
     return suite