Update route count properly when MediaRouteButton detects

grouped/ungrouped events

Change-Id: I1b00678cc3b548f347c51c3fcbfce18ecc8fbabe

1 files changed, 11 insertions, 0 deletions

diff --git a/core/java/android/app/MediaRouteButton.java b/core/java/android/app/MediaRouteButton.java
index 65f9f87d34ca..c34c1639dcc3 100644
--- a/core/java/android/app/MediaRouteButton.java
+++ b/core/java/android/app/MediaRouteButton.java
@@ -374,5 +374,16 @@ public class MediaRouteButton extends View {
         public void onRouteRemoved(MediaRouter router, RouteInfo info) {
             updateRouteCount();
         }
+
+        @Override
+        public void onRouteGrouped(MediaRouter router, RouteInfo info, RouteGroup group,
+                int index) {
+            updateRouteCount();
+        }
+
+        @Override
+        public void onRouteUngrouped(MediaRouter router, RouteInfo info, RouteGroup group) {
+            updateRouteCount();
+        }
     }
 }