From 5f8ac4337e001ec750d25e8145424c7935bd0907 Mon Sep 17 00:00:00 2001
From: Googler <noreply@google.com>
Date: Mon, 21 Mar 2022 10:26:02 -0700
Subject: [PATCH] Add triggerOnReadyForSpeech() to ShadowSpeechRecognizer.

PiperOrigin-RevId: 436243233
---
 .../shadows/ShadowSpeechRecognizerTest.java    | 18 +++++++++++++++++-
 .../shadows/ShadowSpeechRecognizer.java        |  4 ++++
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/robolectric/src/test/java/org/robolectric/shadows/ShadowSpeechRecognizerTest.java b/robolectric/src/test/java/org/robolectric/shadows/ShadowSpeechRecognizerTest.java
index 8419a5e1480..a20f3940b62 100644
--- a/robolectric/src/test/java/org/robolectric/shadows/ShadowSpeechRecognizerTest.java
+++ b/robolectric/src/test/java/org/robolectric/shadows/ShadowSpeechRecognizerTest.java
@@ -42,6 +42,20 @@ public void onErrorCalled() {
     assertThat(listener.errorReceived).isEqualTo(-1);
   }
 
+  @Test
+  public void onReadyForSpeechCalled() {
+    startListening();
+    Bundle expectedBundle = new Bundle();
+    ArrayList<String> results = new ArrayList<>();
+    String result = "onReadyForSpeech";
+    results.add(result);
+    expectedBundle.putStringArrayList(SpeechRecognizer.RESULTS_RECOGNITION, results);
+
+    shadowOf(speechRecognizer).triggerOnReadyForSpeech(expectedBundle);
+
+    assertThat(listener.bundleReceived).isEqualTo(expectedBundle);
+  }
+
   @Test
   public void onPartialResultsCalled() {
     startListening();
@@ -183,7 +197,9 @@ public void onPartialResults(Bundle bundle) {
     }
 
     @Override
-    public void onReadyForSpeech(Bundle params) {}
+    public void onReadyForSpeech(Bundle bundle) {
+      bundleReceived = bundle;
+    }
 
     @Override
     public void onResults(Bundle bundle) {
diff --git a/shadows/framework/src/main/java/org/robolectric/shadows/ShadowSpeechRecognizer.java b/shadows/framework/src/main/java/org/robolectric/shadows/ShadowSpeechRecognizer.java
index 29d8bcdc171..3bccfa8ddf4 100644
--- a/shadows/framework/src/main/java/org/robolectric/shadows/ShadowSpeechRecognizer.java
+++ b/shadows/framework/src/main/java/org/robolectric/shadows/ShadowSpeechRecognizer.java
@@ -97,6 +97,10 @@ public void triggerOnError(int error) {
     recognitionListener.onError(error);
   }
 
+  public void triggerOnReadyForSpeech(Bundle bundle) {
+    recognitionListener.onReadyForSpeech(bundle);
+  }
+
   public void triggerOnPartialResults(Bundle bundle) {
     recognitionListener.onPartialResults(bundle);
   }