- Make the (internal) thread local registry manager maintain a stack

  of registries in order to make it possible to call one BFG
  application from inside another.

1 files changed, 19 insertions, 8 deletions

diff --git a/repoze/bfg/tests/test_registry.py b/repoze/bfg/tests/test_registry.py
index 0d051c131..51883b196 100644
--- a/repoze/bfg/tests/test_registry.py
+++ b/repoze/bfg/tests/test_registry.py
@@ -49,16 +49,28 @@ class TestThreadLocalRegistryManager(unittest.TestCase, PlacelessSetup):
     def test_init(self):
         local = self._makeOne()
         from zope.component import getGlobalSiteManager
-        self.assertEqual(local.registry, getGlobalSiteManager())
+        self.assertEqual(local.stack, [])
+        self.assertEqual(local.get(), getGlobalSiteManager())
+
+    def test_push_and_pop(self):
+        local = self._makeOne()
+        from zope.component import getGlobalSiteManager
+        local.push(True)
+        self.assertEqual(local.get(), True)
+        self.assertEqual(local.pop(), True)
+        self.assertEqual(local.pop(), None)
+        self.assertEqual(local.get(), getGlobalSiteManager())
 
     def test_set_get_and_clear(self):
         local = self._makeOne()
         from zope.component import getGlobalSiteManager
         local.set(None)
-        self.failIfEqual(local.registry, getGlobalSiteManager())
+        self.assertEqual(local.stack, [None])
         self.assertEqual(local.get(), None)
         local.clear()
-        self.assertEqual(local.registry, getGlobalSiteManager())
+        self.assertEqual(local.get(), getGlobalSiteManager())
+        local.clear()
+        self.assertEqual(local.get(), getGlobalSiteManager())
 
 class GetSiteManagerTests(unittest.TestCase):
     def _getFUT(self):
@@ -76,16 +88,15 @@ class GetSiteManagerTests(unittest.TestCase):
         self.assertRaises(ComponentLookupError, gsm, object)
         
 class DummyRegistrationManager:
-    registry = None
-    def set(self, registry):
+    def push(self, registry):
         self.registry = registry
 
+    def pop(self):
+        self.popped = True
+
     def get(self):
         return self.registry
 
-    def clear(self):
-        self.cleared = True
-
 class DummyLock:
     def acquire(self):
         self.acquired = True