def test_create_tunnel(self): response = requests.get(networking.GRADIO_API_SERVER) payload = response.json()[0] io = Interface(lambda x: x, "text", "text") _, path_to_local_server, _ = io.launch(prevent_thread_lock=True, share=False) _, localhost, port = path_to_local_server.split(":") threading.Thread.start = mock.MagicMock(return_value=None) paramiko.SSHClient.connect = mock.MagicMock(return_value=None) tunneling.create_tunnel(payload, localhost, port) threading.Thread.start.assert_called_once() paramiko.SSHClient.connect.assert_called_once() io.close()
def setup_tunnel(local_server_port): response = url_request(GRADIO_API_SERVER) if response and response.code == 200: try: payload = json.loads(response.read().decode("utf-8"))[0] return create_tunnel(payload, LOCALHOST_NAME, local_server_port) except Exception as e: raise RuntimeError(str(e))
def setup_tunnel(local_server_port: int, endpoint: str) -> str: response = url_request( endpoint + '/v1/tunnel-request' if endpoint is not None else GRADIO_API_SERVER) if response and response.code == 200: try: payload = json.loads(response.read().decode("utf-8"))[0] return create_tunnel(payload, LOCALHOST_NAME, local_server_port) except Exception as e: raise RuntimeError(str(e))
def setup_tunnel(local_server_port: int, endpoint: str) -> str: response = requests.get( endpoint + "/v1/tunnel-request" if endpoint is not None else GRADIO_API_SERVER) if response and response.status_code == 200: try: payload = response.json()[0] return create_tunnel(payload, LOCALHOST_NAME, local_server_port) except Exception as e: raise RuntimeError(str(e)) else: raise RuntimeError("Could not get share link from Gradio API Server.")