Function test_socket_send_recv_partial

src/tests/test_python_socket.py:295–324

Test partial recv when buffer is smaller than data.

(selenium_nodesock)

def test_socket_send_recv_partial(selenium_nodesock):
    """Test partial recv when buffer is smaller than data."""
    FULL_MESSAGE = b"A" * 1000

    def handler(conn, _addr):
        conn.recv(1024)
        conn.sendall(FULL_MESSAGE)

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

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

        s.sendall(b"start")

        received = b""
        while len(received) < len(full_message):
            chunk = s.recv(100)
            if not chunk:
                break
            received += chunk

        s.close()
        return received.decode()

    with tcp_server(handler) as (host, port):
        result = run(selenium_nodesock, host, port, FULL_MESSAGE)
        assert len(result) == len(FULL_MESSAGE)


def test_socket_create_multiple(selenium_nodesock):