state_machine_test.go

package hsm

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestNewStateMachine(t *testing.T) {
	startState := NewMockState(nil)
	stateMachine := getStateMachine(t, startState)
	assert.Equal(t, startState, stateMachine.currentState)
}

type handleEventTest struct {
	eventID          string
	result           bool
	currentStateName string
}

var handleEventTests = []handleEventTest{
	{
		eventID:          mockStartEventID,
		result:           true,
		currentStateName: mockStateName,
	},
	{
		eventID:          mockSkipEventID,
		result:           false,
		currentStateName: mockStateName,
	},
	{
		eventID:          mockEndEventID,
		result:           true,
		currentStateName: "",
	},
}

func TestHandleEvent(t *testing.T) {
	for _, tt := range handleEventTests {
		startState := NewMockState(nil)
		stateMachine := getStateMachine(t, startState)

		event := NewMockEvent(tt.eventID)
		result := stateMachine.HandleEvent(event)

		assert.Equal(t, result, tt.result)
		if tt.currentStateName != "" {
			assert.NotNil(t, stateMachine.currentState)
			assert.Equal(t, stateMachine.currentState.Name(), tt.currentStateName)
		} else {
			assert.Nil(t, stateMachine.currentState)
		}
	}
}