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 = {}