From 17d3d592e980720a8baa9a98e91a3de9fec06af7 Mon Sep 17 00:00:00 2001
From: Artemiy Ryabinkov <getlag@ya.ru>
Date: Sat, 14 Sep 2019 19:11:26 +0300
Subject: [PATCH] add test for custom lookup func

Signed-off-by: Artemiy Ryabinkov <getlag@ya.ru>
---
 pgconn_test.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/pgconn_test.go b/pgconn_test.go
index 4a67a2e0..36499b68 100644
--- a/pgconn_test.go
+++ b/pgconn_test.go
@@ -188,6 +188,24 @@ func TestConnectCustomDialer(t *testing.T) {
 	closeConn(t, conn)
 }
 
+func TestConnectCustomLookup(t *testing.T) {
+	t.Parallel()
+
+	config, err := pgconn.ParseConfig(os.Getenv("PGX_TEST_CONN_STRING"))
+	require.NoError(t, err)
+
+	looked := false
+	config.LookupFunc = func(ctx context.Context, host string) (addrs []string, err error) {
+		looked = true
+		return net.LookupHost(host)
+	}
+
+	conn, err := pgconn.ConnectConfig(context.Background(), config)
+	require.NoError(t, err)
+	require.True(t, looked)
+	closeConn(t, conn)
+}
+
 func TestConnectWithRuntimeParams(t *testing.T) {
 	t.Parallel()