github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/worker/runtime/process_killer_test.go (about)

     1  package runtime_test
     2  
     3  import (
     4  	"context"
     5  	"errors"
     6  	"math"
     7  	"syscall"
     8  	"time"
     9  
    10  	"github.com/pf-qiu/concourse/v6/worker/runtime"
    11  	"github.com/pf-qiu/concourse/v6/worker/runtime/libcontainerd/libcontainerdfakes"
    12  	"github.com/containerd/containerd"
    13  	"github.com/stretchr/testify/require"
    14  	"github.com/stretchr/testify/suite"
    15  )
    16  
    17  type ProcessKillerSuite struct {
    18  	suite.Suite
    19  	*require.Assertions
    20  
    21  	signal syscall.Signal
    22  	proc   *libcontainerdfakes.FakeProcess
    23  	killer runtime.ProcessKiller
    24  
    25  	goodEnoughTimeout time.Duration
    26  	notEnoughTimeout  time.Duration
    27  }
    28  
    29  func (s *ProcessKillerSuite) SetupTest() {
    30  	s.proc = new(libcontainerdfakes.FakeProcess)
    31  	s.killer = runtime.NewProcessKiller()
    32  
    33  	s.signal = 142
    34  	s.goodEnoughTimeout = math.MaxInt64
    35  	s.notEnoughTimeout = time.Nanosecond
    36  }
    37  
    38  func (s *ProcessKillerSuite) TestKillCallsWaitWithDeadline() {
    39  	ch := make(chan containerd.ExitStatus, 1)
    40  	ch <- *containerd.NewExitStatus(0, time.Now(), nil)
    41  	s.proc.WaitReturns(ch, nil)
    42  
    43  	err := s.killer.Kill(context.Background(), s.proc, s.signal, s.goodEnoughTimeout)
    44  	s.NoError(err)
    45  
    46  	s.Equal(1, s.proc.WaitCallCount())
    47  	waitCtx := s.proc.WaitArgsForCall(0)
    48  	_, deadlineSet := waitCtx.Deadline()
    49  	s.True(deadlineSet)
    50  }
    51  
    52  func (s *ProcessKillerSuite) TestKillWaitError() {
    53  	expectedErr := errors.New("wait-err")
    54  	s.proc.WaitReturns(nil, expectedErr)
    55  
    56  	err := s.killer.Kill(context.Background(), s.proc, s.signal, s.goodEnoughTimeout)
    57  	s.True(errors.Is(err, expectedErr))
    58  }
    59  
    60  func (s *ProcessKillerSuite) TestKillKillError() {
    61  	expectedErr := errors.New("kill-err")
    62  	s.proc.KillReturns(expectedErr)
    63  
    64  	err := s.killer.Kill(context.Background(), s.proc, s.signal, s.goodEnoughTimeout)
    65  	s.True(errors.Is(err, expectedErr))
    66  }
    67  
    68  func (s *ProcessKillerSuite) TestKillWaitContextDeadlineReached() {
    69  	err := s.killer.Kill(context.Background(), s.proc, s.signal, s.notEnoughTimeout)
    70  	s.True(errors.Is(err, runtime.ErrGracePeriodTimeout))
    71  }
    72  
    73  func (s *ProcessKillerSuite) TestKillWaitContextCancelled() {
    74  	ctx, cancel := context.WithCancel(context.Background())
    75  	cancel()
    76  
    77  	err := s.killer.Kill(ctx, s.proc, s.signal, s.goodEnoughTimeout)
    78  	s.True(errors.Is(err, context.Canceled))
    79  }
    80  
    81  func (s *ProcessKillerSuite) TestKillExitStatusError() {
    82  	ch := make(chan containerd.ExitStatus, 1)
    83  
    84  	expectedErr := errors.New("status-err")
    85  	ch <- *containerd.NewExitStatus(0, time.Now(), expectedErr)
    86  	s.proc.WaitReturns(ch, nil)
    87  
    88  	err := s.killer.Kill(context.Background(), s.proc, s.signal, s.goodEnoughTimeout)
    89  	s.True(errors.Is(err, expectedErr))
    90  }