Set and restore the site for functional tests as well

lib/python/Testing/ZopeTestCase/functional.py

@@ -37,6 +37,7 @@ class Functional(sandbox.Sandboxed):
                 request_method='GET', stdin=None, handle_errors=True):
         '''Publishes the object at 'path' returning a response object.'''
 
+        from zope.app.component.hooks import setSite, getSite
         from StringIO import StringIO
         from ZPublisher.Response import Response
         from ZPublisher.Test import publish_module
@@ -47,6 +48,10 @@ class Functional(sandbox.Sandboxed):
         # Save current security manager
         sm = getSecurityManager()
 
+        # And we need to store the old site
+        old_site = getSite()
+        setSite(None)
+
         # Commit the sandbox for good measure
         transaction.commit()
 
@@ -89,6 +94,9 @@ class Functional(sandbox.Sandboxed):
         # Restore security manager
         setSecurityManager(sm)
 
+        # And we need to restore the site again
+        setSite(old_site)
+
         return ResponseWrapper(response, outstream, path)
 
 

lib/python/Testing/ZopeTestCase/zopedoctest/functional.py

@@ -117,6 +117,7 @@ def http(request_string, handle_errors=True):
     """
     import urllib
     import rfc822
+    from zope.app.component.hooks import setSite, getSite
     from cStringIO import StringIO
     from ZPublisher.Response import Response
     from ZPublisher.Test import publish_module
@@ -126,6 +127,10 @@ def http(request_string, handle_errors=True):
     # Save current Security Manager
     old_sm = getSecurityManager()
 
+    # And we need to store the old site
+    old_site = getSite()
+    setSite(None)
+
     # Commit work done by previous python code.
     transaction.commit()
 
@@ -193,7 +198,10 @@ def http(request_string, handle_errors=True):
     # by calling the publish method above
     setSecurityManager(old_sm)
 
+    # And we need to restore the site again
+    setSite(old_site)
     # Sync connection
+
     sync()
 
     return DocResponseWrapper(response, outstream, path, header_output)