[Fix] mount: state and setState should throw an explicit error …

…message when called on an SFC
enzymejs · Aug 16, 2018 · 8feee5a · 8feee5a
1 parent 12b150f
commit 8feee5a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/enzyme/src/ReactWrapper.js b/packages/enzyme/src/ReactWrapper.js
@@ -302,6 +302,9 @@ class ReactWrapper {
     if (this[ROOT] !== this) {
       throw new Error('ReactWrapper::setState() can only be called on the root');
     }
+    if (this.instance() === null || this[RENDERER].getNode().nodeType === 'function') {
+      throw new Error('ReactWrapper::setState() can only be called on class components');
+    }
     if (typeof callback !== 'function') {
       throw new TypeError('ReactWrapper::setState() expects a function as its second argument');
     }
@@ -622,6 +625,9 @@ class ReactWrapper {
     if (this[ROOT] !== this) {
       throw new Error('ReactWrapper::state() can only be called on the root');
     }
+    if (this.instance() === null || this[RENDERER].getNode().nodeType === 'function') {
+      throw new Error('ReactWrapper::state() can only be called on class components');
+    }
     const _state = this.single('state', () => this.instance().state);
     if (typeof name !== 'undefined') {
       return _state[name];