From df18c3b6c603b3c7f2de6fe175a749df48f081b7 Mon Sep 17 00:00:00 2001
From: Michele Bertoli <michele@berto.li>
Date: Sun, 3 Jun 2018 22:06:15 +0200
Subject: [PATCH] support cond in tests

---
 README.md                                     |   4 +-
 src/testStatechart.js                         |  10 +-
 .../__snapshots__/testStatechart.spec.js.snap |  60 +++----
 test/testStatechart.spec.js                   | 146 ++++++++++++------
 4 files changed, 135 insertions(+), 85 deletions(-)

diff --git a/README.md b/README.md
index 8f1aba9..a476ead 100644
--- a/README.md
+++ b/README.md
@@ -248,10 +248,10 @@ The component to define which parts of the tree should be rendered for a given s
 />
 ```
 
-## testStatechart({ statechart[, fixtures] }, Component)
+## testStatechart({ statechart[, fixtures][, extendedState] }, Component)
 
 The method to automagically generate tests given a statechart definition, and a component.
-It accepts an optional `fixtures` configuration to describe which data should be injected into the component for a given transition.
+It accepts an additional `fixtures` option to describe the data to be injected into the component for a given transition, and an `extendedState` option to control the statechart's conditions - both are optional.
 
 > Please note that the component should be a base component not wrapped into `withStateChart` (see [#46](https://github.com/MicheleBertoli/react-automata/issues/46)).
 
diff --git a/src/testStatechart.js b/src/testStatechart.js
index 24f4e6e..ac18d95 100644
--- a/src/testStatechart.js
+++ b/src/testStatechart.js
@@ -7,19 +7,19 @@ import invariant from 'invariant'
 import { getContextValue } from './utils'
 import withStatechart from './withStatechart'
 
-const testStatechart = (config, Component) => {
+const testStatechart = (options, Component) => {
   invariant(
     !Component.isStateMachine,
     `It seems you are testing a component wrapped into \`withStatechart\`, please use a base component instead.
     See https://github.com/MicheleBertoli/react-automata/issues/46`
   )
 
-  const { channel, statechart } = config
+  const { statechart, extendedState, channel } = options
   const machine = Machine(statechart)
-  const paths = getShortestPaths(machine)
+  const paths = getShortestPaths(machine, extendedState)
 
   Object.keys(paths).forEach(key => {
-    const initialData = idx(config, _ => _.fixtures.initialData)
+    const initialData = idx(options, _ => _.fixtures.initialData)
     const StateMachine = withStatechart(statechart, { channel })(Component)
     const renderer = TestRenderer.create(
       <StateMachine initialData={initialData} />
@@ -27,7 +27,7 @@ const testStatechart = (config, Component) => {
     const instance = renderer.getInstance()
 
     paths[key].forEach(({ event, state }) => {
-      const fixtures = idx(config, _ => _.fixtures[state][event])
+      const fixtures = idx(options, _ => _.fixtures[state][event])
 
       instance.handleTransition(event, fixtures)
     })
diff --git a/test/__snapshots__/testStatechart.spec.js.snap b/test/__snapshots__/testStatechart.spec.js.snap
index 69645cf..58c21d1 100644
--- a/test/__snapshots__/testStatechart.spec.js.snap
+++ b/test/__snapshots__/testStatechart.spec.js.snap
@@ -1,30 +1,54 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
-exports[`action: a 1`] = `
+exports[`channels: outer 1`] = `
+<div>
+  outer
+  <div>
+    inner
+    outer
+  </div>
+</div>
+`;
+
+exports[`cond fail: a 1`] = `"A"`;
+
+exports[`cond pass: a 1`] = `"A"`;
+
+exports[`cond pass: b 1`] = `"B"`;
+
+exports[`conditional action: a 1`] = `
 <div>
   a
 </div>
 `;
 
-exports[`action: b.a 1`] = `
+exports[`conditional action: b.a 1`] = `
 <div>
   b.a
 </div>
 `;
 
-exports[`action: b.b 1`] = `
+exports[`conditional action: b.b 1`] = `
 <div>
   b.b
 </div>
 `;
 
-exports[`channels: outer 1`] = `
+exports[`conditional state: a 1`] = `
 <div>
-  outer
-  <div>
-    inner
-    outer
-  </div>
+  a
+</div>
+`;
+
+exports[`conditional state: b.a 1`] = `
+<div>
+  b.a
+</div>
+`;
+
+exports[`conditional state: b.b 1`] = `
+<div>
+  b.b
 </div>
 `;
 
@@ -243,21 +267,3 @@ exports[`parallel: bold.on,underline.on,italics.on,list.numbers 1`] = `
   list.numbers
 </div>
 `;
-
-exports[`state: a 1`] = `
-<div>
-  a
-</div>
-`;
-
-exports[`state: b.a 1`] = `
-<div>
-  b.a
-</div>
-`;
-
-exports[`state: b.b 1`] = `
-<div>
-  b.b
-</div>
-`;
diff --git a/test/testStatechart.spec.js b/test/testStatechart.spec.js
index 5940602..12c3afa 100644
--- a/test/testStatechart.spec.js
+++ b/test/testStatechart.spec.js
@@ -1,67 +1,69 @@
 import React from 'react'
 import { Action, State, testStatechart, withStatechart } from '../src'
 
-const secondMachine = {
-  initial: 'a',
-  states: {
-    a: {
-      on: {
-        SECOND_NEXT: 'b',
+describe('conditional', () => {
+  const secondMachine = {
+    initial: 'a',
+    states: {
+      a: {
+        on: {
+          SECOND_NEXT: 'b',
+        },
+        onEntry: 'enterBA',
       },
-      onEntry: 'enterBA',
-    },
-    b: {
-      on: {
-        SECOND_NEXT: 'a',
+      b: {
+        on: {
+          SECOND_NEXT: 'a',
+        },
+        onEntry: 'enterBB',
       },
-      onEntry: 'enterBB',
     },
-  },
-}
-
-const firstMachine = {
-  initial: 'a',
-  states: {
-    a: {
-      on: {
-        FIRST_NEXT: 'b',
+  }
+
+  const firstMachine = {
+    initial: 'a',
+    states: {
+      a: {
+        on: {
+          FIRST_NEXT: 'b',
+        },
+        onEntry: 'enterA',
       },
-      onEntry: 'enterA',
-    },
-    b: {
-      on: {
-        FIRST_NEXT: 'a',
+      b: {
+        on: {
+          FIRST_NEXT: 'a',
+        },
+        onEntry: 'enterB',
+        ...secondMachine,
       },
-      onEntry: 'enterB',
-      ...secondMachine,
     },
-  },
-}
+  }
 
-test('action', () => {
-  const App = () => (
-    <div>
-      <Action show="enterA" hide="enterB">
-        a
-      </Action>
-      <Action show="enterBA">b.a</Action>
-      <Action show="enterBB">b.b</Action>
-    </div>
-  )
+  test('action', () => {
+    const App = () => (
+      <div>
+        <Action show="enterA" hide="enterB">
+          a
+        </Action>
+        <Action show="enterBA">b.a</Action>
+        <Action show="enterBB">b.b</Action>
+      </div>
+    )
 
-  testStatechart({ statechart: firstMachine }, App)
-})
+    testStatechart({ statechart: firstMachine }, App)
+  })
 
-test('state', () => {
-  const App = () => (
-    <div>
-      <State value="a">a</State>
-      <State value="b.a">b.a</State>
-      <State value="b.b">b.b</State>
-    </div>
-  )
+  test('state', () => {
+    const App = () => (
+      <div>
+        <State value="a">a</State>
+        <State value="b.a">b.a</State>
+        <State value="b.b">b.b</State>
+      </div>
+    )
 
-  testStatechart({ statechart: firstMachine }, App)
+    testStatechart({ statechart: firstMachine }, App)
+  })
 })
 
 test('parallel', () => {
@@ -171,3 +173,45 @@ test('channels', () => {
 
   testStatechart({ statechart: outer, channel: 'outer' }, Outer)
 })
+
+describe('cond', () => {
+  const statechart = {
+    initial: 'a',
+    states: {
+      a: {
+        on: {
+          EVENT: {
+            b: {
+              cond: extState => extState.shouldPass,
+            },
+          },
+        },
+      },
+      b: {},
+    },
+  }
+
+  const Cond = () => (
+    <React.Fragment>
+      <State value="a">A</State>
+      <State value="b">B</State>
+    </React.Fragment>
+  )
+
+  test('pass', () => {
+    const extendedState = {
+      shouldPass: true,
+    }
+    const fixtures = {
+      a: {
+        EVENT: extendedState,
+      },
+    }
+
+    testStatechart({ statechart, fixtures, extendedState }, Cond)
+  })
+
+  test('fail', () => {
+    testStatechart({ statechart }, Cond)
+  })
+})