Function TestTCPTransportWriteToUnreachableAddr

kv/memberlist/tcp_transport_test.go:81–122

(t *testing.T)

}

func TestTCPTransportWriteToUnreachableAddr(t *testing.T) {
	writeCt := 50

	// Listen for TCP connections on a random port
	listener, err := net.Listen("tcp", "127.0.0.1:0")
	require.NoError(t, err)
	defer listener.Close()

	logs := &concurrency.SyncBuffer{}
	logger := log.NewLogfmtLogger(logs)

	cfg := TCPTransportConfig{}
	flagext.DefaultValues(&cfg)
	cfg.BindAddrs = getLocalhostAddrs()
	cfg.MaxConcurrentWrites = writeCt
	cfg.PacketDialTimeout = 500 * time.Millisecond
	transport, err := NewTCPTransport(cfg, logger, nil)
	require.NoError(t, err)

	// Configure TLS only for writes. The dialing should timeout (because of the timeoutReader)
	transport.cfg.TLSEnabled = true
	transport.cfg.TLS = tls.ClientConfig{
		Reader:   &timeoutReader{},
		CertPath: "fake",
		KeyPath:  "fake",
		CAPath:   "fake",
	}

	timeStart := time.Now()

	for i := 0; i < writeCt; i++ {
		_, err = transport.WriteTo([]byte("test"), listener.Addr().String())
		require.NoError(t, err)
	}

	require.NoError(t, transport.Shutdown())

	gotErrorCt := strings.Count(logs.String(), "context deadline exceeded")
	assert.Equal(t, writeCt, gotErrorCt, "expected %d errors, got %d", writeCt, gotErrorCt)
	assert.GreaterOrEqual(t, time.Since(timeStart), 500*time.Millisecond, "expected to take at least 500ms (timeout duration)")
	assert.LessOrEqual(t, time.Since(timeStart), 2*time.Second, "expected to take less than 2s (timeout + a good margin), writing to unreachable addresses should not block")
}

func TestTCPTransportWriterAcquireTimeout(t *testing.T) {
	// Listen for TCP connections on a random port