Add data reduction proxy debug info to net-internals#bandwidth

chrome/test/data/webui/net_internals/bandwidth_view.js

 // Copyright (c) 2012 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.
 
 // Include test fixture.
 GEN_INCLUDE(['net_internals_test.js']);
 
 // Anonymous namespace
 (function() {
 
@@ skipping 11 matching lines @@
  */
 function BandwidthTask(expectedLength, faviconLength) {
   NetInternalsTest.Task.call(this);
   this.url_ = null;
   this.expectedLength_ = expectedLength;
   this.faviconLength_ = faviconLength;
   this.sessionVerified = false;
   this.historicVerified = false;
 }
 
+/**
+ * The task loads data reduction proxy info.
+ *
+ * Checks that we see all relevant events and update the corresponding elements.
+ *
+ * @param {boolean} isEnabled The desired state of the data reduction proxy.
+ * @extends {NetInternalsTest.Task}
+ * @constructor
+ */
+function DataReductionProxyTask(isEnabled) {
+  NetInternalsTest.Task.call(this);
+  this.enabled_ = isEnabled;
+  this.dataReductionProxyInfoVerified_ = false;
+}
+
 BandwidthTask.prototype = {
   __proto__: NetInternalsTest.Task.prototype,
 
   /**
    * Switches to the bandwidth tab, loads a page in the background, and waits
    * for the arrival of network statistics.
    *
    * @param {string} url URL to be fetched.
    */
   start: function(url) {
     assertEquals('string', typeof url);
     this.url_ = url;
     g_browser.addSessionNetworkStatsObserver(this, true);
     g_browser.addHistoricNetworkStatsObserver(this, false);
     NetInternalsTest.switchToView('bandwidth');
     chrome.send('loadPage', [this.url_]);
   },
 
   /**
    * Returns the float value the specified cell of the bandwidth table.
    */
   getBandwidthTableCell_: function(row, col) {
     return parseFloat(NetInternalsTest.getTbodyText(
-        BandwidthView.MAIN_BOX_ID, row, col));
+        BandwidthView.STATS_BOX_ID, row, col));
   },
 
   /**
    * Confirms that the bandwidth usage table displays the expected values.
    * Does not check exact displayed values, to avoid races, but makes sure
    * values are high enough after an event of interest.
    *
    * @param {number} col The column of the table to validate, either 1 or 2.
    * @param {number} expectedReceived Expected received content length.
    * @param {number} expectedOriginal Expected original content length.
@@ skipping 14 matching lines @@
     expectFalse(isNaN(row4));
   },
 
   /**
    * A task is complete only when session and historic counters have been
    * verified to reflect the expected number of bytes received.
    */
   completeIfDone: function() {
     if (this.historicVerified && this.sessionVerified) {
       // Check number of rows in the table.
-      NetInternalsTest.checkTbodyRows(BandwidthView.MAIN_BOX_ID, 4);
+      NetInternalsTest.checkTbodyRows(BandwidthView.STATS_BOX_ID, 4);
       this.onTaskDone();
     }
   },
 
   /**
    * SessionNetworkStatsObserver function.  Sanity checks the received data
    * and constructed table.
 
    * @param {object} networkStats State of the network session.
    */
@@ skipping 28 matching lines @@
     if (!isNaN(this.getBandwidthTableCell_(0, 2))) {
       expectLE(expectedLength, networkStats.historic_original_content_length);
       // Column 2 contains historic information.
       this.validateBandwidthTableColumn_(2, expectedLength, expectedLength);
       this.historicVerified = true;
       this.completeIfDone();
     }
   }
 };
 
+DataReductionProxyTask.prototype = {
+  __proto__: NetInternalsTest.Task.prototype,
+
+  /**
+   * Switches to the bandwidth tab and waits for arrival of data reduction
+   * proxy information.
+   */
+  start: function() {
+    chrome.send('enableDataReductionProxy', [this.enabled_]);
+    g_browser.addDataReductionProxyInfoObserver(this, true);
+    NetInternalsTest.switchToView('bandwidth');
+  },
+
+  /**
+   * A task is complete only when session and historic counters have been
+   * verified to reflect the expected number of bytes received.
+   */
+  completeIfDone: function() {
+    if (this.dataReductionProxyInfoVerified_) {
+      this.onTaskDone();
+    }
+  },
+
+  /**
+   * DataReductionProxyInfoObserver function.  Sanity checks the received data
+   * and constructed table.
+
+   * @param {object} info State of the data reduction proxy.
+   */
+  onDataReductionProxyInfoChanged: function(info) {
+    if (this.isDone() || this.dataReductionProxyInfoVerified_)
+      return;
+
+    if (info) {
+      if (this.enabled_) {
+        expectEquals("true", info.enabled);
+        expectEquals("Enabled", $(BandwidthView.ENABLED_ID).innerText);
+        expectNotEquals('', $(BandwidthView.PRIMARY_PROXY_ID).innerText);
+        expectNotEquals('', $(BandwidthView.SECONDARY_PROXY_ID).innerText);
+      } else {
+        expectEquals("false", info.enabled);
+        expectEquals("Disabled", $(BandwidthView.ENABLED_ID).innerText);
+        expectEquals('', $(BandwidthView.PRIMARY_PROXY_ID).innerText);
+        expectEquals('', $(BandwidthView.SECONDARY_PROXY_ID).innerText);
+        // Each event results in 2 rows, and we get 2 events since the startup
+        // event starts as disabled, and we subsequently manually set it to the
+        // disabled state.
+        expectEquals(
+          4, NetInternalsTest.getTbodyNumRows(BandwidthView.EVENTS_TBODY_ID));
+      }
+
+      this.dataReductionProxyInfoVerified_ = true;
+      this.completeIfDone();
+    }
+
+  }
+};
+
 /**
  * Loads a page and checks bandwidth statistics.
  */
 TEST_F('NetInternalsTest', 'netInternalsSessionBandwidthSucceed', function() {
   var taskQueue = new NetInternalsTest.TaskQueue(true);
   taskQueue.addTask(
       new NetInternalsTest.GetTestServerURLTask('files/title1.html'));
   // Load a page with a content length of 66 bytes and a 45-byte favicon.
   taskQueue.addTask(new BandwidthTask(66, 45));
   taskQueue.run();
 });
 
+/**
+ * Checks data reduction proxy info when it is enabled.
+ */
+TEST_F('NetInternalsTest', 'netInternalsDataReductionProxyEnabled', function() {
+  var taskQueue = new NetInternalsTest.TaskQueue(true);
+  taskQueue.addTask(new DataReductionProxyTask(true));
+  taskQueue.run();
+});
+
+/**
+ * Checks data reduction proxy info when it is disabled.
+ */
+TEST_F('NetInternalsTest',
+    'netInternalsDataReductionProxyDisabled', function() {
+  var taskQueue = new NetInternalsTest.TaskQueue(true);
+  taskQueue.addTask(new DataReductionProxyTask(false));
+  taskQueue.run();
+});
+
 })();  // Anonymous namespace