mirror of
https://gitea.com/gitea/act_runner.git
synced 2026-03-03 02:10:18 +00:00
100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"syscall"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGracefulJobCancellationViaSigint(t *testing.T) {
|
|
ctx, cancel, channel := createGracefulJobCancellationContext()
|
|
defer cancel()
|
|
assert.NotNil(t, ctx)
|
|
assert.NotNil(t, cancel)
|
|
assert.NotNil(t, channel)
|
|
cancelCtx := JobCancelContext(ctx)
|
|
assert.NotNil(t, cancelCtx)
|
|
require.NoError(t, ctx.Err())
|
|
require.NoError(t, cancelCtx.Err())
|
|
channel <- os.Interrupt
|
|
select {
|
|
case <-time.After(1 * time.Second):
|
|
t.Fatal("context not canceled")
|
|
case <-cancelCtx.Done():
|
|
case <-ctx.Done():
|
|
}
|
|
if assert.Error(t, cancelCtx.Err(), "context canceled") {
|
|
assert.Equal(t, context.Canceled, cancelCtx.Err())
|
|
}
|
|
require.NoError(t, ctx.Err())
|
|
channel <- os.Interrupt
|
|
select {
|
|
case <-time.After(1 * time.Second):
|
|
t.Fatal("context not canceled")
|
|
case <-ctx.Done():
|
|
}
|
|
if assert.Error(t, ctx.Err(), "context canceled") {
|
|
assert.Equal(t, context.Canceled, ctx.Err())
|
|
}
|
|
}
|
|
|
|
func TestForceCancellationViaSigterm(t *testing.T) {
|
|
ctx, cancel, channel := createGracefulJobCancellationContext()
|
|
defer cancel()
|
|
assert.NotNil(t, ctx)
|
|
assert.NotNil(t, cancel)
|
|
assert.NotNil(t, channel)
|
|
cancelCtx := JobCancelContext(ctx)
|
|
assert.NotNil(t, cancelCtx)
|
|
require.NoError(t, ctx.Err())
|
|
require.NoError(t, cancelCtx.Err())
|
|
channel <- syscall.SIGTERM
|
|
select {
|
|
case <-time.After(1 * time.Second):
|
|
t.Fatal("context not canceled")
|
|
case <-cancelCtx.Done():
|
|
}
|
|
select {
|
|
case <-time.After(1 * time.Second):
|
|
t.Fatal("context not canceled")
|
|
case <-ctx.Done():
|
|
}
|
|
if assert.Error(t, ctx.Err(), "context canceled") {
|
|
assert.Equal(t, context.Canceled, ctx.Err())
|
|
}
|
|
if assert.Error(t, cancelCtx.Err(), "context canceled") {
|
|
assert.Equal(t, context.Canceled, cancelCtx.Err())
|
|
}
|
|
}
|
|
|
|
func TestCreateGracefulJobCancellationContext(t *testing.T) {
|
|
ctx, cancel := CreateGracefulJobCancellationContext()
|
|
defer cancel()
|
|
assert.NotNil(t, ctx)
|
|
assert.NotNil(t, cancel)
|
|
cancelCtx := JobCancelContext(ctx)
|
|
assert.NotNil(t, cancelCtx)
|
|
require.NoError(t, cancelCtx.Err())
|
|
}
|
|
|
|
func TestCreateGracefulJobCancellationContextCancelFunc(t *testing.T) {
|
|
ctx, cancel := CreateGracefulJobCancellationContext()
|
|
assert.NotNil(t, ctx)
|
|
assert.NotNil(t, cancel)
|
|
cancelCtx := JobCancelContext(ctx)
|
|
assert.NotNil(t, cancelCtx)
|
|
require.NoError(t, cancelCtx.Err())
|
|
cancel()
|
|
if assert.Error(t, ctx.Err(), "context canceled") {
|
|
assert.Equal(t, context.Canceled, ctx.Err())
|
|
}
|
|
if assert.Error(t, cancelCtx.Err(), "context canceled") {
|
|
assert.Equal(t, context.Canceled, cancelCtx.Err())
|
|
}
|
|
}
|