def test_basic(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
check_mpoll(mp)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
mp.add_callback(READABLE, callback)
check_mpoll(mp)
called.wait(0.01)
self.assertEqual(cbargs, [])
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE)])
self.assertEqual(s2.recv(10), b'x')
del cbargs[:]
called.clear()
called.wait(0.01)
self.assertEqual(cbargs, [])
mp.close()
check_mpoll(mp)
s1.close()
s2.close()
def test_update(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
h1 = mp.add_callback(READABLE, callback)
check_mpoll(mp)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE)])
del cbargs[:]; called.clear()
mp.update_callback(h1, READABLE|WRITABLE)
check_mpoll(mp)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE|WRITABLE)])
del cbargs[:]; called.clear()
mp.update_callback(h1, WRITABLE)
check_mpoll(mp)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, WRITABLE)])
del cbargs[:]; called.clear()
mp.update_callback(h1, 0)
check_mpoll(mp)
s1.send(b'x')
called.wait(0.01)
self.assertEqual(cbargs, [])
mp.close()
check_mpoll(mp)
s1.close(); s2.close()
def test_multiple(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
cbargs = []
called = Event()
def callback(arg=0):
def _callback(fd, events):
cbargs.append((fd, events, arg))
called.set()
return _callback
mp.add_callback(READABLE, callback(0))
check_mpoll(mp)
mp.add_callback(READABLE, callback(1))
check_mpoll(mp)
mp.add_callback(WRITABLE, callback(2))
check_mpoll(mp)
mp.add_callback(WRITABLE, callback(3))
check_mpoll(mp)
called.wait(0.1)
self.assertEqual(cbargs, [(fd, WRITABLE, 2), (fd, WRITABLE, 3)])
del cbargs[:]
called.clear()
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE, 0), (fd, READABLE, 1),
(fd, WRITABLE, 2), (fd, WRITABLE, 3)])
self.assertEqual(s2.recv(10), b'x')
mp.close()
check_mpoll(mp)
s1.close()
s2.close()
def test_multiple(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
cbargs = []
called = Event()
def callback(arg=0):
def _callback(fd, events):
cbargs.append((fd, events, arg))
called.set()
return _callback
mp.add_callback(READABLE, callback(0))
check_mpoll(mp)
mp.add_callback(READABLE, callback(1))
check_mpoll(mp)
mp.add_callback(WRITABLE, callback(2))
check_mpoll(mp)
mp.add_callback(WRITABLE, callback(3))
check_mpoll(mp)
called.wait(0.1)
self.assertEqual(cbargs, [(fd, WRITABLE, 2), (fd, WRITABLE, 3)])
del cbargs[:]; called.clear()
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE, 0), (fd, READABLE, 1),
(fd, WRITABLE, 2), (fd, WRITABLE, 3)])
self.assertEqual(s2.recv(10), b'x')
mp.close()
check_mpoll(mp)
s1.close(); s2.close()
def test_remove(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
cbargs = []
called = Event()
def callback(arg=0):
def _callback(fd, events):
cbargs.append((fd, events, arg))
called.set()
return _callback
h1 = mp.add_callback(READABLE, callback(0))
check_mpoll(mp)
h2 = mp.add_callback(READABLE, callback(1))
check_mpoll(mp)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE, 0), (fd, READABLE, 1)])
del cbargs[:]
called.clear()
mp.remove_callback(h1)
check_mpoll(mp)
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE, 1)])
mp.remove_callback(h2)
check_mpoll(mp)
mp.close()
check_mpoll(mp)
s1.close()
s2.close()
def test_remove(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
cbargs = []
called = Event()
def callback(arg=0):
def _callback(fd, events):
cbargs.append((fd, events, arg))
called.set()
return _callback
h1 = mp.add_callback(READABLE, callback(0))
check_mpoll(mp)
h2 = mp.add_callback(READABLE, callback(1))
check_mpoll(mp)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE, 0), (fd, READABLE, 1)])
del cbargs[:]; called.clear()
mp.remove_callback(h1)
check_mpoll(mp)
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE, 1)])
mp.remove_callback(h2)
check_mpoll(mp)
mp.close()
check_mpoll(mp)
s1.close(); s2.close()
def test_update(self):
poll = Poller(get_hub().loop)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
s1, s2 = socketpair()
fd = s2.fileno()
handle = poll.add_callback(fd, WRITABLE, callback)
self.assertIsNotNone(handle)
called.wait(0.1)
self.assertEqual(cbargs, [(fd, WRITABLE)])
del cbargs[:]
called.clear()
poll.update_callback(fd, handle, READABLE | WRITABLE)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE | WRITABLE)])
del cbargs[:]
called.clear()
poll.close()
s1.close()
s2.close()
def test_close(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
h1 = mp.add_callback(READABLE, callback)
h2 = mp.add_callback(READABLE, callback)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE), (fd, READABLE)])
del cbargs[:]
called.clear()
mp.close()
called.wait(0.01)
self.assertEqual(cbargs, [])
self.assertRaises(RuntimeError, mp.add_callback, READABLE, callback)
self.assertRaises(RuntimeError, mp.remove_callback, h1)
self.assertRaises(RuntimeError, mp.remove_callback, h2)
self.assertRaises(RuntimeError, mp.update_callback, h1, WRITABLE)
self.assertRaises(RuntimeError, mp.update_callback, h2, WRITABLE)
s1.close()
s2.close()
def make_socketpair(self):
rd, wr = socketpair()
# Make non-blocking so we get errors if the
# sockets are interacted with but not ready.
rd.settimeout(0.0)
wr.settimeout(0.0)
self.addCleanup(rd.close)
self.addCleanup(wr.close)
return rd, wr
def test_multiple_fds(self):
poll = Poller(get_hub().loop)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
s11, s12 = socketpair()
fd1 = s12.fileno()
poll.add_callback(fd1, READABLE, callback)
s21, s22 = socketpair()
fd2 = s22.fileno()
poll.add_callback(fd2, READABLE, callback)
s11.send(b'x')
s21.send(b'x')
called.wait()
self.assertEqual(cbargs, [(fd1, READABLE), (fd2, READABLE)])
poll.close()
s11.close(); s12.close()
s21.close(); s22.close()
def test_multiple_fds(self):
poll = Poller(get_hub().loop)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
s11, s12 = socketpair()
fd1 = s12.fileno()
poll.add_callback(fd1, READABLE, callback)
s21, s22 = socketpair()
fd2 = s22.fileno()
poll.add_callback(fd2, READABLE, callback)
s11.send(b'x')
s21.send(b'x')
called.wait()
self.assertEqual(cbargs, [(fd1, READABLE), (fd2, READABLE)])
poll.close()
s11.close()
s12.close()
s21.close()
s22.close()
def test_pass_ssl_handle_over_ssl(self):
server = IpcServer(self.ssl_s_args['ssl'])
pipe = self.pipename()
server.listen(pipe, ipc=True, **self.ssl_s_args)
client = IpcClient()
client.connect(pipe, ipc=True, **self.ssl_cp_args)
self.assertEqual(client.call('type'), 'pipe')
s1, s2 = socketpair()
h2 = pyuv.Pipe(get_hub().loop)
h2.open(s2.fileno())
client.call('ssl_handle', h2)
c1 = IpcClient()
c1.connect(s1.fileno(), **self.ssl_cp_args)
self.assertEqual(c1.call('ping'), 'pong')
self.assertEqual(c1.call('type'), 'tcp')
s1.close(); s2.close()
c1.close(); h2.close()
client.close(); server.close()
def test_close(self):
poll = Poller(get_hub().loop)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
s1, s2 = socketpair()
fd = s2.fileno()
handle = poll.add_callback(fd, READABLE, callback)
self.assertIsNotNone(handle)
s1.send(b'x')
poll.close()
called.wait(0.01)
self.assertEqual(cbargs, [])
self.assertRaises(RuntimeError, poll.add_callback, fd, READABLE, callback)
self.assertRaises(RuntimeError, poll.remove_callback, fd, handle)
self.assertRaises(RuntimeError, poll.update_callback, fd, handle, WRITABLE)
s1.close(); s2.close()
def test_debug(self):
if not __debug__:
return
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
def callback(fd, events):
pass
mp.add_callback(0, callback)
mp.add_callback(READABLE, callback)
mp.add_callback(WRITABLE, callback)
mp.add_callback(READABLE|WRITABLE, callback)
check(mp)
with capture_stdio() as (out, err):
dump(mp)
lines = out.readlines()
self.assertGreater(len(lines), 0)
self.assertEqual(err.read(), '')
mp.close()
s1.close(); s2.close()
def test_update(self):
poll = Poller(get_hub().loop)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
s1, s2 = socketpair()
fd = s2.fileno()
handle = poll.add_callback(fd, WRITABLE, callback)
self.assertIsNotNone(handle)
called.wait(0.1)
self.assertEqual(cbargs, [(fd, WRITABLE)])
del cbargs[:]; called.clear()
poll.update_callback(fd, handle, READABLE|WRITABLE)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE|WRITABLE)])
del cbargs[:]; called.clear()
poll.close()
s1.close(); s2.close()
def test_basic(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
check(mp)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
mp.add_callback(READABLE, callback)
check(mp)
called.wait(0.01)
self.assertEqual(cbargs, [])
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE)])
self.assertEqual(s2.recv(10), b'x')
del cbargs[:]; called.clear()
called.wait(0.01)
self.assertEqual(cbargs, [])
mp.close()
check(mp)
s1.close(); s2.close()
def test_close(self):
poll = Poller(get_hub().loop)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
s1, s2 = socketpair()
fd = s2.fileno()
handle = poll.add_callback(fd, READABLE, callback)
self.assertIsNotNone(handle)
s1.send(b'x')
poll.close()
called.wait(0.01)
self.assertEqual(cbargs, [])
self.assertRaises(RuntimeError, poll.add_callback, fd, READABLE,
callback)
self.assertRaises(RuntimeError, poll.remove_callback, fd, handle)
self.assertRaises(RuntimeError, poll.update_callback, fd, handle,
WRITABLE)
s1.close()
s2.close()
def test_close(self):
s1, s2 = socketpair()
fd = s2.fileno()
mp = MultiPoll(get_hub().loop, fd)
cbargs = []
called = Event()
def callback(fd, events):
cbargs.append((fd, events))
called.set()
h1 = mp.add_callback(READABLE, callback)
h2 = mp.add_callback(READABLE, callback)
s1.send(b'x')
called.wait(0.1)
self.assertEqual(cbargs, [(fd, READABLE), (fd, READABLE)])
del cbargs[:]; called.clear()
mp.close()
called.wait(0.01)
self.assertEqual(cbargs, [])
self.assertRaises(RuntimeError, mp.add_callback, READABLE, callback)
self.assertRaises(RuntimeError, mp.remove_callback, h1)
self.assertRaises(RuntimeError, mp.remove_callback, h2)
self.assertRaises(RuntimeError, mp.update_callback, h1, WRITABLE)
self.assertRaises(RuntimeError, mp.update_callback, h2, WRITABLE)
s1.close(); s2.close()
