Browse Source

Add test for SelectPoller.unregister()

Mike Naberezny 10 years ago
parent
commit
bfea3767f8
1 changed files with 12 additions and 0 deletions
  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]})