jscodeshift update assert.expect

update-assert-expect.js

export default function transformer(file, api) {
  const j = api.jscodeshift;

  return j(file.source)
    .find(j.CallExpression, {
      callee: {
        type: 'Identifier',
        name: 'test',
      },
    })
    .forEach((path) => {
      const assertCall = j.memberExpression(
        j.identifier('assert'),
        j.identifier('expect')
      );
      const assertUsage = j(path.node.arguments[1].body).find(j.Identifier, {
        name: 'assert',
      }).length;
      if (assertUsage > 0) {
        const assertStatement = j.expressionStatement(
          j.callExpression(assertCall, [j.literal(assertUsage)])
        );
        const existingExpectCalls = j(path.node.arguments[1].body)
          .find(j.CallExpression, { callee: assertCall })
          .forEach((callPath) => {
            const arg = callPath.node.arguments[0];
            if (j.Literal.check(arg) && typeof arg.value === 'number') {
              arg.value += assertUsage;
            } else {
              const addition = j.binaryExpression(
                '+',
                arg,
                j.literal(assertUsage)
              );
              callPath.node.arguments[0] = addition;
            }
          });
        const blockStatement = path.node.arguments[1].body;
        j(blockStatement).find(j.Statement).at(0).insertBefore(assertStatement);
      } else {
        const assertStatement = j.variableDeclaration('const', [
          j.variableDeclarator(
            j.identifier('assert'),
            j.memberExpression(
              j.callExpression(j.identifier('require'), [j.literal('assert')]),
              j.identifier('strict')
            )
          ),
        ]);
        const assertExpectCall = j.callExpression(assertCall, [j.literal(0)]);
        const assertExpectStatement = j.expressionStatement(assertExpectCall);
        const blockStatement = path.node.arguments[1].body;
        j(blockStatement)
          .find(j.Statement)
          .at(0)
          .insertBefore(assertExpectStatement);
        j(blockStatement).find(j.Statement).at(0).insertBefore(assertStatement);
      }
    })
    .toSource();
}