extract context binding from loop, apply when added (#84)

* extract context binding from loop, apply when added

* pass function as first param to mutate

* corrected to match other test cases

2 files changed, 4 insertions, 4 deletions

diff --git a/fastdom.js b/fastdom.js
index 8579b62..4b7ab70 100644
--- a/fastdom.js
+++ b/fastdom.js
@@ -56,7 +56,7 @@ FastDom.prototype = {
    */
   measure: function(fn, ctx) {
     debug('measure');
-    var task = { fn: fn, ctx: ctx };
+    var task = !ctx ? fn : fn.bind(ctx);
     this.reads.push(task);
     scheduleFlush(this);
     return task;
@@ -72,7 +72,7 @@ FastDom.prototype = {
    */
   mutate: function(fn, ctx) {
     debug('mutate');
-    var task = { fn: fn, ctx: ctx };
+    var task = !ctx ? fn : fn.bind(ctx);
     this.writes.push(task);
     scheduleFlush(this);
     return task;
@@ -203,7 +203,7 @@ function flush(fastdom) {
  */
 function runTasks(tasks) {
   debug('run tasks');
-  var task; while (task = tasks.shift()) task.fn.call(task.ctx);
+  var task; while (task = tasks.shift()) task();
 }
 
 /**
diff --git a/test/fastdom-test.js b/test/fastdom-test.js
index 64f72e5..5dd4ceb 100644
--- a/test/fastdom-test.js
+++ b/test/fastdom-test.js
@@ -340,7 +340,7 @@ suite('fastdom', function() {
 
     test('it removes reference to the job if cleared', function(done) {
       var write = sinon.spy();
-      var id = fastdom.mutate(2, write);
+      var id = fastdom.mutate(write);
 
       fastdom.clear(id);