瀏覽代碼

Add test for SelectPoller.unregister()

Mike Naberezny 10 年之前
父節點
當前提交
bfea3767f8
共有 1 個文件被更改,包括 12 次插入0 次删除
  1. 12 0
      supervisor/tests/test_poller.py

+ 12 - 0
supervisor/tests/test_poller.py

@@ -30,6 +30,18 @@ class SelectPollerTests(unittest.TestCase):
         poller.register_writable(7)
         self.assertEqual(sorted(poller.writables), [6,7])
 
+    def test_unregister(self):
+        poller = self._makeOne(DummyOptions())
+        poller.register_readable(6)
+        poller.register_readable(7)
+        poller.register_writable(8)
+        poller.register_writable(9)
+        poller.unregister(6)
+        poller.unregister(9)
+        poller.unregister(100)  # not registered, ignore error
+        self.assertEqual(list(poller.readables), [7])
+        self.assertEqual(list(poller.writables), [8])
+
     def test_poll_returns_readables_and_writables(self):
         _select = DummySelect(result={'readables': [6],
                                       'writables': [8]})