Show media source icon in the search screen

Bug: 146031656
Test: Run on emulator
Change-Id: Ia4027f72102dcf00c1567a3302832b61a5606557
diff --git a/src/com/android/car/media/MediaActivity.java b/src/com/android/car/media/MediaActivity.java
index d3dc8b1..4d32546 100644
--- a/src/com/android/car/media/MediaActivity.java
+++ b/src/com/android/car/media/MediaActivity.java
@@ -27,6 +27,7 @@
 import android.content.Intent;
 import android.content.pm.ResolveInfo;
 import android.graphics.drawable.BitmapDrawable;
+import android.graphics.drawable.Drawable;
 import android.media.audiofx.AudioEffect;
 import android.os.Bundle;
 import android.support.v4.media.session.PlaybackStateCompat;
@@ -491,9 +492,11 @@
         mCurrentPlaybackStateWrapper = null;
         maybeCancelToast();
         maybeCancelDialog();
-        mAppBarView.setLogo(mediaSource != null
+        Drawable icon = mediaSource != null
                 ? new BitmapDrawable(getResources(), mediaSource.getRoundPackageIcon())
-                : null);
+                : null;
+        mAppBarView.setLogo(icon);
+        mAppBarView.setSearchIcon(icon);
         if (mediaSource != null) {
             if (Log.isLoggable(TAG, Log.INFO)) {
                 Log.i(TAG, "Browsing: " + mediaSource.getDisplayName());