| import pytest | |
| from gradio import Interface, networking | |
| 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) | |
| 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) | |