From c2c49c5f7a0eea2fac532218acbb0eb31cb82e07 Mon Sep 17 00:00:00 2001
From: Jerko Steiner <jerko.steiner@gmail.com>
Date: Wed, 11 Sep 2019 10:14:56 +0700
Subject: [PATCH] Record actions before rendering

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

diff --git a/packages/client/src/test-utils/TestUtils.tsx b/packages/client/src/test-utils/TestUtils.tsx
index 116baef..85b9151 100644
--- a/packages/client/src/test-utils/TestUtils.tsx
+++ b/packages/client/src/test-utils/TestUtils.tsx
@@ -60,7 +60,6 @@ export class TestUtils {
     const {reducers, select} = params
 
     const waitMiddleware = new WaitMiddleware()
-    const recorder = waitMiddleware.record()
 
     let store = this.createStore({
       reducer: this.combineReducers(reducers),
@@ -89,9 +88,12 @@ export class TestUtils {
       let createJSX: CreateJSX | undefined
 
       const render = (props: Props) => {
+        const recorder = waitMiddleware.record()
+
         const jsx = createJSX
           ? createJSX(Component, props)
           : <Component {...props} />
+
         const result = this.render(
           <Provider store={store}>
             {jsx}