javascript - 如何在 Mocha 中进行多次嵌套测试(通过或失败)后清理文件

Question

我刚刚开始使用 Mocha。我正在编写一个需要清理文件的测试，无论测试是否成功。这就是我正在寻找的东西:

describe('example',function(){
      fs.writeFile(file,'foo', function(){
          it('should succeed', function(done){
             done();
             describe('example2', function(){
                it('should do something awesome', function(done){
                  done();
                });
             })
          }

      }
    });
  });
});
afterAllTests(function(){
  fs.unlink(file);
})

我正在寻找的是在测试期间删除我创建的所有文件。现在，我在上次测试完成后删除它们。但如果任何测试失败，这些文件不会被删除。我确实研究过 after()，但它仅在我的第一次测试后运行。我能看到的唯一其他选项是执行 afterEvery()，并在开头设置一个像 var testFinished = false; 的变量，并将值设置为 true 一旦完成。这是最好的方法吗？

谢谢!

Answer 1

我编写了一个文件系统包装器，用于跟踪写入的文件，然后可以删除所有文件。注意:我只包装了我正在使用的函数。您可以轻松添加更多内容。

// trackedfs.js

var fs = require('fs');

function deleteNoFail(filePath) {
  if (filePath && fs.existsSync(filePath)) {
    if (fs.lstatSync(filePath).isDirectory()) {
      fs.rmdirSync(filePath);
    } else {
      fs.unlinkSync(filePath);
    }
  }
};

function makeTrackedFS(options) {
  options = options || {};
  var f = options.fs || fs;

  var files = [];
  var folders = [];
  var links = [];

  function addFile(file) {
    files.push(file);
  }

  function addFolder(folder) {
    folders.push(folder);
  }

  function addLink(link) {
    links.push(link);
  };

  // Expose these in case we want to manually add stuff
  // like if we shell out to another program we can add
  // its output
  f.addFile = addFile;
  f.addFolder = addFolder;
  f.addLink = addLink;

  f.mkdirSync = function(origFn) {
    return function() {
      addFolder(arguments[0]);
      return origFn.apply(f, arguments);
    };
  }(f.mkdirSync);

  f.symlinkSync = function(origFn) {
    return function() {
      addLink(arguments[1]);
      return origFn.apply(f, arguments);
    };
  }(f.symlinkSync);

  f.writeFileSync = function(origFn) {
    return function() {
      addFile(arguments[0]);
      return origFn.apply(f, arguments);
    };
  }(f.writeFileSync);

  function deleteList(list) {
    list.reverse();  // because we want files before dirs
    list.forEach(function(entry) {
      deleteNoFail(entry);
    });
  };

  // call to delete everything
  f.cleanup = function(options) {
    deleteList(links);
    deleteList(files);
    deleteList(folders);
    links = [];
    files = [];
    folders = [];
  };
};

exports.makeTrackedFS = makeTrackedFS;

我现在可以用以下方式包装 fs

var trackedfs = require('trackedfs');     
trackedfs.makeTrackedFS();

并用它进行清理

function cleanUpOnExit() {
  fs.cleanup();
};

function cleanUpOnExitAndExit() {
  cleanUpOnExit();
  process.exit();
}

process.on('exit', cleanUpEncodersOnExit);
process.on('SIGINT', cleanUpEncodersOnExitAndExit);
process.on('uncaughtException', cleanUpEncodersOnExitAndExit);

这是一个测试

var trackedfs = require('../lib/trackedfs');
var fs = require('fs');

trackedfs.makeTrackedFS();

function cleanUpOnExit() {
  fs.cleanup();
};

function cleanUpOnExitAndExit() {
  cleanUpOnExit();
  process.exit();
}

process.on('exit', cleanUpOnExit);
process.on('SIGINT', cleanUpOnExitAndExit);
process.on('uncaughtException', cleanUpOnExitAndExit);


describe("trackedfs", function() {

  it('makes a directory', function() {
    fs.mkdirSync('foo');
  });

  it('writes a file', function() {
    fs.writeFileSync("foo/bar", "hello");
  });

});

运行它，完成后将不会有 foo 文件夹或 foo/bar 文件。

一个问题是它可能无法删除仍打开的文件，因此如果您打开一个文件然后使测试崩溃，它可能无法删除它。