Mock asyncio in mockiavellian tests

src/ZEO/tests/testConfig.py

@@ -20,7 +20,7 @@
 
 from .forker import start_zeo_server
 
-class ZEOConfigTest(setupstack.TestCase):
+class ZEOConfigTestBase(setupstack.TestCase):
 
     setUp = setupstack.setUpDirectory
 
@@ -86,6 +86,8 @@ def _client_assertions(
         self.assertEqual(client.__name__,
                          name if name is not None else str(client._addr))
 
+class ZEOConfigTest(ZEOConfigTestBase):
+
     def test_default_zeo_config(self, **client_settings):
         addr, stop = self.start_server()
 

src/ZEO/tests/testssl.py

@@ -1,3 +1,5 @@
+from .._compat import PY3
+
 import mock
 import os
 import ssl
@@ -7,7 +9,7 @@
 from ..Exceptions import ClientDisconnected
 from .. import runzeo
 
-from .testConfig import ZEOConfigTest
+from .testConfig import ZEOConfigTestBase
 
 here = os.path.dirname(__file__)
 server_cert = os.path.join(here, 'server.pem')
@@ -17,7 +19,7 @@
 client_cert = os.path.join(here, 'client.pem')
 client_key  = os.path.join(here, 'client_key.pem')
 
-class SSLConfigTest(ZEOConfigTest):
+class SSLConfigTest(ZEOConfigTestBase):
 
     def test_ssl_basic(self):
         # This shows that configuring ssl has an actual effect on connections.
@@ -112,8 +114,13 @@ def test_ssl_pw(self):
             )
         stop()
 
+@mock.patch(('asyncio' if PY3 else 'trollius') + '.async')
+@mock.patch(('asyncio' if PY3 else 'trollius') + '.set_event_loop')
+@mock.patch(('asyncio' if PY3 else 'trollius') + '.new_event_loop')
+class SSLConfigTestMockiavellian(ZEOConfigTestBase):
+
     @mock.patch('ssl.create_default_context')
-    def test_ssl_mockiavellian_server_no_ssl(self, factory):
+    def test_ssl_mockiavellian_server_no_ssl(self, factory, *_):
         server = create_server()
         self.assertFalse(factory.called)
         self.assertEqual(server.acceptor.ssl_context, None)
@@ -134,29 +141,29 @@ def assert_context(
         self.assertEqual(context.check_hostname, check_hostname)
 
     @mock.patch('ssl.create_default_context')
-    def test_ssl_mockiavellian_server_ssl_no_auth(self, factory):
+    def test_ssl_mockiavellian_server_ssl_no_auth(self, factory, *_):
         with self.assertRaises(SystemExit):
             # auth is required
             create_server(certificate=server_cert, key=server_key)
 
     @mock.patch('ssl.create_default_context')
-    def test_ssl_mockiavellian_server_ssl_auth_file(self, factory):
+    def test_ssl_mockiavellian_server_ssl_auth_file(self, factory, *_):
         server = create_server(
             certificate=server_cert, key=server_key, authenticate=__file__)
         context = server.acceptor.ssl_context
         self.assert_context(factory, context, cafile=__file__)
         server.close()
 
     @mock.patch('ssl.create_default_context')
-    def test_ssl_mockiavellian_server_ssl_auth_dir(self, factory):
+    def test_ssl_mockiavellian_server_ssl_auth_dir(self, factory, *_):
         server = create_server(
             certificate=server_cert, key=server_key, authenticate=here)
         context = server.acceptor.ssl_context
         self.assert_context(factory, context, capath=here)
         server.close()
 
     @mock.patch('ssl.create_default_context')
-    def test_ssl_mockiavellian_server_ssl_pw(self, factory):
+    def test_ssl_mockiavellian_server_ssl_pw(self, factory, *_):
         server = create_server(
             certificate=server_cert,
             key=server_key,
@@ -170,15 +177,15 @@ def test_ssl_mockiavellian_server_ssl_pw(self, factory):
 
     @mock.patch('ssl.create_default_context')
     @mock.patch('ZEO.ClientStorage.ClientStorage')
-    def test_ssl_mockiavellian_client_no_ssl(self, ClientStorage, factory):
+    def test_ssl_mockiavellian_client_no_ssl(self, ClientStorage, factory, *_):
         client = ssl_client()
         self.assertFalse('ssl' in ClientStorage.call_args[1])
         self.assertFalse('ssl_server_hostname' in ClientStorage.call_args[1])
 
     @mock.patch('ssl.create_default_context')
     @mock.patch('ZEO.ClientStorage.ClientStorage')
     def test_ssl_mockiavellian_client_server_signed(
-        self, ClientStorage, factory
+        self, ClientStorage, factory, *_
         ):
         client = ssl_client(certificate=client_cert, key=client_key)
         context = ClientStorage.call_args[1]['ssl']
@@ -191,7 +198,7 @@ def test_ssl_mockiavellian_client_server_signed(
     @mock.patch('ssl.create_default_context')
     @mock.patch('ZEO.ClientStorage.ClientStorage')
     def test_ssl_mockiavellian_client_auth_dir(
-        self, ClientStorage, factory
+        self, ClientStorage, factory, *_
         ):
         client = ssl_client(
             certificate=client_cert, key=client_key, authenticate=here)
@@ -207,7 +214,7 @@ def test_ssl_mockiavellian_client_auth_dir(
     @mock.patch('ssl.create_default_context')
     @mock.patch('ZEO.ClientStorage.ClientStorage')
     def test_ssl_mockiavellian_client_auth_file(
-        self, ClientStorage, factory
+        self, ClientStorage, factory, *_
         ):
         client = ssl_client(
             certificate=client_cert, key=client_key, authenticate=server_cert)
@@ -223,7 +230,7 @@ def test_ssl_mockiavellian_client_auth_file(
     @mock.patch('ssl.create_default_context')
     @mock.patch('ZEO.ClientStorage.ClientStorage')
     def test_ssl_mockiavellian_client_pw(
-        self, ClientStorage, factory
+        self, ClientStorage, factory, *_
         ):
         client = ssl_client(
             certificate=client_cert, key=client_key,
@@ -241,7 +248,7 @@ def test_ssl_mockiavellian_client_pw(
     @mock.patch('ssl.create_default_context')
     @mock.patch('ZEO.ClientStorage.ClientStorage')
     def test_ssl_mockiavellian_client_server_hostname(
-        self, ClientStorage, factory
+        self, ClientStorage, factory, *_
         ):
         client = ssl_client(
             certificate=client_cert, key=client_key, authenticate=server_cert,
@@ -258,7 +265,7 @@ def test_ssl_mockiavellian_client_server_hostname(
     @mock.patch('ssl.create_default_context')
     @mock.patch('ZEO.ClientStorage.ClientStorage')
     def test_ssl_mockiavellian_client_check_hostname(
-        self, ClientStorage, factory
+        self, ClientStorage, factory, *_
         ):
         client = ssl_client(
             certificate=client_cert, key=client_key, authenticate=server_cert,
@@ -320,7 +327,10 @@ def create_server(**ssl_settings):
 
 
 def test_suite():
-    return unittest.makeSuite(SSLConfigTest)
+    return unittest.TestSuite((
+        unittest.makeSuite(SSLConfigTest),
+        unittest.makeSuite(SSLConfigTestMockiavellian),
+        ))
 
 # Helpers for other tests: