1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
from ratelimit import limits, RateLimitException
from tests import unittest, clock
class TestDecorator(unittest.TestCase):
@limits(calls=1, period=10, clock=clock)
def increment(self):
'''
Increment the counter at most once every 10 seconds.
'''
self.count += 1
@limits(calls=1, period=10, clock=clock, raise_on_limit=False)
def increment_no_exception(self):
'''
Increment the counter at most once every 10 seconds, but w/o rasing an
exception when reaching limit.
'''
self.count += 1
def setUp(self):
self.count = 0
clock.increment(10)
def test_increment(self):
self.increment()
self.assertEqual(self.count, 1)
def test_exception(self):
self.increment()
self.assertRaises(RateLimitException, self.increment)
def test_reset(self):
self.increment()
clock.increment(10)
self.increment()
self.assertEqual(self.count, 2)
def test_no_exception(self):
self.increment_no_exception()
self.increment_no_exception()
self.assertEqual(self.count, 1)
|
