diff --git a/media/tests/MediaRouter/src/com/android/mediaroutertest/MediaRouter2Test.java b/media/tests/MediaRouter/src/com/android/mediaroutertest/MediaRouter2Test.java index daaa8688e2034..e0b454559456d 100644 --- a/media/tests/MediaRouter/src/com/android/mediaroutertest/MediaRouter2Test.java +++ b/media/tests/MediaRouter/src/com/android/mediaroutertest/MediaRouter2Test.java @@ -36,6 +36,7 @@ import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertNotEquals; import static org.junit.Assert.assertNotNull; +import static org.junit.Assert.assertSame; import static org.junit.Assert.assertTrue; import static org.testng.Assert.assertThrows; @@ -693,6 +694,14 @@ public class MediaRouter2Test { assertFalse(systemController.isReleased()); } + @Test + public void testControllers() { + List controllers = mRouter2.getControllers(); + assertNotNull(controllers); + assertFalse(controllers.isEmpty()); + assertSame(mRouter2.getSystemController(), controllers.get(0)); + } + // Helper for getting routes easily static Map createRouteMap(List routes) { Map routeMap = new HashMap<>();