Function test_socket_create_multiple

src/tests/test_python_socket.py:327–361

Test creating multiple sockets simultaneously.

(selenium_nodesock)

def test_socket_create_multiple(selenium_nodesock):
    """Test creating multiple sockets simultaneously."""

    def echo_handler(conn, _addr):
        data = conn.recv(1024)
        _, sport = conn.getsockname()
        conn.sendall(f"Server{sport}:{data.decode()}".encode())

    @run_in_pyodide
    def run(selenium, host1, port1, host2, port2):
        import socket

        s1 = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s2 = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        s1.connect((host1, port1))
        s2.connect((host2, port2))

        s1.sendall(b"Hello1")
        s2.sendall(b"Hello2")

        r1 = s1.recv(1024)
        r2 = s2.recv(1024)

        s1.close()
        s2.close()

        return [r1.decode(), r2.decode()]

    with (
        tcp_server(echo_handler) as (host1, port1),
        tcp_server(echo_handler) as (host2, port2),
    ):
        result = run(selenium_nodesock, host1, port1, host2, port2)
        assert result == [f"Server{port1}:Hello1", f"Server{port2}:Hello2"]


def test_socket_recv_eof(selenium_nodesock):