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_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()