def test_allow_default(self):
self.throttle = Throttle(
self.request,
self.auth,
cache=self.cache,
rate=(1, timedelta(seconds=60)),
)
self.test_allow(wait=(67, ))
def test_allow_period_not_evenly_divisible(self):
self.throttle = Throttle(
self.request,
self.auth,
bucket=timedelta(seconds=10),
cache=self.cache,
rate=(1, timedelta(seconds=61)),
)
self.test_allow(wait=(71, 81))
class ThrottleBaseTest(TestCaseNoDB):
def setUp(self):
request = self.make_request()
self.throttle = Throttle(request, Auth(request))
def test_allow(self):
self.assertEquals(self.throttle.allow(), (True, {}))
def test_allow_suffix(self):
self.assertEquals(self.throttle.allow('suffix'), (True, {}))
class ThrottleBaseTest(TestCaseNoDB):
def setUp(self):
request = self.make_request()
self.throttle = Throttle(request, Auth(request))
def test_allow(self):
self.assertEquals(self.throttle.allow(), (True, {}))
def test_allow_suffix(self):
self.assertEquals(self.throttle.allow('suffix'), (True, {}))
def test_allow_default(self):
self.throttle = Throttle(
self.request,
self.auth,
cache=self.cache,
rate=(1, timedelta(seconds=60)),
)
self.test_allow(wait=(67,))
def test_allow_period_not_evenly_divisible(self):
self.throttle = Throttle(
self.request,
self.auth,
bucket=timedelta(seconds=10),
cache=self.cache,
rate=(1, timedelta(seconds=61)),
)
self.test_allow(wait=(71, 81))
class ThrottleRateTest(TestCaseNoDB):
def setUp(self):
self.request = self.make_request()
self.auth = Auth(self.request)
self.auth.authned = self.auth.AuthnedInterface(FakeUser())
self.cache = cache
self.make_throttle = partial(
Throttle,
bucket=timedelta(seconds=2),
cache=self.cache,
rate=(1, timedelta(seconds=10)),
)
self.throttle = self.make_throttle(self.request, self.auth)
self.now = self.throttle.timer()
self.throttle.timer = lambda: self.now
# make sure cache is clear
self.cache.clear()
def test_allow(self, wait=None):
wait = wait or (13, )
self.assertEquals(self.throttle.allow(), (True, {}))
allowed, headers = self.throttle.allow()
self.assertEquals(allowed, False)
nreq, nsec = self.throttle.rate
self.assertTrue(headers['Retry-After'] in wait)
self.throttle.timer = lambda: self.now + headers['Retry-After']
allowed, headers = self.throttle.allow()
self.assertEquals(allowed, True)
self.assertTrue(not headers)
def test_allow_anon(self):
request = self.make_request()
self.assertEquals(
self.make_throttle(request, Auth(request)).allow(), (False, {})
)
def test_allow_default(self):
self.throttle = Throttle(
self.request,
self.auth,
cache=self.cache,
rate=(1, timedelta(seconds=60)),
)
self.test_allow(wait=(67,))
def test_allow_period_not_evenly_divisible(self):
self.throttle = Throttle(
self.request,
self.auth,
bucket=timedelta(seconds=10),
cache=self.cache,
rate=(1, timedelta(seconds=61)),
)
self.test_allow(wait=(71, 81))
def test_allow_suffix(self, wait=13):
# no suffix
self.assertEquals(self.throttle.allow(), (True, {}))
allowed, headers = self.throttle.allow()
self.assertEquals(allowed, False)
nreq, nsec = self.throttle.rate
self.assertEquals(headers['Retry-After'], wait)
self.throttle.timer = lambda: self.now + wait
allowed, headers = self.throttle.allow()
self.assertEquals(allowed, True)
self.assertTrue(not headers)
# suffix
self.throttle.timer = lambda: self.now
self.assertEquals(
self.throttle.allow(suffix='suffix'), (True, {})
)
allowed, headers = self.throttle.allow(suffix='suffix')
self.assertEquals(allowed, False)
nreq, nsec = self.throttle.rate
self.assertEquals(headers['Retry-After'], wait)
self.throttle.timer = lambda: self.now + wait
allowed, headers = self.throttle.allow(suffix='suffix')
self.assertEquals(allowed, True)
self.assertTrue(not headers)
def setUp(self):
request = self.make_request()
self.throttle = Throttle(request, Auth(request))
class ThrottleRateTest(TestCaseNoDB):
def setUp(self):
self.request = self.make_request()
self.auth = Auth(self.request)
self.auth.authned = self.auth.AuthnedInterface(FakeUser())
self.cache = cache
self.make_throttle = partial(
Throttle,
bucket=timedelta(seconds=2),
cache=self.cache,
rate=(1, timedelta(seconds=10)),
)
self.throttle = self.make_throttle(self.request, self.auth)
self.now = self.throttle.timer()
self.throttle.timer = lambda: self.now
# make sure cache is clear
self.cache.clear()
def test_allow(self, wait=None):
wait = wait or (13, )
self.assertEquals(self.throttle.allow(), (True, {}))
allowed, headers = self.throttle.allow()
self.assertEquals(allowed, False)
nreq, nsec = self.throttle.rate
self.assertTrue(headers['Retry-After'] in wait)
self.throttle.timer = lambda: self.now + headers['Retry-After']
allowed, headers = self.throttle.allow()
self.assertEquals(allowed, True)
self.assertTrue(not headers)
def test_allow_anon(self):
request = self.make_request()
self.assertEquals(
self.make_throttle(request, Auth(request)).allow(), (False, {}))
def test_allow_default(self):
self.throttle = Throttle(
self.request,
self.auth,
cache=self.cache,
rate=(1, timedelta(seconds=60)),
)
self.test_allow(wait=(67, ))
def test_allow_period_not_evenly_divisible(self):
self.throttle = Throttle(
self.request,
self.auth,
bucket=timedelta(seconds=10),
cache=self.cache,
rate=(1, timedelta(seconds=61)),
)
self.test_allow(wait=(71, 81))
def test_allow_suffix(self, wait=13):
# no suffix
self.assertEquals(self.throttle.allow(), (True, {}))
allowed, headers = self.throttle.allow()
self.assertEquals(allowed, False)
nreq, nsec = self.throttle.rate
self.assertEquals(headers['Retry-After'], wait)
self.throttle.timer = lambda: self.now + wait
allowed, headers = self.throttle.allow()
self.assertEquals(allowed, True)
self.assertTrue(not headers)
# suffix
self.throttle.timer = lambda: self.now
self.assertEquals(self.throttle.allow(suffix='suffix'), (True, {}))
allowed, headers = self.throttle.allow(suffix='suffix')
self.assertEquals(allowed, False)
nreq, nsec = self.throttle.rate
self.assertEquals(headers['Retry-After'], wait)
self.throttle.timer = lambda: self.now + wait
allowed, headers = self.throttle.allow(suffix='suffix')
self.assertEquals(allowed, True)
self.assertTrue(not headers)
def setUp(self):
request = self.make_request()
self.throttle = Throttle(request, Auth(request))
