Function reply_next_layer

test/mitmproxy/proxy/tutils.py:456–467

Helper function to simplify the syntax for next_layer events to this: << NextLayerHook(nl) >> reply_next_layer(tutils.EchoLayer)

(
    child_layer: type[Layer] | Callable[[context.Context], Layer], *args, **kwargs
)

def reply_next_layer(
    child_layer: type[Layer] | Callable[[context.Context], Layer], *args, **kwargs
) -> reply:
    """Helper function to simplify the syntax for next_layer events to this:
    << NextLayerHook(nl)
    >> reply_next_layer(tutils.EchoLayer)
    """

    def set_layer(next_layer: layer.NextLayer) -> None:
        next_layer.layer = child_layer(next_layer.context)

    return reply(*args, side_effect=set_layer, **kwargs)