improve Enzyme return types

voodoocreation · Dec 12, 2019 · 0539064 · 0539064
1 parent 30fcaa1
commit 0539064
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/package.json b/package.json
@@ -3,7 +3,7 @@
   "author": "Raice Hannay <voodoocreation@gmail.com>",
   "description": "A set of classes to make setting up React components for unit tests easy.",
   "license": "ISC",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "keywords": [
     "component",
     "enzyme",

diff --git a/src/Wrapper.tsx b/src/Wrapper.tsx
@@ -54,7 +54,7 @@ export default class Wrapper<
     this.beforeMount();
 
     const props = this.defineProps() as P;
-    const wrapper = mount<C>(<this.Component {...props} />, {
+    const wrapper = mount<P>(<this.Component {...props} />, {
       wrappingComponent: this.WrappingComponent
     });
 
@@ -67,7 +67,7 @@ export default class Wrapper<
     this.beforeMount();
 
     const props = this.defineProps() as P;
-    const wrapper = render(
+    const wrapper = render<P, React.ComponentState>(
       this.WrappingComponent ? (
         <this.WrappingComponent>
           <this.Component {...props} />
@@ -86,7 +86,7 @@ export default class Wrapper<
     this.beforeMount();
 
     const props = this.defineProps() as P;
-    const wrapper = shallow(
+    const wrapper = shallow<P>(
       this.WrappingComponent ? (
         <this.WrappingComponent>
           <this.Component {...props} />