summaryrefslogtreecommitdiffstatshomepage
path: root/tests/extmod/uselect_poll_udp.py
blob: 2a56a122b5f2144c05550d18e96f8394dc325b39 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# test select.poll on UDP sockets

try:
    import usocket as socket, uselect as select
except ImportError:
    try:
        import socket, select

        select.poll  # Raises AttributeError for CPython implementations without poll()
    except (ImportError, AttributeError):
        print("SKIP")
        raise SystemExit


s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.bind(socket.getaddrinfo("127.0.0.1", 8000)[0][-1])
poll = select.poll()

# UDP socket should not be readable
poll.register(s, select.POLLIN)
print(len(poll.poll(0)))

# UDP socket should be writable
poll.modify(s, select.POLLOUT)
print(poll.poll(0)[0][1] == select.POLLOUT)

# same test for select.select, but just skip it if the function isn't available
if hasattr(select, "select"):
    r, w, e = select.select([s], [], [], 0)
    assert not r and not w and not e