Set tracking state properly

When user selects printer from Select printer screen, we started
tracking last selected printer instead of selected printer. It is
happening due to race condition between onStart and onActivityResult
callback.
Add a support to start tracking printer from onActivityResult function.

Bug: 341016728
Bug: 335214357
Test: atest frameworks/base/packages/PrintSpooler/tests/outofprocess/src/com/android/printspooler/outofprocess/tests/WorkflowTest.java
Change-Id: I096bf15d3d3ebc45ffbb05db33ac5936dedc14fa

1 files changed, 3 insertions, 0 deletions

diff --git a/packages/PrintSpooler/src/com/android/printspooler/ui/PrintActivity.java b/packages/PrintSpooler/src/com/android/printspooler/ui/PrintActivity.java
index d25d5dcaac87..ff09084e24cd 100644
--- a/packages/PrintSpooler/src/com/android/printspooler/ui/PrintActivity.java
+++ b/packages/PrintSpooler/src/com/android/printspooler/ui/PrintActivity.java
@@ -785,6 +785,9 @@ public class PrintActivity extends Activity implements RemotePrintDocument.Updat
                 } else {
                     onPrinterUnavailable(printerInfo);
                 }
+                if (mPrinterRegistry != null) {
+                    mPrinterRegistry.setTrackedPrinter(mCurrentPrinter.getId());
+                }
 
                 mDestinationSpinnerAdapter.ensurePrinterInVisibleAdapterPosition(printerInfo);