From 770ccb9262f4b38db60c9157678331ea6c41e831 Mon Sep 17 00:00:00 2001 From: "David T.H. Kao" Date: Wed, 9 May 2018 15:52:11 -0700 Subject: Add functionality to specify timeout for waitForSms. (#95) --- .../java/com/google/android/mobly/snippet/bundled/SmsSnippet.java | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'src/main/java/com/google/android/mobly') diff --git a/src/main/java/com/google/android/mobly/snippet/bundled/SmsSnippet.java b/src/main/java/com/google/android/mobly/snippet/bundled/SmsSnippet.java index 4a38798..ec6d470 100644 --- a/src/main/java/com/google/android/mobly/snippet/bundled/SmsSnippet.java +++ b/src/main/java/com/google/android/mobly/snippet/bundled/SmsSnippet.java @@ -115,12 +115,11 @@ public class SmsSnippet implements Snippet { @TargetApi(Build.VERSION_CODES.KITKAT) @Rpc(description = "Wait for incoming SMS message.") - public JSONObject waitForSms() throws Throwable { + public JSONObject waitForSms(int timeoutMillis) throws Throwable { String callbackId = SMS_CALLBACK_ID_PREFIX + (++mCallbackCounter); SmsReceiver receiver = new SmsReceiver(mContext, callbackId); mContext.registerReceiver(receiver, new IntentFilter(Intents.SMS_RECEIVED_ACTION)); - return Utils.waitForSnippetEvent( - callbackId, SMS_RECEIVED_EVENT_NAME, DEFAULT_TIMEOUT_MILLISECOND) + return Utils.waitForSnippetEvent(callbackId, SMS_RECEIVED_EVENT_NAME, timeoutMillis) .toJson(); } -- cgit v1.2.3