def test_quota_reached_adds_retry_metadata(self):
quota_checker = standard.RetryQuotaChecker(
quota.RetryQuota(initial_capacity=0))
context = self.create_context()
self.assertFalse(quota_checker.acquire_retry_quota(context))
self.assertEqual(context.get_retry_metadata(),
{'RetryQuotaReached': True})
def test_acquire_quota_fails(self):
quota_checker = standard.RetryQuotaChecker(
quota.RetryQuota(initial_capacity=5))
# The first one succeeds.
self.assertTrue(
quota_checker.acquire_retry_quota(self.create_context()))
# But we should fail now because we're out of quota.
self.request_context.pop('retry_quota_capacity')
self.assertFalse(
quota_checker.acquire_retry_quota(self.create_context()))
self.assertNotIn('retry_quota_capacity', self.request_context)
def setUp(self):
self.quota = quota.RetryQuota(500)
self.quota_checker = standard.RetryQuotaChecker(self.quota)
self.request_context = {}