broken fsm on req-rep

import click
import zmq

context = zmq.Context()


@click.group()
def cli():
    pass


@cli.command()
def server():
    socket = context.socket(zmq.REP)
    socket.bind("tcp://*:5555")

    while True:
        msg = socket.recv()
        print(msg)
        socket.send(b"A" + msg[1:])


@cli.command()
def client():
    socket = context.socket(zmq.REQ)
    socket.connect("tcp://127.0.0.1:5556")

    cnt = 0
    while True:
        socket.send(b"D" + str(cnt).encode())
        msg = socket.recv()
        print(msg)
        cnt += 1


if __name__ == "__main__":
    cli()