From 0ba64fa1105270eb499c0c6904f0c5cfb57f1b16 Mon Sep 17 00:00:00 2001
From: Hyundo Moon <hdmoon@google.com>
Date: Thu, 20 Feb 2020 14:38:21 +0900
Subject: [PATCH] Add MediaRouter2.RoutingController#getOriginalId() as
 @TestApi

Bug: 149886370
Test: atest mediaroutertest and CTS
Change-Id: I4adc8041763756747e4c725d682533fb5707f183
---
 api/test-current.txt                       |  4 ++++
 media/java/android/media/MediaRouter2.java | 18 ++++++++++++++++--
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/api/test-current.txt b/api/test-current.txt
index 957794cac0573..9bc4637a52c0a 100644
--- a/api/test-current.txt
+++ b/api/test-current.txt
@@ -1473,6 +1473,10 @@ package android.media {
     method @NonNull public String getOriginalId();
   }
 
+  public class MediaRouter2.RoutingController {
+    method @NonNull public String getOriginalId();
+  }
+
   public final class PlaybackParams implements android.os.Parcelable {
     method public int getAudioStretchMode();
     method public android.media.PlaybackParams setAudioStretchMode(int);
diff --git a/media/java/android/media/MediaRouter2.java b/media/java/android/media/MediaRouter2.java
index 28bb4c1924255..5942a3d05e67e 100644
--- a/media/java/android/media/MediaRouter2.java
+++ b/media/java/android/media/MediaRouter2.java
@@ -21,6 +21,7 @@ import static com.android.internal.util.function.pooled.PooledLambda.obtainMessa
 import android.annotation.CallbackExecutor;
 import android.annotation.NonNull;
 import android.annotation.Nullable;
+import android.annotation.TestApi;
 import android.content.Context;
 import android.os.Bundle;
 import android.os.Handler;
@@ -721,8 +722,7 @@ public class MediaRouter2 {
             RoutingController newController) {
         for (TransferCallbackRecord record: mTransferCallbackRecords) {
             record.mExecutor.execute(
-                    () -> record.mTransferCallback.onTransferred(oldController,
-                            newController));
+                    () -> record.mTransferCallback.onTransferred(oldController, newController));
         }
     }
 
@@ -865,6 +865,20 @@ public class MediaRouter2 {
             }
         }
 
+        /**
+         * Gets the original session id set by
+         * {@link RoutingSessionInfo.Builder#Builder(String, String)}.
+         *
+         * @hide
+         */
+        @NonNull
+        @TestApi
+        public String getOriginalId() {
+            synchronized (mControllerLock) {
+                return mSessionInfo.getOriginalId();
+            }
+        }
+
         /**
          * @return the control hints used to control routing session if available.
          */