Fix RetriableHttp

infra_libs/test/httplib2_utils_test.py

 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import os
 import socket
 import time
 import unittest
 
 import infra_libs
@@ skipping 120 matching lines @@
 
 class RetriableHttplib2Test(unittest.TestCase):
   def setUp(self):
     super(RetriableHttplib2Test, self).setUp()
     self.http = infra_libs.RetriableHttp(httplib2.Http())
     self.http._http.request = mock.create_autospec(self.http._http.request,
                                                    spec_set=True)
 
   _MOCK_REQUEST = mock.call('http://foo/', 'GET', None)
 
+  def test_authorize(self):
+    http = infra_libs.RetriableHttp(httplib2.Http())
+    creds = infra_libs.get_signed_jwt_assertion_credentials(
+      'valid_creds.json',
+      service_accounts_creds_root=DATA_DIR)
+    creds.authorize(http)
+
   def test_delegate_get_attr(self):
     """RetriableHttp should delegate getting attribute except request() to
        Http"""
     self.http._http.clear_credentials = mock.create_autospec(
         self.http._http.clear_credentials, spec_set=True)
     self.http.clear_credentials()
     self.http._http.clear_credentials.assert_called_once_with()
 
   def test_delegate_set_attr(self):
     """RetriableHttp should delegate setting attributes to Http"""
@@ skipping 193 matching lines @@
                                 ('.*', {'status': 404}, '')])
     response, body = http.request('https://mywebserver.woo.hoo', 'GET')
     self.assertIsInstance(response, httplib2.Response)
     self.assertEqual(response.status, 404)
     self.assertEqual(body, '')
 
     self.assertEqual(http.requests_made[0].uri, 'https://mywebserver.woo.hoo')
     self.assertEqual(http.requests_made[0].method, 'GET')
     self.assertEqual(http.requests_made[0].body, None)
     self.assertEqual(http.requests_made[0].headers, None)