javascript - Node - 遍历 JS 文件的文件夹并提取某些位

Question

我目前正在以电子表格形式维护一份测试文档 - 所以我有 3 列:

• 文件名
• 来自 it 的文本 block - 这是摩卡测试 it('does this thing', function(){ ... })
• 描述 - 我正在写。

但是我不喜欢这个想法，因为维护起来很耗时。

是否可以自动执行此操作，我认为存在一些 npm 包。因此，这将运行一个文件夹，获取文件名并查看每个 JS 以在某些定义的字符串正则表达式之后收集信息。例如，查找 //DESCRIPTION 之后的任何文本和it( - 像这样:

//DESCRIPTION: Some text here
it('does this thing', function(){ ... })

Answer 1

使用esprima获取AST，过滤出想要的测试 block ，通过escodegen生成源代码

这是代码:

const fs = require('fs')
const path = require('path')
const esprima = require('esprima')
const escodegen = require('escodegen')


function walkSync (dir) {
  return fs.statSync(dir).isDirectory()
    ? Array.prototype.concat(...fs.readdirSync(dir).map(f => walkSync(path.join(dir, f))))
    : dir
}

const result = walkSync('./test').map(path => {
  const ast = esprima.parse(fs.readFileSync(path, 'utf8'))
  ast.body = ast.body.filter(elem => {
    return elem.type === 'ExpressionStatement'
      && elem.expression && elem.expression.callee
      && elem.expression.callee.name === 'it'
  })
  const code = escodegen.generate(ast)
  return {
    path,
    code
  }
})

console.log(result)

测试文件夹结构如下:

测试用例是什么样的:

执行结果如下: