Class TTunnelLayer

test/mitmproxy/proxy/test_tunnel.py:43–62

class TTunnelLayer(tunnel.TunnelLayer):
    def start_handshake(self) -> layer.CommandGenerator[None]:
        yield SendData(self.tunnel_connection, b"handshake-hello")

    def receive_handshake_data(
        self, data: bytes
    ) -> layer.CommandGenerator[tuple[bool, str | None]]:
        yield SendData(self.tunnel_connection, data)
        if data == b"handshake-success":
            return True, None
        else:
            return False, "handshake error"

    def send_data(self, data: bytes) -> layer.CommandGenerator[None]:
        yield SendData(self.tunnel_connection, b"tunneled-" + data)

    def receive_data(self, data: bytes) -> layer.CommandGenerator[None]:
        yield from self.event_to_child(
            DataReceived(self.conn, data.replace(b"tunneled-", b""))
        )


@pytest.mark.parametrize("success", ["success", "fail"])