Function TestSQLDataWarmup

internal/proxy/proxy_test.go:706–760

(t *testing.T)

}

func TestSQLDataWarmup(t *testing.T) {
	tcs := []struct {
		desc string
		in   *proxy.Config
		want int
	}{
		{
			desc: "standard warmup",
			in: &proxy.Config{
				Instances: []proxy.InstanceConnConfig{{Name: "proj:reg:inst", Port: 30104}},
			},
			want: 1,
		},
		{
			desc: "warmup skipped globally",
			in: &proxy.Config{
				SQLDataEnabled: true,
				Instances:      []proxy.InstanceConnConfig{{Name: "proj:reg:inst", Port: 30105}},
			},
			want: 0,
		},
		{
			desc: "warmup skipped per-instance",
			in: &proxy.Config{
				Instances: []proxy.InstanceConnConfig{{
					Name:           "proj:reg:inst",
					Port:           30106,
					SQLDataEnabled: pointer(true),
				}},
			},
			want: 0,
		},
	}

	for _, tc := range tcs {
		t.Run(tc.desc, func(t *testing.T) {
			d := &fakeDialer{}
			c, err := proxy.NewClient(context.Background(), d, testLogger, tc.in, nil)
			if err != nil {
				t.Fatalf("proxy.NewClient error: %v", err)
			}
			defer c.Close()

			var got int
			for i := 0; i < 10; i++ {
				got = d.engineVersionAttempts()
				if got == tc.want {
					return
				}
				time.Sleep(100 * time.Millisecond)
			}
			t.Fatalf("warmup attempts, want = %v, got = %v", tc.want, got)
		})
	}
}

func TestCheckConnections(t *testing.T) {
	in := &proxy.Config{