src/third_party/WebKit/Tools/GardeningServer/lib/network-simulator.html

   1 <!--
   2 Copyright 2014 The Chromium Authors. All rights reserved.
   3 Use of this source code is governed by a BSD-style license that can be
   4 found in the LICENSE file.
   5 -->
   6
   7 <script>
   8 function NetworkSimulator(assert, done) {
   9   this._assert = assert;
  10   this._done = done;
  11   this._pendingPromises = [];
  12 }
  13
  14 NetworkSimulator._testInProgress = false;
  15
  16 NetworkSimulator.prototype.schedulePromise = function(promise) {
  17   this._pendingPromises.push(promise);
  18   return promise;
  19 };
  20
  21 NetworkSimulator.prototype.resolvePromises = function() {
  22   var self = this;
  23   return new Promise(function(resolve, reject) {
  24     var pendingPromises = self._pendingPromises;
  25     self._pendingPromises = [];
  26     function allResolved(results) {
  27       if (self._pendingPromises.length) {
  28         resolve(self.resolvePromises());
  29         return;
  30       }
  31       resolve(results);
  32     }
  33     Promise.all(pendingPromises).then(allResolved, allResolved);
  34   });
  35 };
  36
  37 NetworkSimulator.prototype.runTest = function(testCase) {
  38   if (NetworkSimulator._testInProgress) {
  39     this._assert(false, "runTest calls cannot be nested");
  40     this._done();
  41     return;
  42   }
  43
  44   NetworkSimulator._testInProgress = true;
  45
  46   var self = this;
  47   return new Promise(function(resolve, reject) {
  48     var realNet = window.net;
  49
  50     function reset() {
  51       window.net = realNet;
  52       NetworkSimulator._testInProgress = false;
  53     }
  54
  55     // All net.* methods should return promises. This watches all
  56     // promises generated by test-overridden methods.
  57     window.net = {};
  58     ['probe', 'jsonp', 'get', 'post',
  59      'ajax', 'json', 'xml'].forEach(function(method) {
  60        if (method in self) {
  61          net[method] = function() {
  62            return self.schedulePromise(self[method].apply(self, arguments));
  63          };
  64        };
  65      });
  66
  67     try {
  68       testCase();
  69     } catch(e) {
  70       // Make sure errors thrown in the test case don't leave window.net in a bad state.
  71       reset();
  72       self._assert(false, "Test case threw an error:" + e);
  73     }
  74
  75     self.resolvePromises().then(function() {
  76       reset();
  77       self._assert(window.net == realNet);
  78       resolve();
  79     }).catch(function(e) {
  80       reset();
  81       self._assert(false, "Failed to finish test: " + e);
  82     });
  83   });
  84 };
  85 </script>