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)
Exemple #2
0
    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 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())
Exemple #4
0
    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)
Exemple #6
0
    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())
Exemple #8
0
 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 __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)
Exemple #10
0
 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)