def test_with_suffix(self): http_helpers.set_user_agent_suffix("foobar") mock_http = MockHttp() http = http_helpers.build_http(mock_http) response, content = http.request(DUMMY_URL) self.assertIn(UA_KEY, mock_http.headers) self.assertRegex(mock_http.headers[UA_KEY], r'foobar\s+gce$')
def test_without_suffix(self): http_helpers.set_user_agent_suffix("") mock_http = MockHttp() http = http_helpers.build_http(mock_http) response, content = http.request(DUMMY_URL) self.assertIn(UA_KEY, mock_http.headers) self.assertRegex( mock_http.headers[UA_KEY], r'forseti-security/[0-9.]+$')
def test_with_suffix_set_platform(self): with mock.patch.dict('os.environ', {'FORSETI_PLATFORM': 'gke'}): http_helpers.set_user_agent_suffix("foobar") mock_http = MockHttp() http = http_helpers.build_http(mock_http) response, content = http.request(DUMMY_URL) self.assertIn(UA_KEY, mock_http.headers) self.assertRegex(mock_http.headers[UA_KEY], r'foobar\s+gke$')
def http(self): """A thread local instance of httplib2.Http. Returns: google_auth_httplib2.AuthorizedHttp: An Http instance authorized by the credentials. """ if self._use_cached_http and hasattr(self._local, 'http'): return self._local.http authorized_http = google_auth_httplib2.AuthorizedHttp( self._credentials, http=http_helpers.build_http()) if self._use_cached_http: self._local.http = authorized_http return authorized_http