[Cast,Android] Handle some PresentationAPI edge cases

chrome/android/junit/src/org/chromium/chrome/browser/media/router/cast/CastMediaRouteProviderTest.java

Index: chrome/android/junit/src/org/chromium/chrome/browser/media/router/cast/CastMediaRouteProviderTest.java
diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/media/router/cast/CastMediaRouteProviderTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/media/router/cast/CastMediaRouteProviderTest.java
index 04578c6842daf99d312fa95f16b3549b8b53c4dc..1974aa14d2327f0cfabed5d4ef69f4c0a6cf9b60 100644
--- a/chrome/android/junit/src/org/chromium/chrome/browser/media/router/cast/CastMediaRouteProviderTest.java
+++ b/chrome/android/junit/src/org/chromium/chrome/browser/media/router/cast/CastMediaRouteProviderTest.java
@@ -18,6 +18,7 @@ import org.robolectric.annotation.Config;
 
 import org.chromium.base.test.util.Feature;
 import org.chromium.chrome.browser.media.router.ChromeMediaRouter;
+import org.chromium.chrome.browser.media.router.MediaRoute;
 import org.chromium.chrome.browser.media.router.MediaRouteManager;
 import org.chromium.testing.local.LocalRobolectricTestRunner;
 
@@ -62,4 +63,37 @@ public class CastMediaRouteProviderTest {
                 .onSinksReceived(
                         eq(UNSUPPORTED_SOURCE), same(provider), eq(new ArrayList<MediaSink>()));
     }
+
+    @Test
+    @Feature({"MediaRouter"})
+    public void testOnSessionClosedNoClientRecord() {
+        ChromeMediaRouter.setAndroidMediaRouterForTest(mock(MediaRouter.class));
+
+        MediaRouteManager mockManager = mock(MediaRouteManager.class);
+        CastMediaRouteProvider provider = CastMediaRouteProvider.create(mockManager);
+
+        CastSession mockSession = mock(CastSession.class);
+        provider.onSessionCreated(mockSession);
+
+        MediaRoute route = new MediaRoute("sink", SUPPORTED_SOURCE, "");
+        provider.addRoute(route, "", -1);
+        provider.onSessionClosed();
+
+        verify(mockManager).onRouteClosed(route.id);
+    }
+
+    @Test
+    @Feature({"MediaRouter"})
+    public void testCloseRouteWithNoSession() {
+        ChromeMediaRouter.setAndroidMediaRouterForTest(mock(MediaRouter.class));
+
+        MediaRouteManager mockManager = mock(MediaRouteManager.class);
+        CastMediaRouteProvider provider = CastMediaRouteProvider.create(mockManager);
+
+        MediaRoute route = new MediaRoute("sink", SUPPORTED_SOURCE, "");
+        provider.addRoute(route, "", -1);
+        provider.closeRoute(route.id);
+
+        verify(mockManager).onRouteClosed(route.id);
+    }
 }