Function TestRun_ConfigFileWithPeers

proxy/run_test.go:105–201

(t *testing.T)

}

func TestRun_ConfigFileWithPeers(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())

	proxy1Addr, proxy2Addr := testAddr, generateTestAddr(testAddr, 1)
	clusterPort, clusterAddr, proxy1BindAddr, httpBindAddr := generateTestAddrs(proxy1Addr)
	cluster := proxycore.NewMockCluster(net.ParseIP(testStartAddr), clusterPort)

	defer cluster.Shutdown()
	err := cluster.Add(ctx, 1)
	require.NoError(t, err)

	configFileName, err := writeTempYaml(struct {
		Bind          string
		Port          int
		RPCAddr       string   `yaml:"rpc-address"`
		DataCenter    string   `yaml:"data-center"`
		ContactPoints []string `yaml:"contact-points"`
		HealthCheck   bool     `yaml:"health-check"`
		HttpBind      string   `yaml:"http-bind"`
		Peers         []PeerConfig
	}{
		Bind:          proxy1BindAddr,
		RPCAddr:       proxy1Addr,
		DataCenter:    "dc-1",
		Port:          clusterPort,
		ContactPoints: []string{clusterAddr},
		HealthCheck:   true,
		HttpBind:      httpBindAddr,
		Peers: []PeerConfig{{
			RPCAddr: proxy1Addr,
			DC:      "dc-1",
		}, {
			RPCAddr: proxy2Addr,
			DC:      "dc-2",
		}},
	})

	var wg sync.WaitGroup
	wg.Add(1)

	go func() {
		rc := Run(ctx, []string{
			"--config", configFileName,
		})
		assert.Equal(t, 0, rc)
		wg.Done()
	}()

	defer func() {
		cancel()
		wg.Wait()
	}()

	require.True(t, waitUntil(10*time.Second, func() bool {
		return checkLiveness(httpBindAddr)
	}))

	cl := connectTestClient(t, ctx, proxy1BindAddr)