Use Iterable.isIterable for Immutable check

astorije · Dec 12, 2015 · b34220f · b34220f
1 parent 6a9c9ed
commit b34220f
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/chai-immutable.js b/chai-immutable.js
@@ -24,6 +24,10 @@
 
     var Assertion = chai.Assertion;
 
+    function isImmutable(obj) {
+      return Immutable.Iterable.isIterable(obj);
+    }
+
     /**
      * ## BDD API Reference
      */
@@ -46,7 +50,7 @@
       return function () {
         var obj = this._obj;
 
-        if (obj && obj instanceof Collection) {
+        if (isImmutable(obj)) {
           var size = obj.size;
           new Assertion(size).a('number');
 
@@ -96,7 +100,7 @@
       return function (collection) {
         var obj = this._obj;
 
-        if (obj && obj instanceof Collection) {
+        if (isImmutable(obj)) {
           this.assert(
             Immutable.is(obj, collection),
             'expected #{act} to equal #{exp}',
@@ -141,7 +145,7 @@
       return function (val) {
         var obj = this._obj;
 
-        if (obj && obj instanceof Collection) {
+        if (isImmutable(obj)) {
           this.assert(
             obj.includes(val),
             'expected #{act} to include #{exp}',