From 2410fe8c2578fbd11275cfc7fc1897173eecd41a Mon Sep 17 00:00:00 2001
From: Barry Warsaw
Date: Sun, 8 Apr 2012 10:15:29 -0600
Subject:  - Refactor the way databases are schema-migrated so that
 load_migrations()    can be tested separately.  - Add an `until` argument to
 load_migrations() so that we can load only up to    a given timestamp.  - In
 load_migrations(), ignore files in which the `version` part of the file   
 name is empty.  - In migrations.rst, use the new, better way of ensuring
 post-test cleanup.  - Add tests for partial upgrades.  - LP: #971013 - schema
 migrations for beta 2.  - LP: #967238 - IMailingList.archive +
 IMailingList.archive_private ->    IMailingList.archive_policy, and add
 ArchivePolicy enum.  - Move the `chdir` context manager to helpers.py and add
 `temporary_db`    context manager.

---
 src/mailman/testing/helpers.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

(limited to 'src/mailman/testing/helpers.py')

diff --git a/src/mailman/testing/helpers.py b/src/mailman/testing/helpers.py
index 3648a6710..d9885cbac 100644
--- a/src/mailman/testing/helpers.py
+++ b/src/mailman/testing/helpers.py
@@ -25,6 +25,7 @@ __all__ = [
     'TestableMaster',
     'body_line_iterator',
     'call_api',
+    'chdir',
     'configuration',
     'digest_mbox',
     'event_subscribers',
@@ -35,6 +36,7 @@ __all__ = [
     'reset_the_world',
     'specialized_message_from_string',
     'subscribe',
+    'temporary_db',
     'wait_for_webservice',
     ]
 
@@ -391,6 +393,34 @@ class configuration:
         return wrapper
 
 
+
+@contextmanager
+def temporary_db(db):
+    real_db = config.db
+    config.db = db
+    try:
+        yield
+    finally:
+        config.db = real_db
+
+
+
+class chdir:
+    """A context manager for temporary directory changing."""
+    def __init__(self, directory):
+        self._curdir = None
+        self._directory = directory
+
+    def __enter__(self):
+        self._curdir = os.getcwd()
+        os.chdir(self._directory)
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        os.chdir(self._curdir)
+        # Don't suppress exceptions.
+        return False
+
+
 
 def subscribe(mlist, first_name, role=MemberRole.member):
     """Helper for subscribing a sample person to a mailing list."""
-- 
cgit v1.2.3-70-g09d2