signal_test.go

package goka

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestSignal_SetState(t *testing.T) {
	sig := NewSignal(0, 1, 2)
	require.True(t, sig.IsState(0))
	require.False(t, sig.IsState(1))

	sig.SetState(1)
	require.True(t, sig.IsState(1))
	require.False(t, sig.IsState(0))

	defer func() {
		err := recover()
		if err == nil {
			t.Fatalf("Expected panic, which didn't occur")
		}
	}()

	// set some invalid state, this will panic
	sig.SetState(3)
}

func TestSignal_Wait(t *testing.T) {
	sig := NewSignal(0, 1, 2)

	<-sig.WaitForState(0)
	// should continue right now since

	var (
		done     = make(chan struct{})
		hasState bool
	)
	go func() {
		defer close(done)
		<-sig.WaitForState(1)
		hasState = true
	}()

	require.False(t, hasState)
	sig.SetState(1)
	// wait for the goroutine to catchup with the state
	<-done
	require.True(t, hasState)
}

func TestSignalWaitMin(t *testing.T) {
	sig := NewSignal(0, 1, 2)

	var (
		done     = make(chan struct{})
		hasState bool
	)
	go func() {
		defer close(done)
		<-sig.WaitForStateMin(1)
		hasState = true
	}()

	require.False(t, hasState)
	sig.SetState(2)
	// wait for the goroutine to catchup with the state
	<-done
	require.True(t, hasState)
}