Add test cases for timeout

asyncmax · Oct 12, 2016 · 6e482df · 6e482df
1 parent 8c5e103
commit 6e482df
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/lib/index.js b/lib/index.js
@@ -6,5 +6,6 @@ module.exports = {
   promisify: require("./promisify"),
   repeat: require("./repeat"),
   forEach: require("./forEach"),
-  queue: require("./queue")
+  queue: require("./queue"),
+  timeout: require("./timeout")
 };
diff --git a/test/test.js b/test/test.js
@@ -274,3 +274,21 @@ test("promisify: error", function(t) {
     t.equal(err.message, "TEST");
   }).then(t.end, t.end);
 });
+
+test("timeout: basic", function(t) {
+  pbox.timeout(Promise.resolve(), 1000).then(function() {
+    t.pass("already resolved promise");
+  }).catch(function() {
+    t.fail("should not get here");
+  }).then(t.end, t.end);
+});
+
+test("timeout: error", function(t) {
+  pbox.timeout(new Promise(function(resolve) {
+    setTimeout(resolve, 500);
+  }), 1).then(function() {
+    t.fail("should not get here");
+  }).catch(function(err) {
+    t.equal(err.code, "TIMEOUT_ERROR");
+  }).then(t.end, t.end);
+});