Adding unittests for media casting sink observation and route handling

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

Index: chrome/android/junit/src/org/chromium/chrome/browser/media/router/ChromeMediaRouterRouteTest.java
diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/media/router/ChromeMediaRouterRouteTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/media/router/ChromeMediaRouterRouteTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..f74aca510dfe0da676db4bdc9a2297c187f22fc2
--- /dev/null
+++ b/chrome/android/junit/src/org/chromium/chrome/browser/media/router/ChromeMediaRouterRouteTest.java
@@ -0,0 +1,154 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.chrome.browser.media.router;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.verify;
+
+import org.chromium.base.test.util.Feature;
+import org.chromium.testing.local.LocalRobolectricTestRunner;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.robolectric.annotation.Config;
+
+/**
+ * Route tests for ChromeMediaRouter.
+ */
+@RunWith(LocalRobolectricTestRunner.class)
+@Config(manifest = Config.NONE,
+        shadows = { ChromeMediaRouterTestBase.FakeActivityManager.class })
+public class ChromeMediaRouterRouteTest extends ChromeMediaRouterTestBase {
+    @Test
+    @Feature({"MediaRouter"})
+    public void testCreateOneRoute() {
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 0);
+
+        mChromeMediaRouter.createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+        verify(mRouteProvider).createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+
+        String routeId1 = new MediaRoute(SINK_ID1, SOURCE_ID1, PRESENTATION_ID1).id;
+        mChromeMediaRouter.onRouteCreated(
+                routeId1, SINK_ID1, REQUEST_ID1, mRouteProvider, true);
+
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 1);
+        assertTrue(mChromeMediaRouter.getRouteIdsToProvidersForTest().containsKey(routeId1));
+    }
+
+    @Test
+    @Feature({"MediaRouter"})
+    public void testCreateTwoRoutes() {
+        mChromeMediaRouter.createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+
+        String routeId1 = new MediaRoute(SINK_ID1, SOURCE_ID1, PRESENTATION_ID1).id;
+        mChromeMediaRouter.onRouteCreated(
+                routeId1, SINK_ID1, REQUEST_ID1, mRouteProvider, true);
+
+        mChromeMediaRouter.createRoute(
+                SOURCE_ID2, SINK_ID2, PRESENTATION_ID2, ORIGIN2, TAB_ID2, REQUEST_ID2);
+
+        verify(mRouteProvider).createRoute(
+                SOURCE_ID2, SINK_ID2, PRESENTATION_ID2, ORIGIN2, TAB_ID2, REQUEST_ID2);
+        String routeId2 = new MediaRoute(SINK_ID2, SOURCE_ID2, PRESENTATION_ID2).id;
+        mChromeMediaRouter.onRouteCreated(
+                routeId2, SINK_ID2, REQUEST_ID2, mRouteProvider, true);
+
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 2);
+        assertTrue(mChromeMediaRouter.getRouteIdsToProvidersForTest().containsKey(routeId2));
+    }
+
+    @Test
+    @Feature({"MediaRouter"})
+    public void testCreateRouteFails() {
+        mChromeMediaRouter.createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+
+        verify(mRouteProvider).createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+        mChromeMediaRouter.onRouteRequestError("ERROR", REQUEST_ID1);
+
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 0);
+    }
+
+    @Test
+    @Feature({"MediaRouter"})
+    public void testJoinRoute() {
+        mChromeMediaRouter.createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+
+        String routeId1 = new MediaRoute(SINK_ID1, SOURCE_ID1, PRESENTATION_ID1).id;
+        mChromeMediaRouter.onRouteCreated(
+                routeId1, SINK_ID1, REQUEST_ID1, mRouteProvider, true);
+
+        mChromeMediaRouter.joinRoute(
+                SOURCE_ID2, PRESENTATION_ID1, ORIGIN1, TAB_ID2, REQUEST_ID2);
+        verify(mRouteProvider).joinRoute(
+                SOURCE_ID2, PRESENTATION_ID1, ORIGIN1, TAB_ID2, REQUEST_ID2);
+
+        String routeId2 = new MediaRoute(SINK_ID1, SOURCE_ID2, PRESENTATION_ID2).id;
+        mChromeMediaRouter.onRouteCreated(
+                routeId2, SINK_ID1, REQUEST_ID2, mRouteProvider, true);
+
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 2);
+        assertTrue(mChromeMediaRouter.getRouteIdsToProvidersForTest().containsKey(routeId2));
+    }
+
+    @Test
+    @Feature({"MediaRouter"})
+    public void testJoinRouteFails() {
+        mChromeMediaRouter.createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+
+        String routeId1 = new MediaRoute(SINK_ID1, SOURCE_ID1, PRESENTATION_ID1).id;
+        mChromeMediaRouter.onRouteCreated(
+                routeId1, SINK_ID1, REQUEST_ID1, mRouteProvider, true);
+
+        mChromeMediaRouter.joinRoute(
+                SOURCE_ID2, PRESENTATION_ID1, ORIGIN1, TAB_ID2, REQUEST_ID2);
+        verify(mRouteProvider).joinRoute(
+                SOURCE_ID2, PRESENTATION_ID1, ORIGIN1, TAB_ID2, REQUEST_ID2);
+
+        mChromeMediaRouter.onRouteRequestError("error", REQUEST_ID2);
+
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 1);
+    }
+
+    @Test
+    @Feature({"MediaRouter"})
+    public void testDetachRoute() {
+        mChromeMediaRouter.createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+
+        String routeId1 = new MediaRoute(SINK_ID1, SOURCE_ID1, PRESENTATION_ID1).id;
+        mChromeMediaRouter.onRouteCreated(
+                routeId1, SINK_ID1, REQUEST_ID1, mRouteProvider, true);
+
+        mChromeMediaRouter.detachRoute(routeId1);
+        verify(mRouteProvider).detachRoute(routeId1);
+
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 0);
+    }
+
+    @Test
+    @Feature({"MediaRouter"})
+    public void testCloseRoute() {
+        mChromeMediaRouter.createRoute(
+                SOURCE_ID1, SINK_ID1, PRESENTATION_ID1, ORIGIN1, TAB_ID1, REQUEST_ID1);
+
+        String routeId1 = new MediaRoute(SINK_ID1, SOURCE_ID1, PRESENTATION_ID1).id;
+        mChromeMediaRouter.onRouteCreated(
+                routeId1, SINK_ID1, REQUEST_ID1, mRouteProvider, true);
+
+        mChromeMediaRouter.closeRoute(routeId1);
+        verify(mRouteProvider).closeRoute(routeId1);
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 1);
+
+        mChromeMediaRouter.onRouteClosed(routeId1);
+        assertEquals(mChromeMediaRouter.getRouteIdsToProvidersForTest().size(), 0);
+    }
+}