Merge pull request #19 from okhowang/master

fix initial transition with multiple sub states

statemachine.go

@@ -437,12 +437,12 @@ func (sm *StateMachine) enterState(ctx context.Context, sr *stateRepresentation,
 	}
 	// Recursively enter substates that have an initial transition
 	if sr.HasInitialState {
-		isValidForInitialState := len(sr.Substates) != 0
+		isValidForInitialState := false
 		for _, substate := range sr.Substates {
 			// Verify that the target state is a substate
 			// Check if state has substate(s), and if an initial transition(s) has been set up.
-			if substate.State != sr.InitialTransitionTarget {
-				isValidForInitialState = false
+			if substate.State == sr.InitialTransitionTarget {
+				isValidForInitialState = true
 				break
 			}
 		}

statemachine_test.go

@@ -1261,6 +1261,15 @@ func TestStateMachine_InitialTransition_DoNotAllowTransitionToSelf(t *testing.T)
 	})
 }
 
+func TestStateMachine_InitialTransition_WithMultipleSubStates(t *testing.T) {
+	sm := NewStateMachine(stateA)
+	sm.Configure(stateA).Permit(triggerX, stateB)
+	sm.Configure(stateB).InitialTransition(stateC)
+	sm.Configure(stateC).SubstateOf(stateB)
+	sm.Configure(stateD).SubstateOf(stateB)
+	assert.NoError(t, sm.Fire(triggerX))
+}
+
 func TestStateMachine_InitialTransition_DoNotAllowTransitionToAnotherSuperstate(t *testing.T) {
 	sm := NewStateMachine(stateA)