From e9802185b450a0d94d2771f6456ee33f64bf648c Mon Sep 17 00:00:00 2001 From: Kevin Kubasik Date: Wed, 22 Jul 2009 13:17:49 +0000 Subject: [PATCH] [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 --- django/test/rollback_importer.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 django/test/rollback_importer.py 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