459 lines
14 KiB
Python
459 lines
14 KiB
Python
import errno
|
|
import os
|
|
import random
|
|
import selectors
|
|
import signal
|
|
import socket
|
|
from test import support
|
|
from time import sleep
|
|
import unittest
|
|
import unittest.mock
|
|
from time import monotonic as time
|
|
try:
|
|
import resource
|
|
except ImportError:
|
|
resource = None
|
|
|
|
|
|
if hasattr(socket, 'socketpair'):
|
|
socketpair = socket.socketpair
|
|
else:
|
|
def socketpair(family=socket.AF_INET, type=socket.SOCK_STREAM, proto=0):
|
|
with socket.socket(family, type, proto) as l:
|
|
l.bind((support.HOST, 0))
|
|
l.listen()
|
|
c = socket.socket(family, type, proto)
|
|
try:
|
|
c.connect(l.getsockname())
|
|
caddr = c.getsockname()
|
|
while True:
|
|
a, addr = l.accept()
|
|
# check that we've got the correct client
|
|
if addr == caddr:
|
|
return c, a
|
|
a.close()
|
|
except OSError:
|
|
c.close()
|
|
raise
|
|
|
|
|
|
def find_ready_matching(ready, flag):
|
|
match = []
|
|
for key, events in ready:
|
|
if events & flag:
|
|
match.append(key.fileobj)
|
|
return match
|
|
|
|
|
|
class BaseSelectorTestCase(unittest.TestCase):
|
|
|
|
def make_socketpair(self):
|
|
rd, wr = socketpair()
|
|
self.addCleanup(rd.close)
|
|
self.addCleanup(wr.close)
|
|
return rd, wr
|
|
|
|
def test_register(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
key = s.register(rd, selectors.EVENT_READ, "data")
|
|
self.assertIsInstance(key, selectors.SelectorKey)
|
|
self.assertEqual(key.fileobj, rd)
|
|
self.assertEqual(key.fd, rd.fileno())
|
|
self.assertEqual(key.events, selectors.EVENT_READ)
|
|
self.assertEqual(key.data, "data")
|
|
|
|
# register an unknown event
|
|
self.assertRaises(ValueError, s.register, 0, 999999)
|
|
|
|
# register an invalid FD
|
|
self.assertRaises(ValueError, s.register, -10, selectors.EVENT_READ)
|
|
|
|
# register twice
|
|
self.assertRaises(KeyError, s.register, rd, selectors.EVENT_READ)
|
|
|
|
# register the same FD, but with a different object
|
|
self.assertRaises(KeyError, s.register, rd.fileno(),
|
|
selectors.EVENT_READ)
|
|
|
|
def test_unregister(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
s.register(rd, selectors.EVENT_READ)
|
|
s.unregister(rd)
|
|
|
|
# unregister an unknown file obj
|
|
self.assertRaises(KeyError, s.unregister, 999999)
|
|
|
|
# unregister twice
|
|
self.assertRaises(KeyError, s.unregister, rd)
|
|
|
|
def test_unregister_after_fd_close(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
rd, wr = self.make_socketpair()
|
|
r, w = rd.fileno(), wr.fileno()
|
|
s.register(r, selectors.EVENT_READ)
|
|
s.register(w, selectors.EVENT_WRITE)
|
|
rd.close()
|
|
wr.close()
|
|
s.unregister(r)
|
|
s.unregister(w)
|
|
|
|
@unittest.skipUnless(os.name == 'posix', "requires posix")
|
|
def test_unregister_after_fd_close_and_reuse(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
rd, wr = self.make_socketpair()
|
|
r, w = rd.fileno(), wr.fileno()
|
|
s.register(r, selectors.EVENT_READ)
|
|
s.register(w, selectors.EVENT_WRITE)
|
|
rd2, wr2 = self.make_socketpair()
|
|
rd.close()
|
|
wr.close()
|
|
os.dup2(rd2.fileno(), r)
|
|
os.dup2(wr2.fileno(), w)
|
|
self.addCleanup(os.close, r)
|
|
self.addCleanup(os.close, w)
|
|
s.unregister(r)
|
|
s.unregister(w)
|
|
|
|
def test_unregister_after_socket_close(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
rd, wr = self.make_socketpair()
|
|
s.register(rd, selectors.EVENT_READ)
|
|
s.register(wr, selectors.EVENT_WRITE)
|
|
rd.close()
|
|
wr.close()
|
|
s.unregister(rd)
|
|
s.unregister(wr)
|
|
|
|
def test_modify(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
key = s.register(rd, selectors.EVENT_READ)
|
|
|
|
# modify events
|
|
key2 = s.modify(rd, selectors.EVENT_WRITE)
|
|
self.assertNotEqual(key.events, key2.events)
|
|
self.assertEqual(key2, s.get_key(rd))
|
|
|
|
s.unregister(rd)
|
|
|
|
# modify data
|
|
d1 = object()
|
|
d2 = object()
|
|
|
|
key = s.register(rd, selectors.EVENT_READ, d1)
|
|
key2 = s.modify(rd, selectors.EVENT_READ, d2)
|
|
self.assertEqual(key.events, key2.events)
|
|
self.assertNotEqual(key.data, key2.data)
|
|
self.assertEqual(key2, s.get_key(rd))
|
|
self.assertEqual(key2.data, d2)
|
|
|
|
# modify unknown file obj
|
|
self.assertRaises(KeyError, s.modify, 999999, selectors.EVENT_READ)
|
|
|
|
# modify use a shortcut
|
|
d3 = object()
|
|
s.register = unittest.mock.Mock()
|
|
s.unregister = unittest.mock.Mock()
|
|
|
|
s.modify(rd, selectors.EVENT_READ, d3)
|
|
self.assertFalse(s.register.called)
|
|
self.assertFalse(s.unregister.called)
|
|
|
|
def test_close(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
s.register(rd, selectors.EVENT_READ)
|
|
s.register(wr, selectors.EVENT_WRITE)
|
|
|
|
s.close()
|
|
self.assertRaises(KeyError, s.get_key, rd)
|
|
self.assertRaises(KeyError, s.get_key, wr)
|
|
|
|
def test_get_key(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
key = s.register(rd, selectors.EVENT_READ, "data")
|
|
self.assertEqual(key, s.get_key(rd))
|
|
|
|
# unknown file obj
|
|
self.assertRaises(KeyError, s.get_key, 999999)
|
|
|
|
def test_get_map(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
keys = s.get_map()
|
|
self.assertFalse(keys)
|
|
self.assertEqual(len(keys), 0)
|
|
self.assertEqual(list(keys), [])
|
|
key = s.register(rd, selectors.EVENT_READ, "data")
|
|
self.assertIn(rd, keys)
|
|
self.assertEqual(key, keys[rd])
|
|
self.assertEqual(len(keys), 1)
|
|
self.assertEqual(list(keys), [rd.fileno()])
|
|
self.assertEqual(list(keys.values()), [key])
|
|
|
|
# unknown file obj
|
|
with self.assertRaises(KeyError):
|
|
keys[999999]
|
|
|
|
# Read-only mapping
|
|
with self.assertRaises(TypeError):
|
|
del keys[rd]
|
|
|
|
def test_select(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
s.register(rd, selectors.EVENT_READ)
|
|
wr_key = s.register(wr, selectors.EVENT_WRITE)
|
|
|
|
result = s.select()
|
|
for key, events in result:
|
|
self.assertTrue(isinstance(key, selectors.SelectorKey))
|
|
self.assertTrue(events)
|
|
self.assertFalse(events & ~(selectors.EVENT_READ |
|
|
selectors.EVENT_WRITE))
|
|
|
|
self.assertEqual([(wr_key, selectors.EVENT_WRITE)], result)
|
|
|
|
def test_context_manager(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
with s as sel:
|
|
sel.register(rd, selectors.EVENT_READ)
|
|
sel.register(wr, selectors.EVENT_WRITE)
|
|
|
|
self.assertRaises(KeyError, s.get_key, rd)
|
|
self.assertRaises(KeyError, s.get_key, wr)
|
|
|
|
def test_fileno(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
if hasattr(s, 'fileno'):
|
|
fd = s.fileno()
|
|
self.assertTrue(isinstance(fd, int))
|
|
self.assertGreaterEqual(fd, 0)
|
|
|
|
def test_selector(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
NUM_SOCKETS = 12
|
|
MSG = b" This is a test."
|
|
MSG_LEN = len(MSG)
|
|
readers = []
|
|
writers = []
|
|
r2w = {}
|
|
w2r = {}
|
|
|
|
for i in range(NUM_SOCKETS):
|
|
rd, wr = self.make_socketpair()
|
|
s.register(rd, selectors.EVENT_READ)
|
|
s.register(wr, selectors.EVENT_WRITE)
|
|
readers.append(rd)
|
|
writers.append(wr)
|
|
r2w[rd] = wr
|
|
w2r[wr] = rd
|
|
|
|
bufs = []
|
|
|
|
while writers:
|
|
ready = s.select()
|
|
ready_writers = find_ready_matching(ready, selectors.EVENT_WRITE)
|
|
if not ready_writers:
|
|
self.fail("no sockets ready for writing")
|
|
wr = random.choice(ready_writers)
|
|
wr.send(MSG)
|
|
|
|
for i in range(10):
|
|
ready = s.select()
|
|
ready_readers = find_ready_matching(ready,
|
|
selectors.EVENT_READ)
|
|
if ready_readers:
|
|
break
|
|
# there might be a delay between the write to the write end and
|
|
# the read end is reported ready
|
|
sleep(0.1)
|
|
else:
|
|
self.fail("no sockets ready for reading")
|
|
self.assertEqual([w2r[wr]], ready_readers)
|
|
rd = ready_readers[0]
|
|
buf = rd.recv(MSG_LEN)
|
|
self.assertEqual(len(buf), MSG_LEN)
|
|
bufs.append(buf)
|
|
s.unregister(r2w[rd])
|
|
s.unregister(rd)
|
|
writers.remove(r2w[rd])
|
|
|
|
self.assertEqual(bufs, [MSG] * NUM_SOCKETS)
|
|
|
|
def test_timeout(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
s.register(wr, selectors.EVENT_WRITE)
|
|
t = time()
|
|
self.assertEqual(1, len(s.select(0)))
|
|
self.assertEqual(1, len(s.select(-1)))
|
|
self.assertLess(time() - t, 0.5)
|
|
|
|
s.unregister(wr)
|
|
s.register(rd, selectors.EVENT_READ)
|
|
t = time()
|
|
self.assertFalse(s.select(0))
|
|
self.assertFalse(s.select(-1))
|
|
self.assertLess(time() - t, 0.5)
|
|
|
|
t0 = time()
|
|
self.assertFalse(s.select(1))
|
|
t1 = time()
|
|
dt = t1 - t0
|
|
self.assertTrue(0.8 <= dt <= 1.6, dt)
|
|
|
|
@unittest.skipUnless(hasattr(signal, "alarm"),
|
|
"signal.alarm() required for this test")
|
|
def test_select_interrupt(self):
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
rd, wr = self.make_socketpair()
|
|
|
|
orig_alrm_handler = signal.signal(signal.SIGALRM, lambda *args: None)
|
|
self.addCleanup(signal.signal, signal.SIGALRM, orig_alrm_handler)
|
|
self.addCleanup(signal.alarm, 0)
|
|
|
|
signal.alarm(1)
|
|
|
|
s.register(rd, selectors.EVENT_READ)
|
|
t = time()
|
|
self.assertFalse(s.select(2))
|
|
self.assertLess(time() - t, 2.5)
|
|
|
|
|
|
class ScalableSelectorMixIn:
|
|
|
|
# see issue #18963 for why it's skipped on older OS X versions
|
|
@support.requires_mac_ver(10, 5)
|
|
@unittest.skipUnless(resource, "Test needs resource module")
|
|
def test_above_fd_setsize(self):
|
|
# A scalable implementation should have no problem with more than
|
|
# FD_SETSIZE file descriptors. Since we don't know the value, we just
|
|
# try to set the soft RLIMIT_NOFILE to the hard RLIMIT_NOFILE ceiling.
|
|
soft, hard = resource.getrlimit(resource.RLIMIT_NOFILE)
|
|
try:
|
|
resource.setrlimit(resource.RLIMIT_NOFILE, (hard, hard))
|
|
self.addCleanup(resource.setrlimit, resource.RLIMIT_NOFILE,
|
|
(soft, hard))
|
|
NUM_FDS = min(hard, 2**16)
|
|
except (OSError, ValueError):
|
|
NUM_FDS = soft
|
|
|
|
# guard for already allocated FDs (stdin, stdout...)
|
|
NUM_FDS -= 32
|
|
|
|
s = self.SELECTOR()
|
|
self.addCleanup(s.close)
|
|
|
|
for i in range(NUM_FDS // 2):
|
|
try:
|
|
rd, wr = self.make_socketpair()
|
|
except OSError:
|
|
# too many FDs, skip - note that we should only catch EMFILE
|
|
# here, but apparently *BSD and Solaris can fail upon connect()
|
|
# or bind() with EADDRNOTAVAIL, so let's be safe
|
|
self.skipTest("FD limit reached")
|
|
|
|
try:
|
|
s.register(rd, selectors.EVENT_READ)
|
|
s.register(wr, selectors.EVENT_WRITE)
|
|
except OSError as e:
|
|
if e.errno == errno.ENOSPC:
|
|
# this can be raised by epoll if we go over
|
|
# fs.epoll.max_user_watches sysctl
|
|
self.skipTest("FD limit reached")
|
|
raise
|
|
|
|
self.assertEqual(NUM_FDS // 2, len(s.select()))
|
|
|
|
|
|
class DefaultSelectorTestCase(BaseSelectorTestCase):
|
|
|
|
SELECTOR = selectors.DefaultSelector
|
|
|
|
|
|
class SelectSelectorTestCase(BaseSelectorTestCase):
|
|
|
|
SELECTOR = selectors.SelectSelector
|
|
|
|
|
|
@unittest.skipUnless(hasattr(selectors, 'PollSelector'),
|
|
"Test needs selectors.PollSelector")
|
|
class PollSelectorTestCase(BaseSelectorTestCase, ScalableSelectorMixIn):
|
|
|
|
SELECTOR = getattr(selectors, 'PollSelector', None)
|
|
|
|
|
|
@unittest.skipUnless(hasattr(selectors, 'EpollSelector'),
|
|
"Test needs selectors.EpollSelector")
|
|
class EpollSelectorTestCase(BaseSelectorTestCase, ScalableSelectorMixIn):
|
|
|
|
SELECTOR = getattr(selectors, 'EpollSelector', None)
|
|
|
|
|
|
@unittest.skipUnless(hasattr(selectors, 'KqueueSelector'),
|
|
"Test needs selectors.KqueueSelector)")
|
|
class KqueueSelectorTestCase(BaseSelectorTestCase, ScalableSelectorMixIn):
|
|
|
|
SELECTOR = getattr(selectors, 'KqueueSelector', None)
|
|
|
|
|
|
@unittest.skipUnless(hasattr(selectors, 'DevpollSelector'),
|
|
"Test needs selectors.DevpollSelector")
|
|
class DevpollSelectorTestCase(BaseSelectorTestCase, ScalableSelectorMixIn):
|
|
|
|
SELECTOR = getattr(selectors, 'DevpollSelector', None)
|
|
|
|
|
|
|
|
def test_main():
|
|
tests = [DefaultSelectorTestCase, SelectSelectorTestCase,
|
|
PollSelectorTestCase, EpollSelectorTestCase,
|
|
KqueueSelectorTestCase, DevpollSelectorTestCase]
|
|
support.run_unittest(*tests)
|
|
support.reap_children()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|