def test_update_writers(self): hub = Hub() P = hub.poller = Mock() write_A = Mock() write_B = Mock() hub.update_writers({20: write_A, File(21): write_B}) P.register.assert_has_calls([ call(20, hub.WRITE), call(File(21), hub.WRITE), ], any_order=True) self.assertIs(hub.writers[20], write_A) self.assertIs(hub.writers[21], write_B) hub.remove(20) self.assertNotIn(20, hub.writers) hub.remove(File(21)) self.assertNotIn(21, hub.writers) P.unregister.assert_has_calls([ call(20), call(21), ])
def test_update_readers(self): hub = Hub() P = hub.poller = Mock() read_A = Mock() read_B = Mock() hub.update_readers({10: read_A, File(11): read_B}) P.register.assert_has_calls([ call(10, hub.READ | hub.ERR), call(File(11), hub.READ | hub.ERR), ], any_order=True) self.assertIs(hub.readers[10], read_A) self.assertIs(hub.readers[11], read_B) hub.remove(10) self.assertNotIn(10, hub.readers) hub.remove(File(11)) self.assertNotIn(11, hub.readers) P.unregister.assert_has_calls([ call(10), call(11), ])
def test_update_writers(self): hub = Hub() P = hub.poller = Mock() write_A = Mock() write_B = Mock() hub.update_writers({20: write_A, File(21): write_B}) P.register.assert_has_calls([call(20, hub.WRITE), call(File(21), hub.WRITE)], any_order=True) self.assertIs(hub.writers[20], write_A) self.assertIs(hub.writers[21], write_B) hub.remove(20) self.assertNotIn(20, hub.writers) hub.remove(File(21)) self.assertNotIn(21, hub.writers) P.unregister.assert_has_calls([call(20), call(21)])
def test_update_readers(self): hub = Hub() P = hub.poller = Mock() read_A = Mock() read_B = Mock() hub.update_readers({10: read_A, File(11): read_B}) P.register.assert_has_calls([call(10, hub.READ | hub.ERR), call(File(11), hub.READ | hub.ERR)], any_order=True) self.assertIs(hub.readers[10], read_A) self.assertIs(hub.readers[11], read_B) hub.remove(10) self.assertNotIn(10, hub.readers) hub.remove(File(11)) self.assertNotIn(11, hub.readers) P.unregister.assert_has_calls([call(10), call(11)])
def test_remove__unregister_raises(self): hub = Hub() hub.poller = Mock() hub.poller.unregister.side_effect = OSError() hub.remove(313)
def test_can_remove_unknown_fds(self): hub = Hub() hub.poller = Mock() hub.remove(30) hub.remove(File(301))