Function setupProxyTestWithConfig

proxy/proxy_test.go:429–485

(ctx context.Context, numNodes int, cfg *proxyTestConfig)

}

func setupProxyTestWithConfig(ctx context.Context, numNodes int, cfg *proxyTestConfig) (tester *proxyTester, proxyContactPoint string, err error) {
	tester = &proxyTester{
		wg: sync.WaitGroup{},
	}

	clusterPort, clusterAddr, proxyAddr, _ := generateTestAddrs(testAddr)

	tester.cluster = proxycore.NewMockCluster(net.ParseIP(testStartAddr), clusterPort)
	tester.cluster.DseVersion = cfg.dseVersion

	if cfg == nil {
		cfg = &proxyTestConfig{}
	}

	if cfg.handlers != nil {
		tester.cluster.Handlers = proxycore.NewMockRequestHandlers(cfg.handlers)
	}

	for i := 1; i <= numNodes; i++ {
		err = tester.cluster.Add(ctx, i)
		if err != nil {
			return tester, proxyAddr, err
		}
	}
	tester.proxy = NewProxy(ctx, Config{
		Version:           primitive.ProtocolVersion4,
		Resolver:          proxycore.NewResolverWithDefaultPort([]string{clusterAddr}, clusterPort),
		ReconnectPolicy:   proxycore.NewReconnectPolicyWithDelays(200*time.Millisecond, time.Second),
		NumConns:          2,
		HeartBeatInterval: 30 * time.Second,
		ConnectTimeout:    10 * time.Second,
		IdleTimeout:       60 * time.Second,
		RPCAddr:           cfg.rpcAddr,
		Peers:             cfg.peers,
		IdempotentGraph:   cfg.idempotentGraph,
		Logger:            zap.L(),
	})

	err = tester.proxy.Connect()
	if err != nil {
		return tester, proxyAddr, err
	}

	l, err := resolveAndListen(proxyAddr, "", "")
	if err != nil {
		return tester, proxyAddr, err
	}

	tester.wg.Add(1)

	go func() {
		_ = tester.proxy.Serve(l)
		tester.wg.Done()
	}()

	return tester, proxyAddr, nil
}