test: fix proxy-based tests confused by the PGHOST env var

diff --git a/tests/fix_proxy.py b/tests/fix_proxy.py
index 7101aac3df3406ddfb3a800bdc1d14870f6213d8..e50f5ec05f28b460cc7c9c349f055db763a64ad3 100644 (file)
--- a/tests/fix_proxy.py
+++ b/tests/fix_proxy.py
@@ -1,3 +1,4 @@
+import os
 import time
 import socket
 import logging
@@ -57,11 +58,9 @@ class Proxy:
         cdict = conninfo.conninfo_to_dict(server_dsn)
 
         # Get server params
+        host = cdict.get("host") or os.environ.get("PGHOST")
+        self.server_host = host if host and not host.startswith("/") else "localhost"
         self.server_port = cdict.get("port", "5432")
-        if "host" not in cdict or cdict["host"].startswith("/"):
-            self.server_host = "localhost"
-        else:
-            self.server_host = cdict["host"]
 
         # Get client params
         self.client_host = "localhost"