Adding first unit tests with mocks

chrome/test/data/webui/test_api.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.
 
 /**
  * @fileoverview Library providing basic test framework functionality.
  */
 
 /**
  * Namespace for |Test|.
@@ skipping 180 matching lines @@
      * @see registerMockGlobals
      */
     makeAndRegisterMockGlobals: function(functionNames) {
       var MockClass = makeMockClass(functionNames);
       this.mockGlobals = mock(MockClass);
       registerMockGlobals(this.mockGlobals, MockClass);
       return this.mockGlobals;
     },
 
     /**
+      * Create a container of mocked standalone functions to handle
+      * '.'-separated |apiNames|, assign it to |this.mockApis|, register its API
+      * overrides and return it.
+      * @return {Mock} Mock handler class.
+      * @see makeMockFunctions
+      * @see registerMockApis
+      */
+    makeAndRegisterMockApis: function (apiNames) {
+      var apiMockNames = [];
+      for(var i = 0; i < apiNames.length; i++) {
+        apiMockNames.push(apiNames [i].replace(/\./g, '_'));
+      }
+
+      this.mockApis = makeMockFunctions(apiMockNames);
+      registerMockApis(this.mockApis);
+      return this.mockApis;
+    },
+
+    /**
+      * Create a container of mocked standalone functions to handle
+      * |functionNames|, assign it to |this.mockLocalFunctions| and return it.
+      * @return {Mock} Mock handler class.
+      * @see makeMockFunctions
+      */
+    makeMockLocalFunctions: function(functionNames) {
+      this.mockLocalFunctions = makeMockFunctions(functionNames);
+      return this.mockLocalFunctions;
+    },
+
+    /**
      * Override this method to perform initialization during preload (such as
      * creating mocks and registering handlers).
      * @type {Function}
      */
     preLoad: function() {},
 
     /**
      * Override this method to perform tasks before running your test.
      * @type {Function}
      */
@@ skipping 288 matching lines @@
     globalOverrides[name] = {
       object: object,
       callback: callback,
     };
 
     if (!deferGlobalOverrides)
       overrideGlobal(name);
   }
 
   /**
+   * Registers the mock API call and its function.
+   * @param {string} name The '_'-separated name of the API call.
+   * @param {function(...)} theFunction Mock function for this API call.
+   */
+  function registerMockApi(name, theFunction) {
+    var path = name.split('_');
+
+    var namespace = this;
+    for(var i = 0; i < path.length-1; ++i) {
+      var fieldName = path[i];
+      if(!namespace[fieldName])
+        namespace[fieldName] = {};
+
+      namespace = namespace[fieldName];
+    }
+
+    var fieldName = path[path.length-1];
+    assertEquals(undefined, namespace[fieldName]);
+    namespace[fieldName] = theFunction;
+  }
+
+  /**
    * Empty function for use in making mocks.
    * @const
    */
   function emptyFunction() {}
 
   /**
    * Make a mock from the supplied |methodNames| array.
    * @param {Array.<string>} methodNames Array of names of methods to mock.
    * @return {Function} Constructor with prototype filled in with methods
    *     matching |methodNames|.
    */
   function makeMockClass(methodNames) {
     function MockConstructor() {}
     for(var i = 0; i < methodNames.length; i++)
       MockConstructor.prototype[methodNames[i]] = emptyFunction;
     return MockConstructor;
   }
 
   /**
+    * Create a new class to handle |functionNames|, add method 'functions()'
+    * that returns a container of standalone functions based on the mock class
+    * members, and return it.
+    * @return {Mock} Mock handler class.
+    */
+  function makeMockFunctions(functionNames) {
+    var MockClass = makeMockClass(functionNames);
+    var mockFunctions = mock(MockClass);
+    var mockProxy = mockFunctions.proxy();
+
+    mockFunctions._functions = {};

[arv (Not doing code reviews), 2013/07/11 20:59:44]

functions_

[vadimt, 2013/07/11 21:28:10]

Done.

+
+    for (var func in MockClass.prototype) {
+      if (typeof MockClass.prototype[func] === 'function')
+        mockFunctions._functions[func] = mockProxy[func].bind(mockProxy);
+    }
+
+    mockFunctions.functions = function () {
+      return this._functions;
+    };
+
+    return mockFunctions;
+  }
+
+  /**
    * Register all methods of {@code mockClass.prototype} as overrides to global
    * functions of the same name as the method, using the proxy of the
    * |mockObject| to handle the functions.
    * @param {Mock4JS.Mock} mockObject The mock to register callbacks against.
    * @param {function(new:Object)} mockClass Constructor for the mocked class.
    * @see registerMockGlobal
    */
   function registerMockGlobals(mockObject, mockClass) {
     var mockProxy = mockObject.proxy();
     for (var func in mockClass.prototype) {
       if (typeof mockClass.prototype[func] === 'function')
         registerMockGlobal(func, mockProxy, mockProxy[func]);
     }
   }
 
   /**
+   * Register all functions in |mockObject.functions()| as global API calls.
+   * @param {Mock4JS.Mock} mockObject The mock to register callbacks against.
+   * @see registerMockApi
+   */
+  function registerMockApis(mockObject) {
+    var functions = mockObject.functions();
+    for (var func in functions) {
+      if (typeof functions[func] === 'function')
+        registerMockApi(func, functions[func]);
+    }
+  }
+
+  /**
    * Overrides {@code chrome.send} for routing messages to javascript
    * functions. Also falls back to sending with the original chrome object.
    * @param {string} messageName The message to route.
    */
   function send(messageName) {
     var callback = sendCallbacks[messageName];
     if (callback != undefined)
       callback[1].apply(callback[0], Array.prototype.slice.call(arguments, 1));
     else
       this.__proto__.send.apply(this.__proto__, arguments);
@@ skipping 982 matching lines @@
   exports.TEST = TEST;
   exports.TEST_F = TEST_F;
   exports.RUNTIME_TEST_F = TEST_F;
   exports.GEN = GEN;
   exports.GEN_INCLUDE = GEN_INCLUDE;
   exports.WhenTestDone = WhenTestDone;
 
   // Import the Mock4JS helpers.
   Mock4JS.addMockSupport(exports);
 })(this);