test(speech to text): Add more tests for speech to text & websockets

Author: maxnussbaum

lib/watson_apis/websocket/speech_to_text_websocket_listener.rb

@@ -27,7 +27,8 @@ def initialize(audio: nil, chunk_data:, options:, recognize_callback:, url:, hea
   end
 
   def start
-    on_open = lambda do |_event|
+    on_open = lambda do |event|
+      on_connect(event)
       @client.send(build_start_message(options: @options))
       @mic_running = true if @chunk_data
       send_audio(data: @audio)
@@ -84,6 +85,7 @@ def start
       @client.add_listener(Faye::WebSocket::API::Event.create("open"))
       @client.add_listener(Faye::WebSocket::API::Event.create("message"))
       @client.add_listener(Faye::WebSocket::API::Event.create("close"))
+      @client.add_listener(Faye::WebSocket::API::Event.create("error"))
     end
   end
 

test/integration/test_speech_to_text_v1.rb

@@ -126,6 +126,30 @@ def test_acoustic_model
     )
   end
 
+  def test_recognize_websocket_as_chunks
+    audio_file = File.open(Dir.getwd + "/resources/speech.wav")
+    mycallback = MyRecognizeCallback.new
+    speech = @service.recognize_with_websocket(
+      chunk_data: true,
+      recognize_callback: mycallback,
+      interim_results: true,
+      timestamps: true,
+      max_alternatives: 2,
+      word_alternatives_threshold: 0.5,
+      model: "en-US_BroadbandModel"
+    )
+    Thread.new do
+      until audio_file.eof?
+        chunk = audio_file.read(1024)
+        speech.add_audio_chunk(chunk: chunk)
+      end
+      sleep(1)
+      speech.stop_audio # Tell the websocket object that no more audio will be added
+    end
+    thr = Thread.new { speech.start }
+    thr.join
+  end
+
   def test_recognize_websocket
     audio_file = File.open(Dir.getwd + "/resources/speech.wav")
     mycallback = MyRecognizeCallback.new

test/unit/test_speech_to_text_v1.rb

@@ -360,7 +360,15 @@ def test_custom_corpora
     service_response = service.add_corpus(
       customization_id: "customid",
       corpus_name: "corpus",
-      corpus_file: corpus_file
+      corpus_file: corpus_file,
+      corpus_filename: "corpus-short-1.txt"
+    )
+    assert_nil(service_response)
+
+    service_response = service.add_corpus(
+      customization_id: "customid",
+      corpus_name: "corpus",
+      corpus_file: "corpus_file"
     )
     assert_nil(service_response)
 
@@ -693,4 +701,55 @@ def test_recognize_async
     output = future.value.body
     assert_equal(recognize_response["results"][0]["alternatives"][0]["transcript"], output["results"][0]["alternatives"][0]["transcript"])
   end
+
+  def test_reset_language_model
+    service = WatsonAPIs::SpeechToTextV1.new(
+      username: "username",
+      password: "password"
+    )
+    stub_request(:post, "https://stream.watsonplatform.net/speech-to-text/api/v1/customizations/customization_id/reset")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "stream.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: "", headers: {})
+    service_response = service.reset_language_model(customization_id: "customization_id")
+    assert_nil(service_response)
+  end
+
+  def test_upgrade_language_model
+    service = WatsonAPIs::SpeechToTextV1.new(
+      username: "username",
+      password: "password"
+    )
+    stub_request(:post, "https://stream.watsonplatform.net/speech-to-text/api/v1/customizations/customization_id/upgrade_model")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "stream.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: "", headers: {})
+    service_response = service.upgrade_language_model(customization_id: "customization_id")
+    assert_nil(service_response)
+  end
+
+  def test_upgrade_acoustic_model
+    service = WatsonAPIs::SpeechToTextV1.new(
+      username: "username",
+      password: "password"
+    )
+    stub_request(:post, "https://stream.watsonplatform.net/speech-to-text/api/v1/acoustic_customizations/customization_id/upgrade_model")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "stream.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: "", headers: {})
+    service_response = service.upgrade_acoustic_model(customization_id: "customization_id")
+    assert_nil(service_response)
+  end
 end