test/test_tunneling.py

import pytest

from gradio import Interface, networking


@pytest.mark.flaky
def test_setup_tunnel():
    io = Interface(lambda x: x, "number", "number")
    io.launch(show_error=True, prevent_thread_lock=True)
    share_url = networking.setup_tunnel(
        io.server_name, io.server_port, io.share_token, io.share_server_address
    )
    assert isinstance(share_url, str)


@pytest.mark.flaky
def test_setup_custom_tunnel():
    io = Interface(lambda x: x, "number", "number")
    io.launch(
        show_error=True,
        prevent_thread_lock=True,
        share_server_address="my-gpt-wrapper.com:7000",
    )
    share_url = networking.setup_tunnel(
        io.server_name, io.server_port, io.share_token, io.share_server_address
    )
    assert isinstance(share_url, str)