diff --git a/packages/SystemUI/src/com/android/systemui/media/taptotransfer/MediaTttCommandLineHelper.kt b/packages/SystemUI/src/com/android/systemui/media/taptotransfer/MediaTttCommandLineHelper.kt index e4b8874ff601e..87579040372f6 100644 --- a/packages/SystemUI/src/com/android/systemui/media/taptotransfer/MediaTttCommandLineHelper.kt +++ b/packages/SystemUI/src/com/android/systemui/media/taptotransfer/MediaTttCommandLineHelper.kt @@ -66,11 +66,14 @@ class MediaTttCommandLineHelper @Inject constructor( as StatusBarManager val routeInfo = MediaRoute2Info.Builder("id", args[0]) .addFeature("feature") - .setPackageName(TEST_PACKAGE_NAME) - .build() + val useAppIcon = !(args.size >= 3 && args[2] == "useAppIcon=false") + if (useAppIcon) { + routeInfo.setPackageName(TEST_PACKAGE_NAME) + } + statusBarManager.updateMediaTapToTransferSenderDisplay( - displayState, - routeInfo, + displayState, + routeInfo.build(), getUndoExecutor(displayState), getUndoCallback(displayState) ) @@ -106,7 +109,8 @@ class MediaTttCommandLineHelper @Inject constructor( } override fun help(pw: PrintWriter) { - pw.println("Usage: adb shell cmd statusbar $SENDER_COMMAND ") + pw.println("Usage: adb shell cmd statusbar $SENDER_COMMAND " + + " useAppIcon=[true|false]") } } @@ -128,7 +132,8 @@ class MediaTttCommandLineHelper @Inject constructor( as StatusBarManager val routeInfo = MediaRoute2Info.Builder("id", "Test Name") .addFeature("feature") - if (args.size >= 2 && args[1] == "useAppIcon=true") { + val useAppIcon = !(args.size >= 2 && args[1] == "useAppIcon=false") + if (useAppIcon) { routeInfo.setPackageName(TEST_PACKAGE_NAME) }