Ejemplos de MessageFactory.build en Python

Ejemplo n.º 1

def test_message_fragment(arg2, arg3, connection):
    msg = CallRequestMessage(args=["", arg2, arg3])
    origin_msg = CallRequestMessage(args=["", arg2, arg3])
    message_factory = MessageFactory(connection)
    fragments = message_factory.fragment(msg)
    recv_msg = None
    for fragment in fragments:
        output = message_factory.build(fragment)
        if output:
            recv_msg = output
    header = yield recv_msg.get_header()
    body = yield recv_msg.get_body()
    assert header == origin_msg.args[1]
    assert body == origin_msg.args[2]

Ejemplo n.º 2

Archivo: test_messages.py Proyecto: jonasrosland/tchannel

def test_message_fragment(arg2, arg3, connection):
    msg = CallRequestMessage(args=["", arg2, arg3])
    origin_msg = CallRequestMessage(args=["", arg2, arg3])
    message_factory = MessageFactory(connection)
    fragments = message_factory.fragment(msg)
    recv_msg = None
    for fragment in fragments:
        output = message_factory.build(fragment)
        if output:
            recv_msg = output
    header = yield recv_msg.get_header()
    body = yield recv_msg.get_body()
    assert header == origin_msg.args[1]
    assert body == origin_msg.args[2]