machine/transition_test.go

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

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

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")
  }
}