如何在Mocha中进行多次嵌套测试后清理文件,传递或失败

时间:2015-10-24 16:13:03

标签: javascript unit-testing mocha

我刚刚开始使用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。这是最好的方式吗?

谢谢!

1 个答案:

答案 0 :(得分: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文件。

一个问题是,它可能无法删除仍然打开的文件,因此如果您打开文件然后导致测试崩溃,则可能无法将其删除。