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

describe '_.stream', ->

  it 'wraps an object implementing the Readable stream interface', (done) ->
    {Readable} = require 'stream'
    input = ['a', 'b', 'c']
    rs = new Readable objectMode: true
    rs.push item for item in input
    rs.push null
    new Understream(rs).run (err, result) ->
      assert.ifError err
      assert.deepEqual input, result
      done()

  it 'wraps a mongoose stream', (done) ->
    mongoose.connect "#{process.env.MONGO_URL}/test-understream", {useNewUrlParser: true}
    Doc = mongoose.model "Doc", new mongoose.Schema { foo: String }
    input = ['a', 'b', 'c']
    async.waterfall [
      (cb_wf) -> Doc.remove (err, dontcare) -> cb_wf err
      (cb_wf) ->
        async.forEach input, (str, cb_fe) ->
          new Doc({foo: str}).save cb_fe
        , (err, dontcare) ->
          cb_wf err
      (cb_wf) ->
        cnt = 0
        new Understream(Doc.find().stream()).each (doc) ->
          cnt++
          assert doc.foo in input
        .run (err) ->
          assert.ifError err
          assert.equal cnt, 3
          cb_wf()
    ], done