Function TestRun_ConfigFileWithNoPeerTokens

proxy/run_test.go:332–372

(t *testing.T)

}

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

	proxy1Addr, proxy2Addr := testAddr, generateTestAddr(testAddr, 1)
	clusterPort, clusterAddr, proxy1BindAddr, _ := generateTestAddrs(proxy1Addr)

	cluster := proxycore.NewMockCluster(net.ParseIP(testStartAddr), clusterPort)

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

	defer cluster.Shutdown()

	configFileName, err := writeTempYaml(struct {
		Bind          string
		Port          int
		RPCAddr       string `yaml:"rpc-address"`
		DataCenter    string `yaml:"data-center"`
		Tokens        []string
		ContactPoints []string `yaml:"contact-points"`
		Peers         []PeerConfig
	}{
		Bind:          proxy1BindAddr,
		ContactPoints: []string{clusterAddr},
		Port:          clusterPort,
		RPCAddr:       proxy1Addr,
		Tokens:        []string{"0"}, // Local tokens set
		Peers: []PeerConfig{{
			RPCAddr: proxy2Addr,
			DC:      "dc-2",
			// No peer tokens
		}},
	})
	require.NoError(t, err)

	rc := Run(ctx, []string{
		"--config", configFileName,
	})
	require.Equal(t, 1, rc)
}

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