lotus/node/shutdown_test.go

package node

import (
	"context"
	"sync"
	"testing"
	"time"

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

func TestMonitorShutdown(t *testing.T) {
	signalCh := make(chan struct{})

	// Three shutdown handlers.
	var wg sync.WaitGroup
	wg.Add(3)
	//stm: @NODE_SHUTDOWN_001
	h := ShutdownHandler{
		Component: "handler",
		StopFunc: func(_ context.Context) error {
			wg.Done()
			return nil
		},
	}

	//stm: @NODE_SHUTDOWN_002
	finishCh := MonitorShutdown(signalCh, h, h, h)

	// Nothing here after 10ms.
	time.Sleep(10 * time.Millisecond)
	require.Len(t, finishCh, 0)

	// Now trigger the shutdown.
	close(signalCh)
	wg.Wait()
	<-finishCh
}