Should also work with objects

alfredwesterveld · Jul 4, 2014 · b06ed2b · b06ed2b
1 parent f9a4052
commit b06ed2b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -6,8 +6,10 @@ const util = require("util")
 
 // used http://stackoverflow.com/a/2946616/11926 as baseline
 function getAllMethods(object) {
-    return Object.getOwnPropertyNames(object.prototype).filter(function(property) {
-        return typeof object.prototype[property] === "function"
+    const obj = object.prototype || object
+
+    return Object.getOwnPropertyNames(obj).filter(function(property) {
+        return typeof obj[property] === "function"
     })
 }
 

diff --git a/test/test.implement.js b/test/test.implement.js
@@ -41,4 +41,9 @@ describe("implement", function () {
             done()        
         }
     })
+
+    it("Should also work with objects", function (done) {
+        implement(new ArrayQueue(), new Queue())
+        done()
+    })
 })