Function TestRun_ConfigFileWithTokensProvided

proxy/run_test.go:203–293

(t *testing.T)

}

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

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

	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"`
		Tokens        []string
		ContactPoints []string `yaml:"contact-points"`
		HealthCheck   bool     `yaml:"health-check"`
		HttpBind      string   `yaml:"http-bind"`
		Peers         []PeerConfig
	}{
		Bind:          proxy1BindAddr,
		RPCAddr:       proxy1Addr,
		DataCenter:    "dc-1",
		Tokens:        []string{"0", "1"}, // Provide custom tokens
		Port:          clusterPort,
		ContactPoints: []string{clusterAddr},
		HealthCheck:   true,
		HttpBind:      httpBindAddr,
		Peers: []PeerConfig{{
			RPCAddr: proxy2Addr,
			Tokens:  []string{"42", "613"}, // Same here
		}},
	})

	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)