posixsignal_test.go

// Copyright 2024 The seacraft Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http:www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package shutdown

import (
	"syscall"
	"testing"
	"time"

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

type startShutdownFunc func(sm ShutdownManager)

func (f startShutdownFunc) StartShutdown(sm ShutdownManager) {
	f(sm)
}

func (f startShutdownFunc) ReportError(error) {
}

func (f startShutdownFunc) AddShutdownCallback(ShutdownCallback) {
}

func waitSig(t *testing.T, c <-chan int) {
	select {
	case <-c:

	case <-time.After(1 * time.Second):
		t.Error("Timeout waiting for StartShutdown.")
	}
}

func TestStartShutdownCalledOnDefaultSignals(t *testing.T) {
	c := make(chan int, 100)

	psm := NewPosixSignalManager()
	err := psm.Start(startShutdownFunc(func(sm ShutdownManager) {
		c <- 1
	}))
	if err != nil {
		return
	}

	time.Sleep(time.Millisecond)

	err = syscall.Kill(syscall.Getpid(), syscall.SIGINT)
	assert.NoError(t, err)
	waitSig(t, c)

	err = psm.Start(startShutdownFunc(func(sm ShutdownManager) {
		c <- 1
	}))
	assert.NoError(t, err)

	time.Sleep(time.Millisecond)

	err = syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
	assert.NoError(t, err)

	waitSig(t, c)
}

func TestStartShutdownCalledCustomSignal(t *testing.T) {
	c := make(chan int, 100)

	psm := NewPosixSignalManager(syscall.SIGHUP)
	err := psm.Start(startShutdownFunc(func(sm ShutdownManager) {
		c <- 1
	}))
	assert.NoError(t, err)
	time.Sleep(time.Millisecond)

	err = syscall.Kill(syscall.Getpid(), syscall.SIGHUP)
	assert.NoError(t, err)
	waitSig(t, c)
}