From 45f0cb76d30673bb7cb782e916f1bc8dbca482d3 Mon Sep 17 00:00:00 2001
From: Jerko Steiner <jerko.steiner@gmail.com>
Date: Tue, 4 Jun 2019 14:58:41 +1000
Subject: [PATCH] Fix return type of withJSX (used to be any)

---
 packages/client/src/test-utils/TestUtils.tsx | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/packages/client/src/test-utils/TestUtils.tsx b/packages/client/src/test-utils/TestUtils.tsx
index b370dca..8f18c8e 100644
--- a/packages/client/src/test-utils/TestUtils.tsx
+++ b/packages/client/src/test-utils/TestUtils.tsx
@@ -94,7 +94,9 @@ export class TestUtils {
         return self
       }
 
-      const self = {
+      const self: ISelf<
+        Props, typeof store, typeof Component, CreateJSX
+        > = {
         render,
         store,
         Component,
@@ -107,3 +109,11 @@ export class TestUtils {
     return {withState, withComponent}
   }
 }
+
+interface ISelf<Props, Store, Component, CreateJSX> {
+  render: (props: Props) => ReturnType<TestUtils['render']>
+  store: Store
+  Component: Component
+  withJSX: (localCreateJSX: CreateJSX)
+  => ISelf<Props, Store, Component, CreateJSX>
+}