Put provided function in correct context

brycebaril · Aug 7, 2013 · 22051de · 22051de
1 parent 3aa8152
commit 22051de
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/index.js b/index.js
@@ -8,12 +8,14 @@ function ctor(options, fn) {
     fn = options
     options = {}
   }
-  options._index = 0
-  return through2.ctor(options, function (chunk, encoding, callback) {
+
+  var Map = through2.ctor(options, function (chunk, encoding, callback) {
     if (this.options.wantStrings) chunk = chunk.toString()
-    this.push(fn(chunk, this.options._index++))
+    this.push(fn.call(this, chunk, this._index++))
     return callback()
   })
+  Map.prototype._index = 0
+  return Map
 }
 
 function make(options, fn) {

diff --git a/test/index.js b/test/index.js
@@ -29,9 +29,10 @@ test("ctor", function (t) {
 })
 
 test("ctor options", function (t) {
-  t.plan(2)
+  t.plan(7)
 
-  var Map = map.ctor({objectMode: true}, function (record) {
+  var Map = map.ctor({objectMode: true, foo: "bar"}, function (record) {
+    t.equals(this.options.foo, "bar", "can see options")
     record.foo.toUpperCase()
     return record
   })