Fix multi port connection string issue (#1222)

Co-authored-by: CommanderKeynes <[email protected]>

Author: AndrewJackson2020

asyncpg/connect_utils.py

@@ -168,13 +168,15 @@ def _read_password_from_pgpass(
 
 
 def _validate_port_spec(hosts, port):
-    if isinstance(port, list):
+    if isinstance(port, list) and len(port) > 1:
         # If there is a list of ports, its length must
         # match that of the host list.
         if len(port) != len(hosts):
             raise exceptions.ClientConfigurationError(
                 'could not match {} port numbers to {} hosts'.format(
                     len(port), len(hosts)))
+    elif isinstance(port, list) and len(port) == 1:
+        port = [port[0] for _ in range(len(hosts))]
     else:
         port = [port for _ in range(len(hosts))]
 

tests/test_connect.py

@@ -1088,6 +1088,20 @@ class TestConnectParams(tb.TestCase):
                 }
             )
         },
+        {
+            'name': 'multi_host_single_port',
+            'dsn': 'postgres:///postgres?host=127.0.0.1,127.0.0.2&port=5432&user=postgres',
+            'result': (
+                [
+                    ('127.0.0.1', 5432),
+                    ('127.0.0.2', 5432)
+                ], {
+                    'user': 'postgres',
+                    'database': 'postgres',
+                    'target_session_attrs': 'any',
+                }
+            )
+        },
     ]
 
     @contextlib.contextmanager