Function test_socket_connect

src/tests/test_python_socket.py:58–88

Test that Python socket can connect to a server and exchange data.

(selenium_nodesock)

def test_socket_connect(selenium_nodesock):
    """Test that Python socket can connect to a server and exchange data."""
    TEST_MESSAGE = b"Hello from client"
    RESPONSE_MESSAGE = b"Hello from server"

    server_received = []

    def handler(conn, _addr):
        data = conn.recv(1024)
        server_received.append(data)
        conn.sendall(RESPONSE_MESSAGE)

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

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

        s.sendall(message)

        response = s.recv(1024)

        s.close()
        return response.decode()

    with tcp_server(handler) as (host, port):
        result = run(selenium_nodesock, host, port, TEST_MESSAGE)
        assert len(server_received) == 1
        assert server_received[0] == TEST_MESSAGE
        assert result == RESPONSE_MESSAGE.decode()


def test_socket_multiple_send_recv(selenium_nodesock):