Merge pull request #20 from msmith-techempower/test-async-await

Adds compound async action test
TechEmpower · May 14, 2019 · a42ecdb · a42ecdb
2 parents a7fd424 + f46f70b
commit a42ecdb
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/src/examples/counter/Counter.js b/src/examples/counter/Counter.js
@@ -21,6 +21,15 @@ export default function Counter() {
       />
       <button className="removeState" onClick={() => actions.removeState()} />
       <button className="asyncFunc" onClick={actions.asyncFunc} />
+      <button
+        className="compoundAsyncFunc"
+        onClick={async () => {
+          // Should set `state.count` = 256
+          await actions.asyncFunc();
+          // Should set `state.count` = 257
+          actions.increment();
+        }}
+      />
       <input className="googleStatus" value={state.status} />
       <button className="fetchGoogle" onClick={actions.fetchGoogle} />
     </div>

diff --git a/src/tests/counter/Counter.spec.js b/src/tests/counter/Counter.spec.js
@@ -158,6 +158,20 @@ it("can asyncFunc", async () => {
   });
 });
 
+it("can do compound async actions", async () => {
+  const counter = TestRenderer.create(<Counter />);
+  const val = counter.root.findByProps({ className: "val" });
+  const button = counter.root.findByProps({ className: "compoundAsyncFunc" });
+
+  await act(async () => {
+    expect(val.props.value).toBe(0);
+
+    await button.props.onClick();
+
+    expect(val.props.value).toBe(257);
+  });
+});
+
 it("can fetchGoogle", async () => {
   const counter = TestRenderer.create(<Counter />);
   const val = counter.root.findByProps({ className: "googleStatus" });