greentest.py: CountingHuB: do not assume that switch() is argument-less

7ce0b531 · Denis Bilenko · 0001ccf1 · 7ce0b531
Commit 7ce0b531 authored Aug 29, 2011 by Denis Bilenko
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

greentest/greentest.py greentest/greentest.py +2 -2

No files found.
--- a/greentest/greentest.py
+++ b/greentest/greentest.py
@@ -262,9 +262,9 @@ class CountingHub(_original_Hub):

    switch_count = 0

-    def switch(self):
+    def switch(self, *args):
        self.switch_count += 1
-        return _original_Hub.switch(self)
+        return _original_Hub.switch(self, *args)

 if gettotalrefcount is None:
    gevent.hub.Hub = CountingHub