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()
def test_close(): server = TServerSocket(host="localhost", port=12345) client = TSocket(host="localhost", port=12345) server.listen() client.open() c = server.accept() client.close() with pytest.raises(TTransportException) as e: c.read(1024) assert "TSocket read 0 bytes" in e.value.message c.write(b"world") c.close() assert c.handle is None