Function test_socket_recv_eof

src/tests/test_python_socket.py:364–392

Test that recv returns b'' after server closes and all data is consumed.

(selenium_nodesock)

def test_socket_recv_eof(selenium_nodesock):
    """Test that recv returns b'' after server closes and all data is consumed."""

    def handler(conn, _addr):
        conn.sendall(b"goodbye")
        conn.close()

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

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

        data = s.recv(1024)

        # Give time for the FIN to arrive
        time.sleep(1.0)

        eof = s.recv(1024)

        s.close()
        return (data.decode(), len(eof))

    with tcp_server(handler) as (host, port):
        result = run(selenium_nodesock, host, port)
        assert result[0] == "goodbye"
        assert result[1] == 0


def test_socket_send_after_remote_close(selenium_nodesock):