Testing that nested registry's and using objects within nested stacks works (it does)

1 files changed, 32 insertions, 0 deletions

diff --git a/tests/test_registry.py b/tests/test_registry.py
index ab89d76..87966dd 100644
--- a/tests/test_registry.py
+++ b/tests/test_registry.py
@@ -22,6 +22,21 @@ class RegistryUsingApp(object):
         start_response(status, response_headers)
         return ['Hello world!\nThe variable is %s' % str(testobj)]
 
+class RegistryMiddleMan(object):
+    def __init__(self, app, var, value):
+        self.app = app
+        self.var = var
+        self.value = value
+    
+    def __call__(self, environ, start_response):
+        if environ.has_key('paste.registry'):
+            environ['paste.registry'].register(self.var, self.value)
+        app_response = ['Inserted by middleware!\nInsertValue is %s' % str(testobj)]
+        app_response.extend(self.app(environ, start_response))
+        app_response.extend(['\nAppended by middleware!\nAppendValue is %s' % str(testobj)])
+        return app_response
+            
+
 def test_simple():
     app = TestApp(simpleapp)
     response = app.get('/')
@@ -37,4 +52,21 @@ def test_solo_registry():
     assert 'The variable is' in res
     assert "{'hi': 'people'}" in res
     
+def test_double_registry():
+    obj = {'hi':'people'}
+    secondobj = {'bye':'friends'}
+    wsgiapp = RegistryUsingApp(testobj, obj)
+    wsgiapp = RegistryManager(wsgiapp)
+    wsgiapp = RegistryMiddleMan(wsgiapp, testobj, secondobj)
+    wsgiapp = RegistryManager(wsgiapp)
+    app = TestApp(wsgiapp)
+    res = app.get('/')
+    assert 'Hello world' in res
+    assert 'The variable is' in res
+    assert "{'hi': 'people'}" in res
+    assert "Inserted by middleware" in res
+    assert "Appended by middleware" in res
+    assert "InsertValue is {'bye': 'friends'}" in res
+    assert "AppendValue is {'bye': 'friends'}" in res
+
     
\ No newline at end of file