Fix throttle applied to methods

2015-10-11 10:42:42 -07:00 · 2015-10-11 10:42:42 -07:00 · dcfc91e71c
commit dcfc91e71c
parent c2117b3eaf
2 changed files with 29 additions and 4 deletions
--- a/tests/util/test_init.py
+++ b/tests/util/test_init.py
@ -229,3 +229,16 @@ class TestUtil(unittest.TestCase):

        self.assertTrue(Tester().hello())
        self.assertTrue(Tester().hello())
+
+    def test_throttle_on_method(self):
+        """ Test that throttle works when wrapping a method. """
+
+        class Tester(object):
+            def hello(self):
+                return True
+
+        tester = Tester()
+        throttled = util.Throttle(timedelta(seconds=1))(tester.hello)
+
+        self.assertTrue(throttled())
+        self.assertIsNone(throttled())