bluetooth: Fix BluetoothInternalsTest timeout issue

chrome/test/data/webui/bluetooth_internals_browsertest.js

 // 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.
 
 /**
  * @fileoverview Tests for chrome://bluetooth-internals
  */
 
 /** @const {string} Path to source root. */
 var ROOT_PATH = '../../../../';
@@ skipping 126 matching lines @@
           setTestAdapter: function(adapterInfo) {
             this.adapterInfo_ = adapterInfo;
           },
 
           setTestDevices: function(devices) {
             this.devices_ = devices;
           }
         };
 
         frameInterfaces.addInterfaceOverrideForTesting(
-          adapter.AdapterFactory.name,
-          function(handle) {
-            var stub = connection.bindHandleToStub(
-                handle, adapter.AdapterFactory);
-            this.adapterFactory = new TestAdapterFactoryProxy();
-            bindings.StubBindings(stub).delegate = this.adapterFactory;
+            adapter.AdapterFactory.name, function(handle) {
+              var stub = connection.bindHandleToStub(
+                  handle, adapter.AdapterFactory);
 
-            this.setupResolver.resolve();
-          }.bind(this));
+              this.adapterFactory = new TestAdapterFactoryProxy();
+
+              this.adapterFactory.adapter.proxy.setTestDevices([
+                this.fakeDeviceInfo1(),
+                this.fakeDeviceInfo2(),
+              ]);
+              this.adapterFactory.adapter.proxy.setTestAdapter(
+                  this.fakeAdapterInfo());
+
+              bindings.StubBindings(stub).delegate = this.adapterFactory;
+
+              this.setupResolver.resolve();
+            }.bind(this));
+
       }.bind(this));
     }.bind(this);
   },
+
+  /**
+   * Returns a copy of fake adapter info object.
+   * @return {!Object}
+   */
+  fakeAdapterInfo: function() {
+    return {
+      address: '02:1C:7E:6A:11:5A',
+      discoverable: false,
+      discovering: false,
+      initialized: true,
+      name: 'computer.example.com-0',
+      powered: true,
+      present: true,
+    };
+  },
+
+  /**
+   * Returns a copy of a fake device info object (variant 1).
+   * @return {!Object}
+   */
+  fakeDeviceInfo1: function() {
+    return {
+      address: "AA:AA:84:96:92:84",
+      name: "AAA",
+      name_for_display: "AAA",
+      rssi: {value: -40},
+      services: [],
+    };
+  },
+
+  /**
+   * Returns a copy of a fake device info object (variant 2).
+   * @return {!Object}
+   */
+  fakeDeviceInfo2: function() {
+    return {
+      address: "BB:BB:84:96:92:84",
+      name: "BBB",
+      name_for_display: "BBB",
+      rssi: null,
+      services: [],
+    };
+  },
+
+  /**
+   * Returns a copy of fake device info object. The returned device info lack
+   * rssi and services properties.
+   * @return {!Object}
+   */
+  fakeDeviceInfo3: function() {
+    return {
+      address: "CC:CC:84:96:92:84",
+      name: "CCC",
+      name_for_display: "CCC",
+    };
+  },
 };
 
-// Times out. See https://crbug.com/667970.
-TEST_F('BluetoothInternalsTest', 'DISABLED_Startup_BluetoothInternals',
-    function() {
-  var fakeAdapterInfo = {
-    address: '02:1C:7E:6A:11:5A',
-    discoverable: false,
-    discovering: false,
-    initialized: true,
-    name: 'computer.example.com-0',
-    powered: true,
-    present: true,
-  };
+TEST_F('BluetoothInternalsTest', 'Startup_BluetoothInternals', function() {
+  var adapterFactory = null;
+  var deviceTable = null;
 
-  var fakeDeviceInfo1 = {
-    address: "AA:AA:84:96:92:84",
-    name: "AAA",
-    name_for_display: "AAA",
-    rssi: {value: -40},
-    services: []
-  };
-
-  var fakeDeviceInfo2 = {
-    address: "BB:BB:84:96:92:84",
-    name: "BBB",
-    name_for_display: "BBB",
-    rssi: null,
-    services: []
-  };
-
-  var fakeDeviceInfo3 = {
-    address: "CC:CC:84:96:92:84",
-    name: "CCC",
-    name_for_display: "CCC",
-  };
-
-  var adapterFactory = null;
+  var fakeDeviceInfo1 = this.fakeDeviceInfo1;
+  var fakeDeviceInfo2 = this.fakeDeviceInfo2;
+  var fakeDeviceInfo3 = this.fakeDeviceInfo3;
 
   // Before tests are run, make sure setup completes.
   var setupPromise = this.setupResolver.promise.then(function() {
     adapterFactory = this.adapterFactory;
   }.bind(this));
 
 
   suite('BluetoothInternalsUITest', function() {
     var EXPECTED_DEVICES = 2;
 
     suiteSetup(function() {
       return setupPromise.then(function() {
-        adapterFactory.adapter.proxy.setTestDevices([
-          fakeDeviceInfo1,
-          fakeDeviceInfo2
-        ]);
-        adapterFactory.adapter.proxy.setTestAdapter(fakeAdapterInfo);
-
         return Promise.all([
           adapterFactory.whenCalled('getAdapter'),
           adapterFactory.adapter.proxy.whenCalled('getInfo'),
           adapterFactory.adapter.proxy.whenCalled('getDevices'),
           adapterFactory.adapter.proxy.whenCalled('setClient'),
         ]);
       });
     });
 
     setup(function() {
+      deviceTable = document.querySelector('#devices table');
       devices.splice(0, devices.length);
-      adapterBroker.adapterClient_.deviceAdded(fakeDeviceInfo1);
-      adapterBroker.adapterClient_.deviceAdded(fakeDeviceInfo2);
+      adapterBroker.adapterClient_.deviceAdded(fakeDeviceInfo1());
+      adapterBroker.adapterClient_.deviceAdded(fakeDeviceInfo2());
     });
 
     teardown(function() {
       adapterFactory.reset();
     });
 
     /**
      * Updates device info and verifies the contents of the device table.
      * @param {!device_collection.DeviceInfo} deviceInfo
      */
     function changeDevice(deviceInfo) {
-      var deviceRow = document.querySelector('#' + escapeDeviceAddress(
+      var deviceRow = deviceTable.querySelector('#' + escapeDeviceAddress(
           deviceInfo.address));
       var nameForDisplayColumn = deviceRow.children[0];
       var addressColumn = deviceRow.children[1];
       var rssiColumn = deviceRow.children[2];
       var servicesColumn = deviceRow.children[3];
 
       expectTrue(!!nameForDisplayColumn);
       expectTrue(!!addressColumn);
       expectTrue(!!rssiColumn);
       expectTrue(!!servicesColumn);
@@ skipping 23 matching lines @@
     function escapeDeviceAddress(address) {
       return address.replace(/:/g, '\\:');
     }
 
     /**
      * Expects whether device with |address| is removed.
      * @param {string} address
      * @param {boolean} expectRemoved
      */
     function expectDeviceRemoved(address, expectRemoved) {
-      var removedRow = document.querySelector(
+      var removedRow = deviceTable.querySelector(
           '#' + escapeDeviceAddress(address));
 
       expectEquals(expectRemoved, removedRow.classList.contains('removed'));
     }
 
     /**
      * Tests whether a device is added successfully and not duplicated.
      */
     test('DeviceAdded', function() {
-      var devices = document.querySelectorAll('#device-table tbody tr');
+      var devices = deviceTable.querySelectorAll('tbody tr');
       expectEquals(EXPECTED_DEVICES, devices.length);
 
       // Copy device info because device collection will not copy this object.
-      var infoCopy = Object.assign({}, fakeDeviceInfo3);
+      var infoCopy = fakeDeviceInfo3();
       adapterBroker.adapterClient_.deviceAdded(infoCopy);
 
       // Same device shouldn't appear twice.
       adapterBroker.adapterClient_.deviceAdded(infoCopy);
 
-      devices = document.querySelectorAll('#device-table tbody tr');
+      devices = deviceTable.querySelectorAll('tbody tr');
       expectEquals(EXPECTED_DEVICES + 1, devices.length);
     });
 
     /**
      * Tests whether a device is marked properly as removed.
      */
     test('DeviceSetToRemoved', function() {
-      var devices = document.querySelectorAll('#device-table tbody tr');
+      var devices = deviceTable.querySelectorAll('tbody tr');
       expectEquals(EXPECTED_DEVICES, devices.length);
-      adapterBroker.adapterClient_.deviceRemoved(fakeDeviceInfo2);
+
+      var fakeDevice = fakeDeviceInfo2();
+      adapterBroker.adapterClient_.deviceRemoved(fakeDevice);
 
       // The number of rows shouldn't change.
-      devices = document.querySelectorAll('#device-table tbody tr');
+      devices = deviceTable.querySelectorAll('tbody tr');
       expectEquals(EXPECTED_DEVICES, devices.length);
 
-      expectDeviceRemoved(fakeDeviceInfo2.address, true);
+      expectDeviceRemoved(fakeDevice.address, true);
     });
 
     /**
      * Tests whether a changed device updates the device table properly.
      */
     test('DeviceChanged', function() {
-      var devices = document.querySelectorAll('#device-table tbody tr');
+      var devices = deviceTable.querySelectorAll('tbody tr');
       expectEquals(EXPECTED_DEVICES, devices.length);
 
       // Copy device info because device collection will not copy this object.
-      var newDeviceInfo = Object.assign({}, fakeDeviceInfo1);
+      var newDeviceInfo = fakeDeviceInfo1();
       newDeviceInfo.name_for_display = 'DDDD';
       newDeviceInfo.rssi = { value: -20 };
       newDeviceInfo.services = ['service1', 'service2', 'service3'];
 
       changeDevice(newDeviceInfo);
     });
 
     /**
      * Tests the entire device cycle, added -> updated -> removed -> re-added.
      */
     test('DeviceUpdateCycle', function() {
-      var devices = document.querySelectorAll('#device-table tbody tr');
+      var devices = deviceTable.querySelectorAll('tbody tr');
       expectEquals(EXPECTED_DEVICES, devices.length);
 
       // Copy device info because device collection will not copy this object.
-      var originalDeviceInfo = Object.assign({}, fakeDeviceInfo3);
+      var originalDeviceInfo = fakeDeviceInfo3();
       adapterBroker.adapterClient_.deviceAdded(originalDeviceInfo);
 
-      var newDeviceInfo = Object.assign({}, fakeDeviceInfo3);
+      var newDeviceInfo = fakeDeviceInfo3();
       newDeviceInfo.name_for_display = 'DDDD';
       newDeviceInfo.rssi = { value: -20 };
       newDeviceInfo.services = ['service1', 'service2', 'service3'];
 
       changeDevice(newDeviceInfo);
       changeDevice(originalDeviceInfo);
 
       adapterBroker.adapterClient_.deviceRemoved(originalDeviceInfo);
       expectDeviceRemoved(originalDeviceInfo.address, true);
 
       adapterBroker.adapterClient_.deviceAdded(originalDeviceInfo);
       expectDeviceRemoved(originalDeviceInfo.address, false);
     });
 
     test('DeviceAddedRssiCheck', function() {
-      var devices = document.querySelectorAll('#device-table tbody tr');
+      var devices = deviceTable.querySelectorAll('tbody tr');
       expectEquals(EXPECTED_DEVICES, devices.length);
 
       // Copy device info because device collection will not copy this object.
-      var newDeviceInfo = Object.assign({}, fakeDeviceInfo3);
+      var newDeviceInfo = fakeDeviceInfo3();
       adapterBroker.adapterClient_.deviceAdded(newDeviceInfo);
 
-      var deviceRow = document.querySelector('#' + escapeDeviceAddress(
+      var deviceRow = deviceTable.querySelector('#' + escapeDeviceAddress(
           newDeviceInfo.address));
       var rssiColumn = deviceRow.children[2];
       expectEquals('Unknown', rssiColumn.textContent);
 
-      var newDeviceInfo1 = Object.assign({}, fakeDeviceInfo3);
+      var newDeviceInfo1 = fakeDeviceInfo3();
       newDeviceInfo1.rssi = {value: -42};
       adapterBroker.adapterClient_.deviceChanged(newDeviceInfo1);
       expectEquals('-42', rssiColumn.textContent);
 
       // Device table should keep last valid rssi value.
-      var newDeviceInfo2 = Object.assign({}, fakeDeviceInfo3);
+      var newDeviceInfo2 = fakeDeviceInfo3();
       newDeviceInfo2.rssi = null;
       adapterBroker.adapterClient_.deviceChanged(newDeviceInfo2);
       expectEquals('-42', rssiColumn.textContent);
 
-      var newDeviceInfo3 = Object.assign({}, fakeDeviceInfo3);
+      var newDeviceInfo3 = fakeDeviceInfo3();
       newDeviceInfo3.rssi = {value: -17};
       adapterBroker.adapterClient_.deviceChanged(newDeviceInfo3);
       expectEquals('-17', rssiColumn.textContent);
     });
   });
 
   // Run all registered tests.
   mocha.run();
 });