test: Add e2e test to access instance using DNS SRV record. part of #842

Author: hessjcg

e2e_mysql_test.go

@@ -15,7 +15,10 @@
 package cloudsqlconn_test
 
 import (
+	"context"
 	"database/sql"
+	"fmt"
+	"net"
 	"os"
 	"testing"
 	"time"
@@ -54,25 +57,55 @@ func requireMySQLVars(t *testing.T) {
 	}
 }
 
+type mockResolver struct {
+}
+
+func (r *mockResolver) LookupSRV(_ context.Context, _, _, name string) (cname string, addrs []*net.SRV, err error) {
+	if name == "mysql.example.com" {
+		return "", []*net.SRV{
+			&net.SRV{Target: mysqlConnName},
+		}, nil
+	}
+
+	return "", nil, fmt.Errorf("no resolution for %v", name)
+}
+
 func TestMySQLDriver(t *testing.T) {
 	if testing.Short() {
 		t.Skip("skipping MySQL integration tests")
 	}
 
 	tcs := []struct {
-		desc       string
-		driverName string
-		opts       []cloudsqlconn.Option
+		desc         string
+		driverName   string
+		instanceName string
+		user         string
+		password     string
+		opts         []cloudsqlconn.Option
 	}{
 		{
-			desc:       "default options",
-			driverName: "cloudsql-mysql",
-			opts:       nil,
+			desc:         "default options",
+			driverName:   "cloudsql-mysql",
+			opts:         nil,
+			instanceName: mysqlConnName,
+			user:         mysqlUser,
+			password:     mysqlPass,
+		},
+		{
+			desc:         "auto IAM authn",
+			driverName:   "cloudsql-mysql-iam",
+			opts:         []cloudsqlconn.Option{cloudsqlconn.WithIAMAuthN()},
+			instanceName: mysqlIAMConnName,
+			user:         mysqlIAMUser,
+			password:     "password",
 		},
 		{
-			desc:       "auto IAM authn",
-			driverName: "cloudsql-mysql-iam",
-			opts:       []cloudsqlconn.Option{cloudsqlconn.WithIAMAuthN()},
+			desc:         "with dns",
+			driverName:   "cloudsql-mysql-dns",
+			opts:         []cloudsqlconn.Option{cloudsqlconn.WithResolver(&mockResolver{})},
+			instanceName: "mysql.example.com",
+			user:         mysqlUser,
+			password:     mysqlPass,
 		},
 	}
 
@@ -85,18 +118,18 @@ func TestMySQLDriver(t *testing.T) {
 				}
 				t.Log(now)
 			}
-			cleanup, err := mysql.RegisterDriver(tc.driverName)
+			cleanup, err := mysql.RegisterDriver(tc.driverName, tc.opts...)
 			if err != nil {
 				t.Fatalf("failed to register driver: %v", err)
 			}
 			defer cleanup()
 			cfg := gomysql.NewConfig()
 			cfg.CheckConnLiveness = true
-			cfg.User = mysqlUser
-			cfg.Passwd = mysqlPass
+			cfg.User = tc.user
+			cfg.Passwd = tc.password
 			cfg.DBName = mysqlDB
 			cfg.Net = tc.driverName
-			cfg.Addr = mysqlConnName
+			cfg.Addr = tc.instanceName
 			cfg.Params = map[string]string{"parseTime": "true"}
 
 			db, err := sql.Open("mysql", cfg.FormatDSN())

e2e_postgres_test.go

@@ -107,6 +107,67 @@ func TestPostgresPgxPoolConnect(t *testing.T) {
 	t.Log(now)
 }
 
+type pgMockResolver struct {
+}
+
+func (r *pgMockResolver) LookupSRV(_ context.Context, _, _, name string) (cname string, addrs []*net.SRV, err error) {
+	if name == "pg.example.com" {
+		return "", []*net.SRV{
+			&net.SRV{Target: postgresConnName},
+		}, nil
+	}
+
+	return "", nil, fmt.Errorf("no resolution for %v", name)
+}
+
+func TestPostgresPgxPoolConnectDomainName(t *testing.T) {
+	if testing.Short() {
+		t.Skip("skipping Postgres integration tests")
+	}
+	requirePostgresVars(t)
+
+	ctx := context.Background()
+
+	// Configure the driver to connect to the database
+	dsn := fmt.Sprintf("user=%s password=%s dbname=%s sslmode=disable", postgresUser, postgresPass, postgresDB)
+	config, err := pgxpool.ParseConfig(dsn)
+	if err != nil {
+		t.Fatalf("failed to parse pgx config: %v", err)
+	}
+
+	// Create a new dialer with any options
+	d, err := cloudsqlconn.NewDialer(ctx, cloudsqlconn.WithResolver(&pgMockResolver{}))
+	if err != nil {
+		t.Fatalf("failed to init Dialer: %v", err)
+	}
+
+	// call cleanup when you're done with the database connection to close dialer
+	cleanup := func() error { return d.Close() }
+
+	// Tell the driver to use the Cloud SQL Go Connector to create connections
+	// postgresConnName takes the form of 'project:region:instance'.
+	config.ConnConfig.DialFunc = func(ctx context.Context, _ string, addr string) (net.Conn, error) {
+		return d.Dial(ctx, "pg.example.com")
+	}
+
+	// Interact with the driver directly as you normally would
+	pool, err := pgxpool.NewWithConfig(ctx, config)
+	if err != nil {
+		t.Fatalf("failed to create pool: %s", err)
+	}
+	// ... etc
+
+	defer cleanup()
+	defer pool.Close()
+
+	var now time.Time
+	err = pool.QueryRow(context.Background(), "SELECT NOW()").Scan(&now)
+	if err != nil {
+		t.Fatalf("QueryRow failed: %s", err)
+	}
+	t.Log(now)
+}
+
 func TestPostgresConnectWithIAMUser(t *testing.T) {
 	if testing.Short() {
 		t.Skip("skipping Postgres integration tests")