[Fix] shallow/mount: restore fallback when adapter lacks `invokeS…

…etStateCallback`
enzymejs · Aug 16, 2018 · 093b2ed · 093b2ed
1 parent ec3beef
commit 093b2ed
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/packages/enzyme/src/ReactWrapper.js b/packages/enzyme/src/ReactWrapper.js
@@ -313,7 +313,11 @@ class ReactWrapper {
       if (callback) {
         const adapter = getAdapter(this[OPTIONS]);
         const instance = this.instance();
-        adapter.invokeSetStateCallback(instance, callback);
+        if (adapter.invokeSetStateCallback) {
+          adapter.invokeSetStateCallback(instance, callback);
+        } else {
+          callback.call(instance);
+        }
       }
     });
     return this;

diff --git a/packages/enzyme/src/ShallowWrapper.js b/packages/enzyme/src/ShallowWrapper.js
@@ -421,6 +421,9 @@ class ShallowWrapper {
     if (this.instance() === null || this[RENDERER].getNode().nodeType === 'function') {
       throw new Error('ShallowWrapper::setState() can only be called on class components');
     }
+    if (arguments.length > 1 && typeof callback !== 'function') {
+      throw new TypeError('ReactWrapper::setState() expects a function as its second argument');
+    }
     this.single('setState', () => {
       withSetStateAllowed(() => {
         const adapter = getAdapter(this[OPTIONS]);
@@ -481,7 +484,11 @@ class ShallowWrapper {
         this.update();
         // call the setState callback
         if (callback) {
-          adapter.invokeSetStateCallback(instance, callback);
+          if (adapter.invokeSetStateCallback) {
+            adapter.invokeSetStateCallback(instance, callback);
+          } else {
+            callback.call(instance);
+          }
         }
       });
     });