From 622b5d33df631a7bd6fc5482a9a80931eece0c2d Mon Sep 17 00:00:00 2001
From: Matt Casey <mrcasey@google.com>
Date: Wed, 15 Feb 2023 21:01:10 +0000
Subject: Handle refinement responses on the main thread.

Bug: 269170766
Test: atest CtsSharesheetDeviceTest
Change-Id: I539c1f4a00c4572913fb3c0f10f22d9c20eb6ca9
---
 .../src/com/android/intentresolver/ChooserRefinementManager.java | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'java/src')

diff --git a/java/src/com/android/intentresolver/ChooserRefinementManager.java b/java/src/com/android/intentresolver/ChooserRefinementManager.java
index 5997bfed..98c6bddc 100644
--- a/java/src/com/android/intentresolver/ChooserRefinementManager.java
+++ b/java/src/com/android/intentresolver/ChooserRefinementManager.java
@@ -23,6 +23,7 @@ import android.content.Intent;
 import android.content.IntentSender;
 import android.content.IntentSender.SendIntentException;
 import android.os.Bundle;
+import android.os.Handler;
 import android.os.Parcel;
 import android.os.Parcelable;
 import android.os.ResultReceiver;
@@ -94,7 +95,8 @@ public final class ChooserRefinementManager {
                         mOnRefinementCancelled.run();
                     }
                 },
-                mOnRefinementCancelled);
+                mOnRefinementCancelled,
+                mContext.getMainThreadHandler());
 
         Intent refinementRequest = makeRefinementRequest(mRefinementResultReceiver, selectedTarget);
         try {
@@ -136,8 +138,9 @@ public final class ChooserRefinementManager {
 
         RefinementResultReceiver(
                 Consumer<Intent> onSelectionRefined,
-                Runnable onRefinementCancelled) {
-            super(/* handler=*/ null);
+                Runnable onRefinementCancelled,
+                Handler handler) {
+            super(handler);
             mOnSelectionRefined = onSelectionRefined;
             mOnRefinementCancelled = onRefinementCancelled;
         }
-- 
cgit v1.2.3-59-g8ed1b