def test_long_running_recognize(self): # Setup Expected Response expected_response = {} expected_response = cloud_speech_pb2.LongRunningRecognizeResponse( **expected_response) operation = operations_pb2.Operation( name='operations/test_long_running_recognize', done=True) operation.response.Pack(expected_response) # Mock the API response channel = ChannelStub(responses=[operation]) client = speech_v1.SpeechClient(channel=channel) # Setup Request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = 'en-US' config = { 'encoding': encoding, 'sample_rate_hertz': sample_rate_hertz, 'language_code': language_code } uri = 'gs://bucket_name/file_name.flac' audio = {'uri': uri} response = client.long_running_recognize(config, audio) result = response.result() assert expected_response == result assert len(channel.requests) == 1 expected_request = cloud_speech_pb2.LongRunningRecognizeRequest( config=config, audio=audio) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_long_running_recognize(self): # Setup Expected Response expected_response = {} expected_response = cloud_speech_pb2.LongRunningRecognizeResponse( **expected_response) operation = operations_pb2.Operation( name="operations/test_long_running_recognize", done=True) operation.response.Pack(expected_response) # Mock the API response channel = ChannelStub(responses=[operation]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = speech_v1.SpeechClient() # Setup Request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = "en-US" config = { "encoding": encoding, "sample_rate_hertz": sample_rate_hertz, "language_code": language_code, } uri = "gs://bucket_name/file_name.flac" audio = {"uri": uri} response = client.long_running_recognize(config, audio) result = response.result() assert expected_response == result assert len(channel.requests) == 1 expected_request = cloud_speech_pb2.LongRunningRecognizeRequest( config=config, audio=audio) actual_request = channel.requests[0][1] assert expected_request == actual_request