[Media Router] Add a supports_web_ui_controller attribute to MediaRoute

chrome/test/data/webui/media_router/route_details_tests.js

 // Copyright 2015 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.
 
 /** @fileoverview Suite of tests for route-details. */
 cr.define('route_details', function() {
   function registerTests() {
     suite('RouteDetails', function() {
       /**
        * Route Details created before each test.
@@ skipping 30 matching lines @@
       // the |elementId| element are equal.
       var checkSpanText = function(expected, elementId) {
         assertEquals(
             expected,
             details.$$('#' + elementId).querySelector('span').innerText);
       };
 
       // Checks the default route view is shown.
       var checkDefaultViewIsShown = function() {
         assertFalse(details.$$('#route-information').hasAttribute('hidden'));
-        assertTrue(details.$$('extension-view-wrapper').hasAttribute('hidden'));
+        assertTrue(
+            !details.$$('extension-view-wrapper') ||
+            details.$$('extension-view-wrapper').hasAttribute('hidden'));
       };
 
       // Checks the start button is shown.
       var checkStartCastButtonIsShown = function() {
         assertFalse(details.$$('#start-casting-to-route-button')
                         .hasAttribute('hidden'));
       };
 
       // Checks the start button is not shown.
       var checkStartCastButtonIsNotShown = function() {
@@ skipping 18 matching lines @@
       suiteSetup(function() {
         return PolymerTest.importHtml(
             'chrome://media-router/elements/route_details/' +
             'route_details.html');
       });
 
       // Initialize a route-details before each test.
       setup(function(done) {
         PolymerTest.clearBody();
         details = document.createElement('route-details');
-        details.useWebUiRouteControls = false;
         document.body.appendChild(details);
 
         // Initialize routes and sinks.
         fakeRouteOne = new media_router.Route(
             'route id 1', 'sink id 1', 'Video 1', 1, true, false,
             fakeRouteOneControllerPath);
         fakeRouteTwo = new media_router.Route('route id 2', 'sink id 2',
             'Video 2', 2, false, true);
         fakeSinkOne = new media_router.Sink(
             'sink id 1', 'sink 1', 'description', null,
@@ skipping 101 matching lines @@
         details.route = fakeRouteTwo;
         assertEquals(fakeRouteTwo, details.route);
         checkSpanText(loadTimeData.getStringF('castingActivityStatus',
             fakeRouteTwo.description), 'route-information');
         checkDefaultViewIsShown();
         checkStartCastButtonIsShown();
       });
 
       // Tests when |route| exists, has a custom controller, and it loads.
       test('route has custom controller and loading succeeds', function(done) {
-        details.$$('extension-view-wrapper').$$('#custom-controller').load =
-            function(url) {
-          setTimeout(function() {
-            assertEquals(
-                fakeRouteOneControllerPath,
-                url.substring(0, fakeRouteOneControllerPath.length));
-            checkCustomControllerIsShown();
-            done();
-          });
-          return Promise.resolve();
-        };
+        // Get the extension-view-wrapper stamped first, so that we can mock out
+        // the load method.
+        details.route = fakeRouteTwo;
 
-        details.route = fakeRouteOne;
+        setTimeout(function() {
+          details.$$('extension-view-wrapper').$$('#custom-controller').load =
+              function(url) {
+            setTimeout(function() {
+              assertEquals(
+                  fakeRouteOneControllerPath,
+                  url.substring(0, fakeRouteOneControllerPath.length));
+              checkCustomControllerIsShown();
+              done();
+            });
+            return Promise.resolve();
+          };
+
+          details.route = fakeRouteOne;
+        });
       });
 
       // Tests when |route| exists, has a custom controller, but fails to load.
       test('route has custom controller but loading fails', function(done) {
-        details.$$('extension-view-wrapper').$$('#custom-controller').load =
-            function(url) {
-          setTimeout(function() {
-            checkDefaultViewIsShown();
-            done();
-          });
-          return Promise.reject();
-        };
+        // Get the extension-view-wrapper stamped first, so that we can mock out
+        // the load method.
+        details.route = fakeRouteTwo;
 
-        details.route = fakeRouteOne;
+        setTimeout(function() {
+          details.$$('extension-view-wrapper').$$('#custom-controller').load =
+              function(url) {
+            setTimeout(function() {
+              checkDefaultViewIsShown();
+              done();
+            });
+            return Promise.reject();
+          };
+
+          details.route = fakeRouteOne;
+        });
       });
     });
   }
 
   return {
     registerTests: registerTests,
   };
 });