Isolating speech Connection in _http module.

Author: dhermes

speech/google/cloud/speech/_gax.py

@@ -42,7 +42,7 @@ class GAPICSpeechAPI(object):
     """Manage calls through GAPIC wrappers to the Speech API."""
     def __init__(self, client=None):
         self._client = client
-        credentials = self._client._connection.credentials
+        credentials = self._client._credentials
         channel = make_secure_channel(
             credentials, DEFAULT_USER_AGENT,
             SpeechClient.SERVICE_ADDRESS)

speech/google/cloud/speech/_http.py

@@ -57,7 +57,7 @@ class HTTPSpeechAPI(object):
     """
     def __init__(self, client):
         self._client = client
-        self._connection = client._connection
+        self._connection = Connection(client)
 
     def async_recognize(self, sample, language_code=None,
                         max_alternatives=None, profanity_filter=None,

speech/google/cloud/speech/client.py

@@ -20,7 +20,6 @@
 from google.cloud.environment_vars import DISABLE_GRPC
 
 from google.cloud.speech._gax import GAPICSpeechAPI
-from google.cloud.speech._http import Connection
 from google.cloud.speech._http import HTTPSpeechAPI
 from google.cloud.speech.sample import Sample
 
@@ -58,8 +57,6 @@ class Client(BaseClient):
 
     def __init__(self, credentials=None, http=None, use_gax=None):
         super(Client, self).__init__(credentials=credentials, http=http)
-        self._connection = Connection(self)
-
         # Save on the actual client class whether we use GAX or not.
         if use_gax is None:
             self._use_gax = _USE_GAX

speech/unit_tests/test__gax.py

@@ -49,11 +49,16 @@ def test_constructor(self, mocked_stub, mocked_cls, mocked_channel):
         from google.cloud.speech import __version__
         from google.cloud.speech._gax import OPERATIONS_API_HOST
 
+        credentials = _make_credentials()
         mock_cnxn = mock.Mock(
-            credentials=_make_credentials(),
+            credentials=credentials,
             spec=['credentials'],
         )
-        mock_client = mock.Mock(_connection=mock_cnxn, spec=['_connection'])
+        mock_client = mock.Mock(
+            _connection=mock_cnxn,
+            _credentials=credentials,
+            spec=['_connection', '_credentials'],
+        )
 
         speech_api = self._make_one(mock_client)
         self.assertIs(speech_api._client, mock_client)

speech/unit_tests/test_client.py

@@ -90,9 +90,8 @@ def test_ctor(self):
         creds = _make_credentials()
         http = object()
         client = self._make_one(credentials=creds, http=http)
-        self.assertIsInstance(client._connection, Connection)
-        self.assertTrue(client._connection.credentials is creds)
-        self.assertTrue(client._connection.http is http)
+        self.assertTrue(client._credentials is creds)
+        self.assertTrue(client._http is http)
 
     def test_ctor_use_gax_preset(self):
         creds = _make_credentials()
@@ -153,7 +152,9 @@ def test_sync_recognize_content_with_optional_params_no_gax(self):
         }
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials, use_gax=False)
-        client._connection = _Connection(RETURNED)
+        speech_api = client.speech_api
+        connection = _Connection(RETURNED)
+        speech_api._connection = connection
 
         encoding = speech.Encoding.FLAC
 
@@ -165,8 +166,8 @@ def test_sync_recognize_content_with_optional_params_no_gax(self):
                                          profanity_filter=True,
                                          speech_context=self.HINTS)
 
-        self.assertEqual(len(client._connection._requested), 1)
-        req = client._connection._requested[0]
+        self.assertEqual(len(connection._requested), 1)
+        req = connection._requested[0]
         self.assertEqual(len(req), 3)
         self.assertEqual(req['data'], REQUEST)
         self.assertEqual(req['method'], 'POST')
@@ -199,7 +200,9 @@ def test_sync_recognize_source_uri_without_optional_params_no_gax(self):
         }
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials, use_gax=False)
-        client._connection = _Connection(RETURNED)
+        speech_api = client.speech_api
+        connection = _Connection(RETURNED)
+        speech_api._connection = connection
 
         encoding = speech.Encoding.FLAC
 
@@ -208,8 +211,8 @@ def test_sync_recognize_source_uri_without_optional_params_no_gax(self):
 
         response = [i for i in sample.sync_recognize()]
 
-        self.assertEqual(len(client._connection._requested), 1)
-        req = client._connection._requested[0]
+        self.assertEqual(len(connection._requested), 1)
+        req = connection._requested[0]
         self.assertEqual(len(req), 3)
         self.assertEqual(req['data'], REQUEST)
         self.assertEqual(req['method'], 'POST')
@@ -231,7 +234,8 @@ def test_sync_recognize_with_empty_results_no_gax(self):
 
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials, use_gax=False)
-        client._connection = _Connection(SYNC_RECOGNIZE_EMPTY_RESPONSE)
+        speech_api = client.speech_api
+        speech_api._connection = _Connection(SYNC_RECOGNIZE_EMPTY_RESPONSE)
 
         sample = client.sample(source_uri=self.AUDIO_SOURCE_URI,
                                encoding=speech.Encoding.FLAC,
@@ -248,8 +252,7 @@ def test_sync_recognize_with_empty_results_gax(self):
 
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials, use_gax=True)
-        client._connection = _Connection()
-        client._connection.credentials = credentials
+        client._credentials = credentials
 
         channel_args = []
         channel_obj = object()
@@ -291,9 +294,7 @@ def test_sync_recognize_with_gax(self):
 
         creds = _make_credentials()
         client = self._make_one(credentials=creds, use_gax=True)
-        client._connection = _Connection()
-        client._connection.credentials = creds
-        client._speech_api = None
+        client._credentials = creds
 
         alternatives = [{
             'transcript': 'testing 1 2 3',
@@ -352,8 +353,7 @@ def test_async_supported_encodings(self):
         from google.cloud import speech
 
         credentials = _make_credentials()
-        client = self._make_one(credentials=credentials)
-        client._connection = _Connection({})
+        client = self._make_one(credentials=credentials, use_gax=True)
 
         sample = client.sample(source_uri=self.AUDIO_SOURCE_URI,
                                encoding=speech.Encoding.FLAC,
@@ -370,7 +370,8 @@ def test_async_recognize_no_gax(self):
 
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials, use_gax=False)
-        client._connection = _Connection(RETURNED)
+        speech_api = client.speech_api
+        speech_api._connection = _Connection(RETURNED)
 
         sample = client.sample(source_uri=self.AUDIO_SOURCE_URI,
                                encoding=speech.Encoding.LINEAR16,
@@ -393,8 +394,7 @@ def test_async_recognize_with_gax(self):
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials,
                                 use_gax=True)
-        client._connection = _Connection()
-        client._connection.credentials = credentials
+        client._credentials = credentials
 
         channel_args = []
         channel_obj = object()
@@ -434,7 +434,6 @@ def test_streaming_depends_on_gax(self):
 
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials, use_gax=False)
-        client.connection = _Connection()
         sample = client.sample(content=self.AUDIO_CONTENT,
                                encoding=speech.Encoding.LINEAR16,
                                sample_rate=self.SAMPLE_RATE)
@@ -453,8 +452,7 @@ def test_streaming_closed_stream(self):
         stream = BytesIO(b'Some audio data...')
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials)
-        client.connection = _Connection()
-        client.connection.credentials = credentials
+        client._credentials = credentials
 
         channel_args = []
         channel_obj = object()
@@ -495,8 +493,7 @@ def test_stream_recognize_interim_results(self):
         stream = BytesIO(b'Some audio data...')
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials)
-        client.connection = _Connection()
-        client.connection.credentials = credentials
+        client._credentials = credentials
 
         alternatives = [{
             'transcript': 'testing streaming 1 2 3',
@@ -576,8 +573,7 @@ def test_stream_recognize(self):
         stream = BytesIO(b'Some audio data...')
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials)
-        client.connection = _Connection()
-        client.connection.credentials = credentials
+        client._credentials = credentials
 
         alternatives = [{
             'transcript': 'testing streaming 1 2 3',
@@ -633,8 +629,7 @@ def test_stream_recognize_no_results(self):
         stream = BytesIO(b'Some audio data...')
         credentials = _make_credentials()
         client = self._make_one(credentials=credentials)
-        client.connection = _Connection()
-        client.connection.credentials = credentials
+        client._credentials = credentials
 
         responses = [_make_streaming_response()]
 
@@ -670,8 +665,7 @@ def test_speech_api_with_gax(self):
 
         creds = _make_credentials()
         client = self._make_one(credentials=creds, use_gax=True)
-        client._connection = _Connection()
-        client._connection.credentials = creds
+        client._credentials = creds
 
         channel_args = []
         channel_obj = object()