[gsoc2009-testing] Adding my global importer which can force a reload of all modules. (so coverage can account for startup as well)

git-svn-id: http://code.djangoproject.com/svn/django/branches/soc2009/test-improvements@11296 bcc190cf-cafb-0310-a4f2-bffc1f526a37

1 files changed, 19 insertions, 0 deletions

diff --git a/django/test/rollback_importer.py b/django/test/rollback_importer.py
new file mode 100644
index 0000000000..514667979e
--- /dev/null
+++ b/django/test/rollback_importer.py
@@ -0,0 +1,19 @@
+class RollbackImporter:
+    def __init__(self):
+        "Creates an instance and installs as the global importer"
+        self.previousModules = sys.modules.copy()
+        self.realImport = __builtin__.__import__
+        __builtin__.__import__ = self._import
+        self.newModules = {}
+
+    def _import(self, name, globals=None, locals=None, fromlist=[]):
+        result = apply(self.realImport, (name, globals, locals, fromlist))
+        self.newModules[name] = 1
+        return result
+        
+    def uninstall(self):
+        for modname in self.newModules.keys():
+            if not self.previousModules.has_key(modname):
+                # Force reload when modname next imported
+                del(sys.modules[modname])
+        __builtin__.__import__ = self.realImport