Proxy: Add test seams and unit tests for internal/signal.
Introduce signalNotify and osExit indirections so InstallSignals and InstallForceQuit can be exercised without real OS signal delivery or process termination. Coverage is 100% in isolation. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
parent
43576e6036
commit
98d09138b8
|
|
@ -15,9 +15,15 @@ import (
|
|||
"srsx/internal/logger"
|
||||
)
|
||||
|
||||
// Indirections so tests can substitute signal delivery and process exit.
|
||||
var (
|
||||
signalNotify = signal.Notify
|
||||
osExit = os.Exit
|
||||
)
|
||||
|
||||
func InstallSignals(ctx context.Context, cancel context.CancelFunc) {
|
||||
sc := make(chan os.Signal, 1)
|
||||
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
|
||||
signalNotify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
|
||||
|
||||
go func() {
|
||||
for s := range sc {
|
||||
|
|
@ -40,7 +46,7 @@ func InstallForceQuit(ctx context.Context, environment env.Environment) error {
|
|||
<-ctx.Done()
|
||||
time.Sleep(forceTimeout)
|
||||
logger.Wf(ctx, "Force to exit by timeout")
|
||||
os.Exit(1)
|
||||
osExit(1)
|
||||
}()
|
||||
return nil
|
||||
}
|
||||
|
|
|
|||
170
internal/signal/signal_test.go
Normal file
170
internal/signal/signal_test.go
Normal file
|
|
@ -0,0 +1,170 @@
|
|||
// Copyright (c) 2026 Winlin
|
||||
//
|
||||
// SPDX-License-Identifier: MIT
|
||||
package signal
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"syscall"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"srsx/internal/env/envfakes"
|
||||
)
|
||||
|
||||
// swapNotify replaces signalNotify with a capturing fake and returns a getter
|
||||
// for the channel registered by the code under test plus a restore func.
|
||||
func swapNotify(t *testing.T) (func() chan<- os.Signal, func()) {
|
||||
t.Helper()
|
||||
orig := signalNotify
|
||||
var (
|
||||
mu sync.Mutex
|
||||
ch chan<- os.Signal
|
||||
)
|
||||
signalNotify = func(c chan<- os.Signal, _ ...os.Signal) {
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
ch = c
|
||||
}
|
||||
return func() chan<- os.Signal {
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
return ch
|
||||
}, func() {
|
||||
signalNotify = orig
|
||||
}
|
||||
}
|
||||
|
||||
func swapExit(t *testing.T) (*int32, chan int, func()) {
|
||||
t.Helper()
|
||||
orig := osExit
|
||||
var called int32
|
||||
done := make(chan int, 1)
|
||||
osExit = func(code int) {
|
||||
atomic.StoreInt32(&called, 1)
|
||||
select {
|
||||
case done <- code:
|
||||
default:
|
||||
}
|
||||
// Block to mimic os.Exit never returning; the goroutine holding us
|
||||
// here is abandoned when the test ends.
|
||||
select {}
|
||||
}
|
||||
return &called, done, func() { osExit = orig }
|
||||
}
|
||||
|
||||
func TestInstallSignals_CancelsOnSignal(t *testing.T) {
|
||||
getCh, restore := swapNotify(t)
|
||||
defer restore()
|
||||
|
||||
ctx, cancel := context.WithCancel(t.Context())
|
||||
defer cancel()
|
||||
|
||||
InstallSignals(ctx, cancel)
|
||||
|
||||
ch := getCh()
|
||||
if ch == nil {
|
||||
t.Fatal("signalNotify was not called")
|
||||
}
|
||||
ch <- syscall.SIGINT
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
case <-time.After(time.Second):
|
||||
t.Fatal("ctx was not canceled after signal")
|
||||
}
|
||||
}
|
||||
|
||||
func TestInstallSignals_HandlesRepeatedSignals(t *testing.T) {
|
||||
getCh, restore := swapNotify(t)
|
||||
defer restore()
|
||||
|
||||
ctx, cancel := context.WithCancel(t.Context())
|
||||
defer cancel()
|
||||
|
||||
InstallSignals(ctx, cancel)
|
||||
ch := getCh()
|
||||
|
||||
// Multiple signals must not panic; cancel() is idempotent.
|
||||
ch <- syscall.SIGINT
|
||||
ch <- syscall.SIGTERM
|
||||
ch <- os.Interrupt
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
case <-time.After(time.Second):
|
||||
t.Fatal("ctx was not canceled")
|
||||
}
|
||||
}
|
||||
|
||||
func TestInstallForceQuit_InvalidDurationReturnsError(t *testing.T) {
|
||||
fakeEnv := &envfakes.FakeEnvironment{}
|
||||
fakeEnv.ForceQuitTimeoutReturns("not-a-duration")
|
||||
|
||||
err := InstallForceQuit(t.Context(), fakeEnv)
|
||||
if err == nil {
|
||||
t.Fatal("want error for bad duration")
|
||||
}
|
||||
if !strings.Contains(err.Error(), "parse force timeout") {
|
||||
t.Fatalf("err = %v", err)
|
||||
}
|
||||
if !strings.Contains(err.Error(), "not-a-duration") {
|
||||
t.Fatalf("err missing input: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestInstallForceQuit_ExitsAfterTimeout(t *testing.T) {
|
||||
called, done, restore := swapExit(t)
|
||||
defer restore()
|
||||
|
||||
fakeEnv := &envfakes.FakeEnvironment{}
|
||||
fakeEnv.ForceQuitTimeoutReturns("1ms")
|
||||
|
||||
ctx, cancel := context.WithCancel(t.Context())
|
||||
if err := InstallForceQuit(ctx, fakeEnv); err != nil {
|
||||
t.Fatalf("unexpected err: %v", err)
|
||||
}
|
||||
|
||||
// Before cancel, the goroutine is blocked and exit must not fire.
|
||||
if atomic.LoadInt32(called) != 0 {
|
||||
t.Fatal("osExit called before ctx cancel")
|
||||
}
|
||||
cancel()
|
||||
|
||||
select {
|
||||
case code := <-done:
|
||||
if code != 1 {
|
||||
t.Fatalf("exit code = %d, want 1", code)
|
||||
}
|
||||
case <-time.After(time.Second):
|
||||
t.Fatal("osExit not called after cancel + timeout")
|
||||
}
|
||||
}
|
||||
|
||||
func TestInstallForceQuit_WaitsForCancelBeforeSleeping(t *testing.T) {
|
||||
called, done, restore := swapExit(t)
|
||||
defer restore()
|
||||
|
||||
fakeEnv := &envfakes.FakeEnvironment{}
|
||||
fakeEnv.ForceQuitTimeoutReturns("10ms")
|
||||
|
||||
// Intentionally use a never-canceled context and leak the goroutine:
|
||||
// if we canceled at test end, the goroutine would wake and race with
|
||||
// restore() writing osExit.
|
||||
if err := InstallForceQuit(context.Background(), fakeEnv); err != nil {
|
||||
t.Fatalf("unexpected err: %v", err)
|
||||
}
|
||||
|
||||
select {
|
||||
case <-done:
|
||||
t.Fatal("osExit fired without ctx cancel")
|
||||
case <-time.After(30 * time.Millisecond):
|
||||
}
|
||||
if atomic.LoadInt32(called) != 0 {
|
||||
t.Fatal("osExit called unexpectedly")
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user