Convert moar Sheriff-o-Matic tests to Polymer.

Tools/GardeningServer/lib/network-simulator.html

 <!--
 Copyright 2014 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.
 -->
 
 <script>
-function CTFailureGroup(key, failures, annotation) {
-  this.key = key;
-  this.failures = failures;
-  this.annotation = annotation || {};
-  this._computeProperties();
+function NetworkSimulator(assert, done) {
+  this._assert = assert;
+  this._done = done;
+  this._pendingPromises = [];
 }
 
-CTFailureGroup.prototype.snoozeUntil = function(time) {
-  return this._annotate({
-    snoozeTime: time,
+NetworkSimulator._testInProgress = false;
+
+NetworkSimulator.prototype.schedulePromise = function(promise) {
+  this._pendingPromises.push(promise);
+  return promise;
+};
+
+NetworkSimulator.prototype.resolvePromises = function() {
+  var self = this;
+  return new Promise(function(resolve, reject) {
+    var pendingPromises = self._pendingPromises;
+    self._pendingPromises = [];
+    function allResolved(results) {
+      if (self._pendingPromises.length) {
+        resolve(self.resolvePromises());
+        return;
+      }
+      resolve(results);
+    }
+    Promise.all(pendingPromises).then(allResolved, allResolved);
   });
 };
 
-CTFailureGroup.prototype.unsnooze = function() {
-  return this._annotate({
-    snoozeTime: undefined,
+NetworkSimulator.prototype.runTest = function(testCase) {
+  if (NetworkSimulator._testInProgress) {
+    this._assert(false, "runTest calls cannot be nested");
+    this._done();
+    return;
+  }
+
+  NetworkSimulator._testInProgress = true;
+
+  var self = this;
+  return new Promise(function(resolve, reject) {
+    var realNet = window.net;
+
+    function reset() {
+      window.net = realNet;
+      NetworkSimulator._testInProgress = false;
+    }
+
+    // All net.* methods should return promises. This watches all
+    // promises generated by test-overridden methods.
+    window.net = {};
+    ['probe', 'jsonp', 'get', 'post',
+     'ajax', 'json', 'xml'].forEach(function(method) {
+       if (method in self) {
+         net[method] = function() {
+           return self.schedulePromise(self[method].apply(self, arguments));
+         };
+       };
+     });
+
+    try {
+      testCase();
+    } catch(e) {
+      // Make sure errors thrown in the test case don't leave window.net in a bad state.
+      reset();
+      self._assert(false, "Test case threw an error:" + e);
+    }
+
+    self.resolvePromises().then(function() {
+      reset();
+      self._assert(window.net == realNet);
+      resolve();
+    }).catch(function(e) {
+      reset();
+      self._assert(false, "Failed to finish test: " + e);
+    });
   });
 };
-
-CTFailureGroup.prototype._computeProperties = function() {
-  this.isSnoozed = Date.now() < this.annotation.snoozeTime;
-};
-
-CTFailureGroup.prototype._annotate = function(newAnnotation) {
-  // FIXME: Post the new annotation to frontend rather than storing locally.
-  return CTFailureGroup.fetchAnnotations().then(function(annotations) {
-    var annotation = annotations[this.key] || {};
-
-    Object.keys(newAnnotation, function(key, value) {
-      if (value === 'undefined') {
-        delete annotation[key];
-      } else {
-        annotation[key] = value;
-      }
-    });
-
-    if (Object.size(annotation) == 0) {
-      delete annotations[this.key];
-    } else {
-      annotations[this.key] = annotation;
-    }
-
-    localStorage.CTFailureGroupAnnotations = JSON.stringify(annotations);
-
-    this.annotation = annotation;
-    this._computeProperties();
-  }.bind(this));
-};
-
-CTFailureGroup.fetchAnnotations = function() {
-  // FIXME: Fetch annotations from frontend.
-  var stored = localStorage.CTFailureGroupAnnotations;
-  var annotations = stored ? JSON.parse(stored) : {};
-  return Promise.resolve(annotations);
-};
 </script>