@incdevco/framework/mock/index.js

node.js lambda framework

var Mocked = require('./mocked');

function Mock() {
  'use strict';

  this.mockeds = [];

}

Mock.prototype.done = function (done) {
  'use strict';
  
  this.restore();
  
  this.mockeds.forEach(function (mocked) {

    mocked.done();

  });
  
  if (done) {
  
    done();
  
  }
  
};

Mock.prototype.mock = function (obj) {
  'use strict';

  var mocked = new Mocked(obj);

  //console.log('mocked', new Error().stack);

  this.mockeds.push(mocked);

  return mocked;

};

Mock.prototype.restore = function () {
  'use strict';
  
  this.mockeds.forEach(function(mocked) {
    
    mocked.restore();
    
  });
  
};

module.exports = Mock;

module.exports.Angular = require('./angular');