def testDefaultRetryParams(self): if not enable_cloudstorage_tests: return self.assertTrue( api_utils._get_default_retry_params().save_access_token) self.assertEquals("App Engine Python MR", api_utils._get_default_retry_params()._user_agent)
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())
def testInitialization(self): retry_params = api_utils._get_default_retry_params() self.assertEqual('AppEngine-Python-GCS', retry_params._user_agent) user_agent = 'Test User Agent String' retry_params = api_utils.RetryParams(_user_agent=user_agent) self.assertEqual(user_agent, retry_params._user_agent)
def testInitialization(self): retry_params = api_utils._get_default_retry_params() self.assertEqual('App Engine Python GCS Client', retry_params._user_agent) user_agent = 'Test User Agent String' retry_params = api_utils.RetryParams(_user_agent=user_agent) self.assertEqual(user_agent, retry_params._user_agent)
def __init__(self, scopes, service_account_id=None, token_maker=None, retry_params=None): """Constructor. Args: scopes: A scope or a list of scopes. service_account_id: Internal use only. token_maker: An asynchronous function of the form (scopes, service_account_id) -> (token, expires). retry_params: An instance of api_utils.RetryParams. If None, the default for current thread will be used. """ if isinstance(scopes, basestring): scopes = [scopes] self.scopes = scopes self.service_account_id = service_account_id self.make_token_async = token_maker or _make_token_async if not retry_params: retry_params = api_utils._get_default_retry_params() self.retry_params = retry_params self.user_agent = {'User-Agent': retry_params._user_agent} self.expiration_headroom = random.randint(60, 240)
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 testDefaultRetryParams(self): if not enable_cloudstorage_tests: return self.assertTrue( api_utils._get_default_retry_params().save_access_token)
def testDefaultRetryParams(self): if not enable_cloudstorage_tests: return self.assertTrue(api_utils._get_default_retry_params().save_access_token)
def testDefaultRetryParams(self): if not enable_cloudstorage_tests: return self.assertTrue(api_utils._get_default_retry_params().save_access_token) self.assertEquals("App Engine Python MR", api_utils._get_default_retry_params()._user_agent)