a_mock/partialMock/expectCore.js

Sub package of a. Mocking framework

function expect(hasCorrectArgument,index,mockContext) {
	var expect = require('./expect');
	var expectAnything = require('./expectAnything');
	var ignore = require('./ignore');
	var expectArray = require('./expectArray');
	var _return = require('./return');
	var newThrow = require('./newThrow');
	var newThen = require('../newThen');
	mockContext.whenCalledEmitter = require('../eventEmitter')();


	var c = {};
	mockContext.compositeAreCorrectArguments.add(hasCorrectArgument);
	mockContext.numberOfArgs = index+1;

	c.expect = function() {
		var args = [index+1,mockContext];
		for (var i = 0; i < arguments.length; i++) {
			args.push(arguments[i]);
		};
		return expect.apply(null,args);
	};

	c.expectAnything = function() {
		return expectAnything(index+1,mockContext);
	};

	c.ignore = function() {
		return ignore(index+1,mockContext);
	}

	c.expectArray = function(array) {
		return expectArray(index+1,mockContext,array);
	}

	c.return = function(arg) {
		return _return(arg,index+1,mockContext);
	};

	c.whenCalled = function(callback) {
		mockContext.whenCalledEmitter.add(callback);
		return c;
	};

	c.repeat = function(times) {
		return _return(undefined,index+1,mockContext).repeat(times);
	};

	c.throw = function(error) {
		var _throw = newThrow(error);
		return c.whenCalled(_throw);	
	};

	c.repeatAny = function() {
		return _return(undefined,index+1,mockContext).repeatAny();
	};

	c.resolve = function(value) {
		var promise = newThen.resolve(value);
		return c.return(promise);
	};

	c.reject = function(value) {
		var promise = newThen.reject(value);
		return c.return(promise);
	};

	return c;
}

module.exports = expect;