prefer-power-assert.js

'use strict';

const {isDeepStrictEqual} = require('node:util');
const espurify = require('espurify');
const {visitIf} = require('enhance-visitors');
const createAvaRule = require('../create-ava-rule');
const util = require('../util');

const notAllowed = [
	'truthy',
	'true',
	'falsy',
	'false',
	'is',
	'not',
	'regex',
	'notRegex',
	'ifError',
];

const assertionCalleeAst = methodName => ({
	type: 'MemberExpression',
	object: {
		type: 'Identifier',
		name: 't',
	},
	property: {
		type: 'Identifier',
		name: methodName,
	},
	computed: false,
});

const skippedAssertionCalleeAst = methodName => ({
	type: 'MemberExpression',
	object: {
		type: 'MemberExpression',
		object: {
			type: 'Identifier',
			name: 't',
		},
		property: {
			type: 'Identifier',
			name: 'skip',
		},
		computed: false,
	},
	property: {
		type: 'Identifier',
		name: methodName,
	},
	computed: false,
});

const isCalleeMatched = (callee, methodName) =>
	isDeepStrictEqual(callee, assertionCalleeAst(methodName))
	|| isDeepStrictEqual(callee, skippedAssertionCalleeAst(methodName));

const create = context => {
	const ava = createAvaRule();

	return ava.merge({
		CallExpression: visitIf([
			ava.isInTestFile,
			ava.isInTestNode,
		])(node => {
			const callee = espurify(node.callee);

			if (callee.type === 'MemberExpression') {
				for (const methodName of notAllowed) {
					if (isCalleeMatched(callee, methodName)) {
						context.report({
							node,
							message: 'Only asserts with no power-assert alternative are allowed.',
						});
					}
				}
			}
		}),
	});
};

module.exports = {
	create,
	meta: {
		type: 'suggestion',
		docs: {
			description: 'Enforce the use of the asserts that have no [power-assert](https://github.com/power-assert-js/power-assert) alternative.',
			url: util.getDocsUrl(__filename),
		},
		schema: [],
	},
};