Function TestConnectClientWithEvents

proxycore/clientconn_test.go:102–148

(t *testing.T)

}

func TestConnectClientWithEvents(t *testing.T) {
	var server MockServer

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	const supported = primitive.ProtocolVersion2
	const starting = primitive.ProtocolVersion4

	err := server.Serve(ctx, supported, MockHost{
		IP:   "127.0.0.1",
		Port: 9042,
	}, nil)
	require.NoError(t, err)

	events := make(chan *frame.Frame)
	cl, err := ConnectClient(ctx, NewEndpoint("127.0.0.1:9042"), ClientConnConfig{
		Handler: EventHandlerFunc(func(frm *frame.Frame) {
			events <- frm
		}),
	})
	require.NoError(t, err)

	wait := func() *frame.Frame {
		timer := time.NewTimer(2 * time.Second)
		select {
		case <-timer.C:
			require.Fail(t, "timed out waiting for event")
		case event := <-events:
			return event
		}
		require.Fail(t, "event expected")
		return nil
	}

	version, err := cl.Handshake(ctx, starting, nil)
	require.NoError(t, err)
	assert.Equal(t, supported, version)

	status := &message.StatusChangeEvent{ChangeType: primitive.StatusChangeTypeUp, Address: &primitive.Inet{
		Addr: net.ParseIP("192.168.1.42"),
		Port: 9042,
	}}
	server.Event(status)
	received := wait()
	assert.Equal(t, status, received.Body.Message)
}

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