def Target(retry_params): api_utils.set_default_retry_params(retry_params) with cv: set_count[0] += 1 if set_count[0] != 3: cv.wait() cv.notify() self.assertEqual(retry_params, api_utils._get_default_retry_params())
def testPerRequestSetting(self): os.environ['REQUEST_LOG_ID'] = '1' retry_params = api_utils.RetryParams(max_retries=1000) api_utils.set_default_retry_params(retry_params) self.assertEqual(retry_params, api_utils._get_default_retry_params()) os.environ['REQUEST_LOG_ID'] = '2' self.assertEqual(api_utils.RetryParams(), api_utils._get_default_retry_params())