Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
tests/net_hosted: Add test for socket connect() and poll() behaviour.
- Loading branch information
Showing
2 changed files
with
35 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,32 @@ | ||
# 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 | ||
# note: CPython can return POLLHUP, so use the IN|OUT mask | ||
p = poller.poll(0) | ||
print(len(p), p[0][-1] & (select.POLLIN | select.POLLOUT)) | ||
|
||
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]) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,3 @@ | ||
1 4 | ||
1 | ||
1 4 |