fs-stream/test/write.js

Utilities that helps manipulating a file system through stream

'use strict';

var fsnode  = require('fs');
var fsx     = require('fs-extra');
var path    = require('path');
var through = require('through2');
var assert  = require('../tools/test.assert.js');

var fs = require('../index.js');

describe('write', function () {
  var root      = path.resolve('.');
  var dir       = path.join('test', 'dir');
  var fullPath1 = path.join(root, dir, 'a.txt');
  var str       = 'Kikoo!';

  function fn (path) {
    assert.isDirOrFile(path, 'file')
    return str;
  }

  beforeEach(function () {
    fsx.ensureFileSync(fullPath1);

    fsnode.writeFileSync(fullPath1, fullPath1, 'utf8');
  });

  afterEach(function () {
    fsx.removeSync(path.join(root, dir));
  });

  function chkAppend(filepath) {
    assert.fileHasContent(filepath, filepath + str, true);
  }

  function chkOverride(filepath) {
    assert.fileHasContent(filepath, str, true);
  }

  var configuration = [
    {param: [str     ], result: chkOverride},
    {param: [str, 'w'], result: chkOverride},
    {param: [str, 'a'], result: chkAppend  },
    {param: [fn,  'w'], result: chkOverride},
    {param: [fn,  'a'], result: chkAppend  }
  ];

  function testRunner(conf) {
    var title = [
      'Write in files with a ',
      typeof conf.param[0],
      ' (write mode: ',
      conf.param[1] === undefined ? 'default' : conf.param[1],
      ')'
    ].join('');

    it(title, function (done) {
      fs(path.join(dir, '*.*'))
        .pipe(fs.write.apply(fs, conf.param))
        .on('end', done)
        .on('error', done)
        .pipe(through.obj(function (file, enc, cb) {
          conf.result(file.path);

          cb(null, file);
        }));
    });
  }

  configuration.forEach(testRunner);


  // Testing asynchronous writing through writing stream

  it('Write in files with a function (write mode: stream)', function (done) {
    var s = {};
    var c = 0;

    fs(path.join(dir, '*.*'))
      .pipe(fs.write(function (stream) {
        assert.isWritableStream(stream)

        s[stream.path] = stream;
        c += 1;

        stream.write(str);
        stream.end(str);
      }))
      .on('error', done)
      .pipe(through.obj(function (file, enc, cb) {
        var stream = s[file.path];

        assert.isWritableStream(stream)

        stream.on('close', function () {
          assert.fileHasContent(file.path, str + str, true)
          c -= 1;

          if (c <= 0) { done(); }
        })

        cb(null, file);
      }));
  })
});