From af37323172b98441d02aa02447f795f5766419b0 Mon Sep 17 00:00:00 2001
From: Alex LaFroscia <alexlafroscia@nestlabs.com>
Date: Fri, 2 Dec 2016 17:25:08 -0800
Subject: [PATCH] Allow passing in an alternate scheduler function

---
 nprogress.js | 11 ++++++-----
 package.json |  3 ++-
 test/test.js | 13 ++++++++++++-
 3 files changed, 20 insertions(+), 7 deletions(-)

diff --git a/nprogress.js b/nprogress.js
index 708bee1..f86ed40 100644
--- a/nprogress.js
+++ b/nprogress.js
@@ -27,7 +27,8 @@
     barSelector: '[role="bar"]',
     spinnerSelector: '[role="spinner"]',
     parent: 'body',
-    template: '<div class="bar" role="bar"><div class="peg"></div></div><div class="spinner" role="spinner"><div class="spinner-icon"></div></div>'
+    template: '<div class="bar" role="bar"><div class="peg"></div></div><div class="spinner" role="spinner"><div class="spinner-icon"></div></div>',
+    scheduler: window.setTimeout
   };
 
   /**
@@ -88,18 +89,18 @@
         });
         progress.offsetWidth; /* Repaint */
 
-        setTimeout(function() {
+        Settings.scheduler(function() {
           css(progress, {
             transition: 'all ' + speed + 'ms linear',
             opacity: 0
           });
-          setTimeout(function() {
+          Settings.scheduler(function() {
             NProgress.remove();
             next();
           }, speed);
         }, speed);
       } else {
-        setTimeout(next, speed);
+        Settings.scheduler(next, speed);
       }
     });
 
@@ -121,7 +122,7 @@
     if (!NProgress.status) NProgress.set(0);
 
     var work = function() {
-      setTimeout(function() {
+      Settings.scheduler(function() {
         if (!NProgress.status) return;
         NProgress.trickle();
         work();
diff --git a/package.json b/package.json
index 28f7471..cbca09d 100644
--- a/package.json
+++ b/package.json
@@ -17,7 +17,8 @@
     "jquery": "^2.1.1",
     "jsdom": "^5.4.3",
     "mocha": "^2.2.4",
-    "mocha-jsdom": "^0.3.0"
+    "mocha-jsdom": "^0.3.0",
+    "testdouble": "^1.9.0"
   },
   "dependencies": {},
   "jspm": {
diff --git a/test/test.js b/test/test.js
index a381b09..bfff028 100644
--- a/test/test.js
+++ b/test/test.js
@@ -5,6 +5,7 @@
 
   var root = this;
   var assert = (root.chai || require('chai')).assert;
+  var td = require('testdouble');
 
   describe('NProgress', function() {
     var $, NProgress;
@@ -170,6 +171,16 @@
         assert.equal($("#nprogress .spinner").length, 0);
       });
     });
-  });
 
+    describe('.configure(scheduler)', function() {
+      it('should allow for an alternate scheduler to be provided', function() {
+        var scheduler = td.function('custom scheduler');
+
+        NProgress.configure({ scheduler: scheduler });
+        NProgress.start();
+
+        td.verify(scheduler(td.matchers.isA(Function), td.matchers.isA(Number)));
+      });
+    });
+  });
 })();