def test_wait_io_close_is_called(self):
selector = selectors.DefaultSelector()
self.addCleanup(selector.close)
def fake_constructor():
return selector
old_selector = wait.DefaultSelector
wait.DefaultSelector = fake_constructor
self.addCleanup(setattr, wait, "DefaultSelector", old_selector)
rd, wr = self.make_socketpair()
wait.wait_for_write([rd, wr], 0.001)
self.assertIs(selector._map, None)
def make_selector(self):
s = selectors.DefaultSelector()
self.addCleanup(s.close)
return s