WebUI Testing: async support - global mocking, deferred runs, continued run.

chrome/test/data/webui/print_preview.js

 // Copyright (c) 2011 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.
 
 /**
  * Test fixture for print preview WebUI testing.
  * @extends {testing.Test}
  * @constructor
  **/
 function PrintPreviewWebUITest() {}
 
 PrintPreviewWebUITest.prototype = {
   __proto__: testing.Test.prototype,
 
   /**
-   * Browse to the sample page, cause print preview & call PreLoad().
+   * Browse to the sample page, cause print preview & call preLoad().
    * @type {string}
    * @override
    **/
   browsePrintPreload: 'print_preview_hello_world_test.html',
 
   /**
    * Register a mock handler to ensure expectations are met and print preview
    * behaves correctly.
    * @type {Function}
    * @override
    **/
-  PreLoad: function() {
+  preLoad: function() {
     // TODO(scr) remove this after tests pass consistently.
-    console.info('PreLoad');
+    console.info('preLoad');
 
     /**
      * Create a handler class with empty methods to allow mocking to register
      * expectations and for registration of handlers with chrome.send.
+     * @constructor
      **/
     function MockPrintPreviewHandler() {}
 
     MockPrintPreviewHandler.prototype = {
       getDefaultPrinter: function() {},
       getPrinters: function() {},
       getPreview: function(settings) {},
       print: function(settings) {},
       getPrinterCapabilities: function(printerName) {},
       showSystemDialog: function() {},
@@ skipping 17 matching lines @@
           setDefaultPrinter('FooDevice');
         }));
     mockHandler.stubs().getPrinterCapabilities(NOT_NULL).
         will(callFunction(function() {
           updateWithPrinterCapabilities({
             disableColorOption: true,
             setColorAsDefault: true,
             disableCopiesOption: true,
           });
         }));
-    mockHandler.stubs().getPreview(NOT_NULL).
-        will(callFunction(function() {
-          updatePrintPreview('title', true, 1, 1);
+    var savedArgs = new SaveMockArguments();
+    mockHandler.stubs().getPreview(savedArgs.match(NOT_NULL)).
+        will(callFunctionWithSavedArgs(savedArgs, function(options) {
+          updatePrintPreview('title', true, 1, JSON.parse(options).requestID);
         }));
 
     mockHandler.stubs().getPrinters().
         will(callFunction(function() {
           setUseCloudPrint(false, "");

[mmenke, 2011/08/23 15:26:29]

nit:  Single quotes.

[Sheridan Rawlins, 2011/08/24 02:33:53]

Done.

           setPrinters([{
               printerName: 'FooName',
               deviceName: 'FooDevice',
             }, {
               printerName: 'BarName',
               deviceName: 'BarDevice',
             },
           ]);
         }));
 
     // Register mock as a handler of the chrome.send messages.
     registerMockMessageCallbacks(mockHandler, MockPrintPreviewHandler);
 
+    /**
+     * Create a class to hold global functions to watch for.
+     * @constructor
+     **/
+    function MockGlobals() {}
+
+    MockGlobals.prototype = {
+      updateWithPrinterCapabilities: function(settingInfo) {},
+    };
+
+    var mockGlobals = this.mockGlobals = mock(MockGlobals);
+    mockGlobals.stubs().updateWithPrinterCapabilities(
+        savedArgs.match(ANYTHING)).
+            will(callGlobalWithSavedArgs(
+                savedArgs, 'updateWithPrinterCapabilities'));
+
+    // Register globals to mock out for us.
+    registerMockGlobals(mockGlobals, MockGlobals);
+
     // Override checkCompatiblePluginExists to return a value consistent with
     // the state being tested and stub out the pdf viewer if it doesn't exist,
     // such as on non-official builds. When the plugin exists, use the real
     // thing.
     var self = this;
     window.addEventListener('DOMContentLoaded', function() {
       if (!this.checkCompatiblePluginExists()) {
         // TODO(scr) remove this after tests pass consistently.
         console.info('no PDF Plugin; providing fake methods.');
         this.createPDFPlugin = self.createPDFPlugin;
@@ skipping 207 matching lines @@
           disableLandscapeOption: false,
         });
       }));
 
   updateControlsWithSelectedPrinterCapabilities();
   expectFalse(colorSettings.colorRadioButton.checked);
   expectTrue(colorSettings.bwRadioButton.checked);
 });
 
 // Test that changing the selected printer updates the preview.
-TEST_F('PrintPreviewWebUITest', 'TestPrinterChangeUpdatesPreview',
-       function() {
-  var matchAnythingSave = new SaveArgumentsMatcher(ANYTHING);
+TEST_F('PrintPreviewWebUITest', 'TestPrinterChangeUpdatesPreview', function() {
+  var savedArgs = new SaveMockArguments();
+  this.mockHandler.expects(once()).getPreview(savedArgs.match(ANYTHING)).
+      will(callFunctionWithSavedArgs(savedArgs, function(options) {
+        updatePrintPreview('title', true, 2,
+                           JSON.parse(options).requestID);
+      }));
 
-  this.mockHandler.expects(once()).getPreview(matchAnythingSave).
-      will(callFunction(function() {
-        updatePrintPreview('title', true, 2,
-                           matchAnythingSave.argument.requestID);
-      }));
+  this.mockGlobals.expects(once()).updateWithPrinterCapabilities(
+      savedArgs.match(ANYTHING)).
+          will(callGlobalWithSavedArgs(
+              savedArgs, 'updateWithPrinterCapabilities'));
 
   var printerList = $('printer-list');
   assertNotEquals(null, printerList, 'printerList');
   assertGE(printerList.options.length, printerListMinLength);
   expectEquals(fooIndex, printerList.selectedIndex,
                'fooIndex=' + fooIndex);
   var oldLastPreviewRequestID = lastPreviewRequestID;
   ++printerList.selectedIndex;
   updateControlsWithSelectedPrinterCapabilities();
   expectNotEquals(oldLastPreviewRequestID, lastPreviewRequestID);
@@ skipping 28 matching lines @@
 
 // Test that error message is displayed when plugin doesn't exist.
 TEST_F('PrintPreviewNoPDFWebUITest', 'TestErrorMessage', function() {
   var errorButton = $('error-button');
   assertNotEquals(null, errorButton);
   expectFalse(errorButton.disabled);
   var errorText = $('error-text');
   assertNotEquals(null, errorText);
   expectFalse(errorText.classList.contains('hidden'));
 });