Incorporate selectors.py refactoring from CPython repo.

1 files changed, 14 insertions, 0 deletions

diff --git a/asyncio/test_utils.py b/asyncio/test_utils.py
index c278dd1..d7d8442 100644
--- a/asyncio/test_utils.py
+++ b/asyncio/test_utils.py
@@ -142,9 +142,23 @@ def make_test_protocol(base):
 
 class TestSelector(selectors.BaseSelector):
 
+    def __init__(self):
+        self.keys = {}
+
+    def register(self, fileobj, events, data=None):
+        key = selectors.SelectorKey(fileobj, 0, events, data)
+        self.keys[fileobj] = key
+        return key
+
+    def unregister(self, fileobj):
+        return self.keys.pop(fileobj)
+
     def select(self, timeout):
         return []
 
+    def get_map(self):
+        return self.keys
+
 
 class TestLoop(base_events.BaseEventLoop):
     """Loop for unittests.