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]