Make non-temp cluster usable with tests again

Author: elprans

asyncpg/cluster.py

@@ -78,6 +78,9 @@ def __init__(self, data_dir, *, pg_config_path=None):
         self._daemon_process = None
         self._connection_addr = None
 
+    def is_managed(self):
+        return True
+
     def get_data_dir(self):
         return self._data_dir
 
@@ -448,13 +451,15 @@ def __init__(self, host=None, port=None):
         self.host = host
         self.port = port
 
+    def is_managed(self):
+        return False
+
+    def get_connection_addr(self):
+        return self.host, self.port
+
     def get_status(self):
         return 'running'
 
-    async def connect(self, loop=None, **kwargs):
-        return await asyncpg.connect(
-            host=self.host, port=self.port, loop=loop, **kwargs)
-
     def init(self, **settings):
         pass
 
@@ -466,3 +471,10 @@ def stop(self, wait=60):
 
     def destroy(self):
         pass
+
+    def reset_hba(self):
+        raise ClusterError('cannot modify HBA records of unmanaged cluster')
+
+    def add_hba_entry(self, *, type='host', database, user, address=None,
+                      auth_method, auth_options=None):
+        raise ClusterError('cannot modify HBA records of unmanaged cluster')

tests/test_connect.py

@@ -27,6 +27,9 @@ class TestAuthentication(tb.ConnectedTestCase):
     def setUp(self):
         super().setUp()
 
+        if not self.cluster.is_managed():
+            self.skipTest('unmanaged cluster')
+
         methods = [
             ('trust', None),
             ('reject', None),