From 8fa817af554f99fec64384d3864dda2ae167eb37 Mon Sep 17 00:00:00 2001 From: Guido van Rossum Date: Mon, 2 Dec 2013 17:26:25 -0800 Subject: Incorporate selectors.py refactoring from CPython repo. --- asyncio/test_utils.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'asyncio/test_utils.py') 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. -- cgit v1.2.1