Merge pull request #905 from asubiotto/oenv

Make TestCloseBadConn observe PGHOST and PGPORT
lib · Oct 11, 2019 · f91d341 · f91d341
2 parents 931b5ae + 16412e5
commit f91d341
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/conn_test.go b/conn_test.go
@@ -724,7 +724,15 @@ func TestBadConn(t *testing.T) {
 // TestCloseBadConn tests that the underlying connection can be closed with
 // Close after an error.
 func TestCloseBadConn(t *testing.T) {
-	nc, err := net.Dial("tcp", "localhost:5432")
+	host := os.Getenv("PGHOST")
+	if host == "" {
+		host = "localhost"
+	}
+	port := os.Getenv("PGPORT")
+	if port == "" {
+		port = "5432"
+	}
+	nc, err := net.Dial("tcp", host+":"+port)
 	if err != nil {
 		t.Fatal(err)
 	}