Add compatibility with Py2 back.

zopefoundation · Mar 15, 2013 · 1f9ba57 · 1f9ba57
1 parent ec9bafb
commit 1f9ba57
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 9 deletions.
diff --git a/src/ZEO/scripts/tests.py b/src/ZEO/scripts/tests.py
@@ -11,6 +11,7 @@
 # FOR A PARTICULAR PURPOSE.
 #
 ##############################################################################
+from __future__ import print_function
 import doctest, re, unittest
 from zope.testing import renormalizing
 
@@ -21,7 +22,8 @@ def test_suite():
             checker=renormalizing.RENormalizing([
                 (re.compile('usage: Usage: '), 'Usage: '), # Py 2.4
                 (re.compile('options:'), 'Options:'), # Py 2.4
-                ])
+                ]),
+            globs={'print_function': print_function},
             ),
         ))
 
diff --git a/src/ZEO/tests/testZEO.py b/src/ZEO/tests/testZEO.py
@@ -20,6 +20,7 @@
 from ZEO.tests import forker, Cache, CommitLockTests, ThreadTests
 from ZEO.tests import IterationTests
 from ZEO.zrpc.error import DisconnectedError
+from ZEO._compat import PY3
 from ZODB.tests import StorageTestBase, BasicStorage,  \
      TransactionalUndoStorage,  \
      PackableStorage, Synchronization, ConflictResolution, RevisionStorage, \
@@ -1756,15 +1757,18 @@ def test_suite():
     # unit test layer
     zeo = unittest.TestSuite()
     zeo.addTest(unittest.makeSuite(ZODB.tests.util.AAAA_Test_Runner_Hack))
+    patterns = [
+        (re.compile(r"'start': '[^\n]+'"), 'start'),
+        (re.compile("ZODB.POSException.ConflictError"), "ConflictError"),
+        (re.compile("ZODB.POSException.POSKeyError"), "POSKeyError"),
+        (re.compile("ZEO.Exceptions.ClientStorageError"), "ClientStorageError"),
+        ]
+    if not PY3:
+        patterns.append((re.compile("^'(blob[^']*)'"), r"b'\1'"))
+        patterns.append((re.compile("^'Z308'"), "b'Z308'"))
     zeo.addTest(doctest.DocTestSuite(
         setUp=forker.setUp, tearDown=zope.testing.setupstack.tearDown,
-        checker=renormalizing.RENormalizing([
-            (re.compile(r"'start': '[^\n]+'"), 'start'),
-            (re.compile("ZODB.POSException.ConflictError"), "ConflictError"),
-            (re.compile("ZODB.POSException.POSKeyError"), "POSKeyError"),
-            (re.compile("ZEO.Exceptions.ClientStorageError"),
-             "ClientStorageError"),
-            ]),
+        checker=renormalizing.RENormalizing(patterns),
         ))
     zeo.addTest(doctest.DocTestSuite(
             ZEO.tests.IterationTests,
@@ -1774,14 +1778,17 @@ def test_suite():
                      "ClientDisconnected"),
                     )),
             ))
-    zeo.addTest(doctest.DocFileSuite('registerDB.test'))
+    zeo.addTest(doctest.DocFileSuite(
+            'registerDB.test', globs={'print_function': print_function}))
     zeo.addTest(
         doctest.DocFileSuite(
             'zeo-fan-out.test', 'zdoptions.test',
             'drop_cache_rather_than_verify.txt', 'client-config.test',
             'protocols.test', 'zeo_blob_cache.test', 'invalidation-age.txt',
             'dynamic_server_ports.test', 'new_addr.test',
             setUp=forker.setUp, tearDown=zope.testing.setupstack.tearDown,
+            checker=renormalizing.RENormalizing(patterns),
+            globs={'print_function': print_function},
             ),
         )
     zeo.addTest(PackableStorage.IExternalGC_suite(

diff --git a/src/ZEO/tests/testZEO2.py b/src/ZEO/tests/testZEO2.py
@@ -11,6 +11,7 @@
 # FOR A PARTICULAR PURPOSE.
 #
 ##############################################################################
+from __future__ import print_function
 from zope.testing import setupstack, renormalizing
 import doctest
 import logging