A way to track all imported modules.

django_pyvows/context.py

@@ -31,7 +31,7 @@ class DjangoContext(Vows.Context):
 
     def __init__(self, parent):
         super(DjangoContext, self).__init__(parent)
-
+        self.settings = {}
         self.ignore('get_settings', 'template', 'request', 'model', 'url',
                 'start_environment', 'port', 'host', 'get_url', 'get', 'post')
 

django_pyvows/settings_manager.py

@@ -0,0 +1,33 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# django-pyvows extensions
+# https://github.com/rafaelcaricio/django-pyvows
+
+# Licensed under the MIT license:
+# http://www.opensource.org/licenses/mit-license
+# Copyright (c) 2011 Rafael Caricio rafael@caricio.com
+
+import sys
+
+class ModulesTracker(object):
+
+    def install(self):
+        self.previous_modules = sys.modules.copy()
+        self.real_import = __builtins__['__import__']
+        __builtins__['__import__'] = self._import
+        self.new_modules = {}
+
+    def _import(self, name, globals=None, locals=None, fromlist=[], level=-1):
+        result = apply(self.real_import, (name, globals, locals, fromlist, level))
+        self.new_modules[name] = 1
+        return result
+
+    def reload(self):
+        for modname in self.new_modules.keys():
+            if not self.previous_modules.has_key(modname):
+                del(sys.modules[modname])
+
+    def uninstall(self):
+        __builtins__['__import__'] = self.real_import
+

vows/settings_vows.py

@@ -0,0 +1,78 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# django-pyvows extensions
+# https://github.com/rafaelcaricio/django-pyvows
+
+# Licensed under the MIT license:
+# http://www.opensource.org/licenses/mit-license
+# Copyright (c) 2011 Rafael Caricio rafael@caricio.com
+
+from pyvows import Vows, expect
+
+from django_pyvows.context import DjangoContext, DjangoHTTPContext
+from django_pyvows.settings_manager import ModulesTracker
+
+
+@Vows.batch
+class SettingsVows(DjangoContext):
+
+    class WhenIUseTheModulesTracker(DjangoContext):
+
+        def topic(self):
+            modules_tracker = ModulesTracker()
+            modules_tracker.install()
+            import to_test
+            return modules_tracker
+
+        def should_track_the_new_imported_module(self, topic):
+            expect('to_test' in topic.new_modules).to_be_true()
+
+        class WhenIReloadTheImportedModule(DjangoContext):
+
+            def topic(self, modules_tracker):
+                import to_test
+                imported_first_time_at = to_test.imported_at
+                modules_tracker.reload()
+                import to_test
+                return (modules_tracker, imported_first_time_at, to_test.imported_at)
+
+            def should_be_different_instances(self, topic):
+                expect(topic[1]).not_to_equal(topic[2])
+
+            class WhenIUninstall(DjangoContext):
+
+                def topic(self, last_topic):
+                    modules_tracker = last_topic[0]
+                    modules_tracker.uninstall()
+                    return __builtins__['__import__']
+
+                def should_be_restored_to_original_import_function(self, topic):
+                    expect(topic.__module__).to_equal('__builtin__')
+
+    class CannotSayHelloWithoutName(DjangoHTTPContext):
+
+        def topic(self):
+            self.settings['SAY_HELLO_WITHOUT_NAME'] = False
+            self.start_server(port=8002)
+            return self.get('/say/')
+
+        def should_be_ok(self, topic):
+            expect(topic.code).to_equal(200)
+
+        def should_ask_for_my_name(self, topic):
+            expect(topic.read()).to_equal("What's your name?")
+
+    class SayHelloWithoutName(DjangoHTTPContext):
+
+        def topic(self):
+            self.settings['SAY_HELLO_WITHOUT_NAME'] = True
+            self.start_server(port=8003)
+            return self.get('/say/')
+
+        def should_be_ok(self, topic):
+            expect(topic.code).to_equal(200)
+
+        def should_(self, topic):
+            expect(topic.read()).to_equal("Hello, guess!")
+

vows/to_test/__init__.py

@@ -0,0 +1,2 @@
+from datetime import datetime
+imported_at = datetime.now()