machine/transition_test.go

// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.

package machine

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

func setupTransition() Transitioner {
	t := NewTransition("open", "closed", "open")
	if t == nil {
		log.Fatal("Failed creating new transition")
	}
	return t
}

func setupSubscriber() Subscriber {
	c := make(EventChannel, 2)
	return &c
}

func TestNewTransition(t *testing.T) {
	s := NewTransition("connect", "disconnected", "connected")
	if s == nil {
		t.Errorf("Failed creating new transition")
	}
}

func TestTransitionExecution(t *testing.T) {
	s := setupTransition()
	m := setupModel("closed")
	assert.Nil(t, s.Run(m))
	state := m.InspectState()
	if state != "open" {
		t.Errorf("Failed to transition state: %s", state)
	}
}

func TestTransitionSubscribe(t *testing.T) {
	c := setupSubscriber()
	s := setupTransition()
	s.Subscribe(c)
	m := setupModel("closed")
	assert.Nil(t, s.Run(m))
	exitEvent := <-*c.(*EventChannel)
	enterEvent := <-*c.(*EventChannel)
	if exitEvent.on != EXITSTATEEVENT {
		t.Errorf("Invalid exit event")
	}
	if enterEvent.on != ENTERSTATEEVENT {
		t.Errorf("Invalid enter event")
	}
}