Function test_socket_large_data_transfer

src/tests/test_python_socket.py:126–160

Test transferring larger amounts of data.

(selenium_nodesock)

def test_socket_large_data_transfer(selenium_nodesock):
    """Test transferring larger amounts of data."""
    DATA_SIZE = 64 * 1024  # 64KB

    server_received = []

    def handler(conn, _addr):
        received = b""
        while len(received) < DATA_SIZE:
            chunk = conn.recv(8192)
            if not chunk:
                break
            received += chunk
        server_received.append(received)
        conn.sendall(f"Received {len(received)} bytes".encode())

    @run_in_pyodide
    def run(selenium, host, port, data_size):
        import socket

        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.connect((host, port))

        data = b"X" * data_size
        s.sendall(data)

        response = s.recv(1024)
        s.close()
        return response.decode()

    with tcp_server(handler, timeout=10.0) as (host, port):
        result = run(selenium_nodesock, host, port, DATA_SIZE)
        assert len(server_received) == 1
        assert len(server_received[0]) == DATA_SIZE
        assert result == f"Received {DATA_SIZE} bytes"


def test_socket_getpeername(selenium_nodesock):