def test_server_socket_close(): server_socket = TServerSocket(host="localhost", port=12345) server_socket.listen() client_socket = TSocket(host="localhost", port=12345, socket_timeout=100) client_socket.open() conn = server_socket.accept() server_socket.close() client_socket.write(b"Hello world") with pytest.raises(socket.timeout): client_socket.read(1024) conn.close() server_socket.close()
def test_server_socket_close(): server_socket = TServerSocket(host="localhost", port=12345) server_socket.listen() client_socket = TSocket(host="localhost", port=12345, socket_timeout=100) client_socket.open() conn = server_socket.accept() server_socket.close() client_socket.write(b"Hello world") with pytest.raises(socket.timeout): client_socket.read(1024) conn.close() server_socket.close()
def test_client_socket_close(): server_socket = TServerSocket(host="localhost", port=12345) server_socket.listen() client_socket = TSocket(host="localhost", port=12345) client_socket.open() conn = server_socket.accept() client_socket.close() with pytest.raises(TTransportException) as e: conn.read(1024) assert "TSocket read 0 bytes" in e.value.message conn.write(b"world") conn.close() server_socket.close()
def test_client_socket_timeout(): server_socket = TServerSocket(host="localhost", port=12345, client_timeout=100) server_socket.listen() client_socket = TSocket(host="localhost", port=12345, connect_timeout=10, socket_timeout=100) client_socket.open() conn = server_socket.accept() assert client_socket.sock.gettimeout() == 100 / 1000 assert conn.sock.gettimeout() == 100 / 1000 assert server_socket.sock.gettimeout() is None conn.close() client_socket.close() server_socket.close()