Sinon2 jscodeshift transformer for 3 args stub calls

sinon2-stub-callsFake-transform.js

const _ = require('lodash');

module.exports = function(fileInfo, api) {
	const j = api.jscodeshift;

	const root = j(fileInfo.source);

	root.find(j.CallExpression, {
		callee: {
			type: 'MemberExpression',
			object: {
				type: 'Identifier',
				name: 'sinon'
			},
			property: {
				type: 'Identifier',
				name: 'stub'
			}
		}
	})
		.filter(path => _.get(path, 'value.arguments', []).length == 3)
		.replaceWith(path => {
			const last = _.last(path.value.arguments);
			path.value.arguments = _.initial(path.value.arguments);

			return j.callExpression(j.memberExpression(
				path.value,
				j.identifier('callsFake')
			), [last]);
		});

	return root.toSource();
};

https://github.com/hurrymaplelad/sinon-codemod/