2020-03-22 13:12:42 +01:00
|
|
|
import random
|
|
|
|
import threading
|
2022-05-26 00:10:07 +02:00
|
|
|
import time
|
2020-03-22 13:12:42 +01:00
|
|
|
|
|
|
|
from botocore.retries import quota
|
2022-05-26 00:10:07 +02:00
|
|
|
from tests import unittest
|
2020-03-22 13:12:42 +01:00
|
|
|
|
|
|
|
|
|
|
|
class TestRetryQuota(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
|
|
self.max_capacity = 50
|
|
|
|
self.retry_quota = quota.RetryQuota(self.max_capacity)
|
|
|
|
self.shutdown_threads = False
|
|
|
|
self.seen_capacities = []
|
|
|
|
|
|
|
|
def run_in_thread(self):
|
|
|
|
while not self.shutdown_threads:
|
|
|
|
capacity = random.randint(1, self.max_capacity)
|
|
|
|
self.retry_quota.acquire(capacity)
|
|
|
|
self.seen_capacities.append(self.retry_quota.available_capacity)
|
|
|
|
self.retry_quota.release(capacity)
|
|
|
|
self.seen_capacities.append(self.retry_quota.available_capacity)
|
|
|
|
|
|
|
|
def test_capacity_stays_within_range(self):
|
|
|
|
# The main thing we're after is that the available_capacity
|
|
|
|
# should always be 0 <= capacity <= max_capacity.
|
|
|
|
# So what we do is spawn a number of threads and them acquire
|
|
|
|
# random capacity. They'll check that they never see an invalid
|
|
|
|
# capacity.
|
|
|
|
threads = []
|
|
|
|
for i in range(10):
|
|
|
|
threads.append(threading.Thread(target=self.run_in_thread))
|
|
|
|
for thread in threads:
|
|
|
|
thread.start()
|
|
|
|
# We'll let things run for a second.
|
|
|
|
time.sleep(1)
|
|
|
|
self.shutdown_threads = True
|
|
|
|
for thread in threads:
|
|
|
|
thread.join()
|
|
|
|
for seen_capacity in self.seen_capacities:
|
|
|
|
self.assertGreaterEqual(seen_capacity, 0)
|
|
|
|
self.assertLessEqual(seen_capacity, self.max_capacity)
|