def test_release_limit(self): mock_redis = Mock(spec=redis.Redis) limiter = Limiter(limit=10, redis=mock_redis, expiration_in_seconds=20) limiter.release_limit(key='test1') mock_redis.zrem.assert_called_once_with('retools_limiter', 'test1')
def test_release_limit(self): mock_redis = Mock(spec=redis.Redis) limiter = Limiter(limit=10, redis=mock_redis, expiration_in_seconds=20) limiter.release_limit(key='test1') mock_redis.zrem.assert_called_once_with('retools_limiter', 'test1')
def test_has_limit_after_releasing_items(self): limiter = Limiter(prefix='test-%.6f' % time.time(), limit=2, expiration_in_seconds=400, redis=self.redis) has_limit = limiter.acquire_limit(key='test1') eq_(has_limit, True) has_limit = limiter.acquire_limit(key='test2') eq_(has_limit, True) limiter.release_limit(key='test2') has_limit = limiter.acquire_limit(key='test3') eq_(has_limit, True)
def test_has_limit_after_releasing_items(self): limiter = Limiter(prefix='test-%.6f' % time.time(), limit=2, expiration_in_seconds=400) has_limit = limiter.acquire_limit(key='test1') eq_(has_limit, True) has_limit = limiter.acquire_limit(key='test2') eq_(has_limit, True) limiter.release_limit(key='test2') has_limit = limiter.acquire_limit(key='test3') eq_(has_limit, True)