3dc324d3f1
This adds the Python files in the tests/ directory to be formatted with ./tools/codeformat.py. The basics/ subdirectory is excluded for now so we aren't changing too much at once. In a few places `# fmt: off`/`# fmt: on` was used where the code had special formatting for readability or where the test was actually testing the specific formatting.
32 lines
650 B
Python
32 lines
650 B
Python
# test that socket.connect() has correct polling behaviour before, during and after
|
|
|
|
try:
|
|
import usocket as socket, uselect as select
|
|
except:
|
|
import socket, select
|
|
|
|
|
|
def test(peer_addr):
|
|
s = socket.socket()
|
|
poller = select.poll()
|
|
poller.register(s)
|
|
|
|
# test poll before connect
|
|
p = poller.poll(0)
|
|
print(len(p), p[0][-1])
|
|
|
|
s.connect(peer_addr)
|
|
|
|
# test poll during connection
|
|
print(len(poller.poll(0)))
|
|
|
|
# test poll after connection is established
|
|
p = poller.poll(1000)
|
|
print(len(p), p[0][-1])
|
|
|
|
s.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test(socket.getaddrinfo("micropython.org", 80)[0][-1])
|