Merge pull request #2514 from daspecster/tighten-speech-async-encoding

daspecster · web-flow · commit 4bff05054f68 · 2016-10-07T16:33:47.000-04:00
Tighten Speech Asynchronous encoding support.
diff --git a/packages/google-cloud-speech/google/cloud/speech/client.py b/packages/google-cloud-speech/google/cloud/speech/client.py
@@ -19,6 +19,7 @@
 from google.cloud._helpers import _to_bytes
 from google.cloud import client as client_module
 from google.cloud.speech.connection import Connection
+from google.cloud.speech.encoding import Encoding
 from google.cloud.speech.operation import Operation
 from google.cloud.speech.sample import Sample
 
@@ -89,6 +90,9 @@ def async_recognize(self, sample, language_code=None,
         :rtype: `~google.cloud.speech.operation.Operation`
         :returns: ``Operation`` for asynchronous request to Google Speech API.
         """
+        if sample.encoding is not Encoding.LINEAR16:
+            raise ValueError('Only LINEAR16 encoding is supported by '
+                             'asynchronous speech requests.')
 
         data = _build_request_data(sample, language_code, max_alternatives,
                                    profanity_filter, speech_context)
diff --git a/packages/google-cloud-speech/unit_tests/test_client.py b/packages/google-cloud-speech/unit_tests/test_client.py
@@ -160,6 +160,20 @@ def test_sync_recognize_with_empty_results(self):
                             sample_rate=self.SAMPLE_RATE)
             client.sync_recognize(sample)
 
+    def test_async_supported_encodings(self):
+        from google.cloud.speech.encoding import Encoding
+        from google.cloud.speech.sample import Sample
+
+        credentials = _Credentials()
+        client = self._makeOne(credentials=credentials)
+        client.connection = _Connection({})
+
+        sample = Sample(source_uri=self.AUDIO_SOURCE_URI,
+                        encoding=Encoding.FLAC,
+                        sample_rate=self.SAMPLE_RATE)
+        with self.assertRaises(ValueError):
+            client.async_recognize(sample)
+
     def test_async_recognize(self):
         from unit_tests._fixtures import ASYNC_RECOGNIZE_RESPONSE
         from google.cloud.speech.encoding import Encoding
@@ -172,7 +186,7 @@ def test_async_recognize(self):
         client.connection = _Connection(RETURNED)
 
         sample = Sample(source_uri=self.AUDIO_SOURCE_URI,
-                        encoding=Encoding.FLAC,
+                        encoding=Encoding.LINEAR16,
                         sample_rate=self.SAMPLE_RATE)
         operation = client.async_recognize(sample)
         self.assertIsInstance(operation, Operation)
