Function TestClientConn_SetKeyspace

proxycore/clientconn_test.go:238–284

(t *testing.T)

}

func TestClientConn_SetKeyspace(t *testing.T) {
	server := &MockServer{
		Handlers: NewMockRequestHandlers(MockRequestHandlers{
			primitive.OpCodeQuery: func(cl *MockClient, frm *frame.Frame) message.Message {
				if msg := cl.InterceptQuery(frm.Header, frm.Body.Message.(*message.Query)); msg != nil {
					return msg
				} else {
					return &message.Invalid{ErrorMessage: "Doesn't exist"}
				}
			},
		}),
	}

	const supported = primitive.ProtocolVersion4

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

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

	cl, err := ConnectClient(ctx, NewEndpoint("127.0.0.1:9042"), ClientConnConfig{})
	require.NoError(t, err)

	_, err = cl.Handshake(ctx, supported, nil)
	require.NoError(t, err)

	_, err = cl.Query(ctx, supported, &message.Query{
		Query: "SELECT * FROM local",
	})
	if assert.Error(t, err) {
		assert.Contains(t, err.Error(), "Doesn't exist")
	}

	err = cl.SetKeyspace(ctx, supported, "system")
	require.NoError(t, err)

	rs, err := cl.Query(ctx, supported, &message.Query{
		Query: "SELECT * FROM local",
	})
	require.NoError(t, err)
	require.Equal(t, rs.RowCount(), 1)
}

func TestClientConn_Inflight(t *testing.T) {
	server := &MockServer{