You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.6 KiB
93 lines
2.6 KiB
'use strict';
|
|
|
|
const createAstUtils = require('../util/ast');
|
|
|
|
function newDescribeLayer(describeNode) {
|
|
return {
|
|
describeNode,
|
|
hookNodes: [],
|
|
testCount: 0
|
|
};
|
|
}
|
|
|
|
module.exports = {
|
|
meta: {
|
|
type: 'suggestion',
|
|
docs: {
|
|
description: 'Disallow hooks for a single test or test suite'
|
|
},
|
|
schema: [
|
|
{
|
|
type: 'object',
|
|
properties: {
|
|
allow: {
|
|
type: 'array',
|
|
items: {
|
|
type: 'string'
|
|
}
|
|
}
|
|
}
|
|
}
|
|
]
|
|
},
|
|
create(context) {
|
|
const astUtils = createAstUtils(context.settings);
|
|
const options = context.options[0] || {};
|
|
const allowedHooks = options.allow || [];
|
|
let layers = [];
|
|
|
|
function increaseTestCount() {
|
|
layers = layers.map((layer) => {
|
|
return {
|
|
describeNode: layer.describeNode,
|
|
hookNodes: layer.hookNodes,
|
|
testCount: layer.testCount + 1
|
|
};
|
|
});
|
|
}
|
|
|
|
function popLayer(node) {
|
|
const layer = layers[layers.length - 1];
|
|
if (layer.describeNode === node) {
|
|
if (layer.testCount <= 1) {
|
|
layer.hookNodes
|
|
.filter(function (hookNode) {
|
|
return !allowedHooks.includes(hookNode.name);
|
|
})
|
|
.forEach(function (hookNode) {
|
|
context.report({
|
|
node: hookNode,
|
|
message: `Unexpected use of Mocha \`${ hookNode.name }\` hook for a single test case`
|
|
});
|
|
});
|
|
}
|
|
layers.pop();
|
|
}
|
|
}
|
|
|
|
return {
|
|
Program(node) {
|
|
layers.push(newDescribeLayer(node));
|
|
},
|
|
|
|
CallExpression(node) {
|
|
if (astUtils.isDescribe(node)) {
|
|
increaseTestCount();
|
|
layers.push(newDescribeLayer(node));
|
|
return;
|
|
}
|
|
|
|
if (astUtils.isTestCase(node)) {
|
|
increaseTestCount();
|
|
}
|
|
|
|
if (astUtils.isHookIdentifier(node.callee)) {
|
|
layers[layers.length - 1].hookNodes.push(node.callee);
|
|
}
|
|
},
|
|
|
|
'CallExpression:exit': popLayer,
|
|
'Program:exit': popLayer
|
|
};
|
|
}
|
|
};
|