assert = require 'assert'
_      = require 'underscore'
Understream = require "#{__dirname}/../index"

describe '_.transform', ->
  it 'works with a custom _transform function', (done) ->
    input = ['a', 'b', 'c']
    new Understream(input).transform (chunk, enc, cb) ->
      @push chunk
      @push chunk + '2'
      cb()
    .run (err, data) ->
      assert.ifError err
      assert.deepEqual data, ['a', 'a2', 'b', 'b2', 'c', 'c2']
      done()

  it 'works with a custom _flush function', (done) ->
    input = ['a', 'b', 'c']
    new Understream(input).transform(
      (chunk, enc, cb) -> @push chunk; cb()
    , (cb) -> @push 'bye bye'; cb()
    ).run (err, data) ->
      assert.ifError err
      assert.deepEqual data, input.concat ['bye bye']
      done()