def test_register_already_registered(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
s = selectors._BaseSelector()
s.register(fobj, selectors.EVENT_READ)
self.assertRaises(ValueError, s.register, fobj, selectors.EVENT_READ)
def test_key_from_fd(self, m_log):
s = selectors._BaseSelector()
key = s.register(1, selectors.EVENT_READ)
self.assertIs(key, s._key_from_fd(1))
self.assertIsNone(s._key_from_fd(10))
m_log.warning.assert_called_with('No key found for fd %r', 10)
def test_key_from_fd(self, m_log):
s = selectors._BaseSelector()
key = s.register(1, selectors.EVENT_READ)
self.assertIs(key, s._key_from_fd(1))
self.assertIsNone(s._key_from_fd(10))
m_log.warning.assert_called_with('No key found for fd %r', 10)
def test_close(self):
s = selectors._BaseSelector()
s.register(1, selectors.EVENT_READ)
s.close()
self.assertFalse(s._fd_to_key)
self.assertFalse(s._fileobj_to_key)
def test_register_already_registered(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
s = selectors._BaseSelector()
s.register(fobj, selectors.EVENT_READ)
self.assertRaises(ValueError, s.register, fobj, selectors.EVENT_READ)
def test_close(self):
s = selectors._BaseSelector()
s.register(1, selectors.EVENT_READ)
s.close()
self.assertFalse(s._fd_to_key)
self.assertFalse(s._fileobj_to_key)
def test_context_manager(self):
s = selectors._BaseSelector()
with s as sel:
sel.register(1, selectors.EVENT_READ)
self.assertFalse(s._fd_to_key)
self.assertFalse(s._fileobj_to_key)
def test_context_manager(self):
s = selectors._BaseSelector()
with s as sel:
sel.register(1, selectors.EVENT_READ)
self.assertFalse(s._fd_to_key)
self.assertFalse(s._fileobj_to_key)
def test_registered_count(self):
s = selectors._BaseSelector()
self.assertEqual(0, s.registered_count())
s.register(1, selectors.EVENT_READ)
self.assertEqual(1, s.registered_count())
s.unregister(1)
self.assertEqual(0, s.registered_count())
def test_unregister(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
s = selectors._BaseSelector()
s.register(fobj, selectors.EVENT_READ)
s.unregister(fobj)
self.assertFalse(s._fd_to_key)
self.assertFalse(s._fileobj_to_key)
def test_register(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
s = selectors._BaseSelector()
key = s.register(fobj, selectors.EVENT_READ)
self.assertIsInstance(key, selectors.SelectorKey)
self.assertEqual(key.fd, 10)
self.assertIs(key, s._fd_to_key[10])
def test_registered_count(self):
s = selectors._BaseSelector()
self.assertEqual(0, s.registered_count())
s.register(1, selectors.EVENT_READ)
self.assertEqual(1, s.registered_count())
s.unregister(1)
self.assertEqual(0, s.registered_count())
def test_modify(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
s = selectors._BaseSelector()
key = s.register(fobj, selectors.EVENT_READ)
key2 = s.modify(fobj, selectors.EVENT_WRITE)
self.assertNotEqual(key.events, key2.events)
self.assertEqual((selectors.EVENT_WRITE, None), s.get_info(fobj))
def test_register(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
s = selectors._BaseSelector()
key = s.register(fobj, selectors.EVENT_READ)
self.assertIsInstance(key, selectors.SelectorKey)
self.assertEqual(key.fd, 10)
self.assertIs(key, s._fd_to_key[10])
def test_unregister(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
s = selectors._BaseSelector()
s.register(fobj, selectors.EVENT_READ)
s.unregister(fobj)
self.assertFalse(s._fd_to_key)
self.assertFalse(s._fileobj_to_key)
def test_modify(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
s = selectors._BaseSelector()
key = s.register(fobj, selectors.EVENT_READ)
key2 = s.modify(fobj, selectors.EVENT_WRITE)
self.assertNotEqual(key.events, key2.events)
self.assertEqual((selectors.EVENT_WRITE, None), s.get_info(fobj))
def test_modify_same(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
data = object()
s = selectors._BaseSelector()
key = s.register(fobj, selectors.EVENT_READ, data)
key2 = s.modify(fobj, selectors.EVENT_READ, data)
self.assertIs(key, key2)
def test_modify_same(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
data = object()
s = selectors._BaseSelector()
key = s.register(fobj, selectors.EVENT_READ, data)
key2 = s.modify(fobj, selectors.EVENT_READ, data)
self.assertIs(key, key2)
def test_modify_data(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
d1 = object()
d2 = object()
s = selectors._BaseSelector()
key = s.register(fobj, selectors.EVENT_READ, d1)
key2 = s.modify(fobj, selectors.EVENT_READ, d2)
self.assertEqual(key.events, key2.events)
self.assertNotEqual(key.data, key2.data)
self.assertEqual((selectors.EVENT_READ, d2), s.get_info(fobj))
def test_modify_data(self):
fobj = unittest.mock.Mock()
fobj.fileno.return_value = 10
d1 = object()
d2 = object()
s = selectors._BaseSelector()
key = s.register(fobj, selectors.EVENT_READ, d1)
key2 = s.modify(fobj, selectors.EVENT_READ, d2)
self.assertEqual(key.events, key2.events)
self.assertNotEqual(key.data, key2.data)
self.assertEqual((selectors.EVENT_READ, d2), s.get_info(fobj))
def test_register_unknown_event(self):
s = selectors._BaseSelector()
self.assertRaises(ValueError, s.register, unittest.mock.Mock(), 999999)
def test_modify_unknown(self):
s = selectors._BaseSelector()
self.assertRaises(ValueError, s.modify, unittest.mock.Mock(), 1)
def test_unregister_unknown(self):
s = selectors._BaseSelector()
self.assertRaises(ValueError, s.unregister, unittest.mock.Mock())
def test_unregister_unknown(self):
s = selectors._BaseSelector()
self.assertRaises(ValueError, s.unregister, unittest.mock.Mock())
def test_modify_unknown(self):
s = selectors._BaseSelector()
self.assertRaises(ValueError, s.modify, unittest.mock.Mock(), 1)
def test_register_unknown_event(self):
s = selectors._BaseSelector()
self.assertRaises(ValueError, s.register, unittest.mock.Mock(), 999999)
def test_select(self):
s = selectors._BaseSelector()
self.assertRaises(NotImplementedError, s.select)
def test_select(self):
s = selectors._BaseSelector()
self.assertRaises(NotImplementedError, s.select)