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

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package runtimefakes
     3  
     4  import (
     5  	"context"
     6  	"sync"
     7  
     8  	"github.com/pf-qiu/concourse/v6/worker/runtime"
     9  	"github.com/containerd/containerd"
    10  )
    11  
    12  type FakeKiller struct {
    13  	KillStub        func(context.Context, containerd.Task, runtime.KillBehaviour) error
    14  	killMutex       sync.RWMutex
    15  	killArgsForCall []struct {
    16  		arg1 context.Context
    17  		arg2 containerd.Task
    18  		arg3 runtime.KillBehaviour
    19  	}
    20  	killReturns struct {
    21  		result1 error
    22  	}
    23  	killReturnsOnCall map[int]struct {
    24  		result1 error
    25  	}
    26  	invocations      map[string][][]interface{}
    27  	invocationsMutex sync.RWMutex
    28  }
    29  
    30  func (fake *FakeKiller) Kill(arg1 context.Context, arg2 containerd.Task, arg3 runtime.KillBehaviour) error {
    31  	fake.killMutex.Lock()
    32  	ret, specificReturn := fake.killReturnsOnCall[len(fake.killArgsForCall)]
    33  	fake.killArgsForCall = append(fake.killArgsForCall, struct {
    34  		arg1 context.Context
    35  		arg2 containerd.Task
    36  		arg3 runtime.KillBehaviour
    37  	}{arg1, arg2, arg3})
    38  	fake.recordInvocation("Kill", []interface{}{arg1, arg2, arg3})
    39  	fake.killMutex.Unlock()
    40  	if fake.KillStub != nil {
    41  		return fake.KillStub(arg1, arg2, arg3)
    42  	}
    43  	if specificReturn {
    44  		return ret.result1
    45  	}
    46  	fakeReturns := fake.killReturns
    47  	return fakeReturns.result1
    48  }
    49  
    50  func (fake *FakeKiller) KillCallCount() int {
    51  	fake.killMutex.RLock()
    52  	defer fake.killMutex.RUnlock()
    53  	return len(fake.killArgsForCall)
    54  }
    55  
    56  func (fake *FakeKiller) KillCalls(stub func(context.Context, containerd.Task, runtime.KillBehaviour) error) {
    57  	fake.killMutex.Lock()
    58  	defer fake.killMutex.Unlock()
    59  	fake.KillStub = stub
    60  }
    61  
    62  func (fake *FakeKiller) KillArgsForCall(i int) (context.Context, containerd.Task, runtime.KillBehaviour) {
    63  	fake.killMutex.RLock()
    64  	defer fake.killMutex.RUnlock()
    65  	argsForCall := fake.killArgsForCall[i]
    66  	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
    67  }
    68  
    69  func (fake *FakeKiller) KillReturns(result1 error) {
    70  	fake.killMutex.Lock()
    71  	defer fake.killMutex.Unlock()
    72  	fake.KillStub = nil
    73  	fake.killReturns = struct {
    74  		result1 error
    75  	}{result1}
    76  }
    77  
    78  func (fake *FakeKiller) KillReturnsOnCall(i int, result1 error) {
    79  	fake.killMutex.Lock()
    80  	defer fake.killMutex.Unlock()
    81  	fake.KillStub = nil
    82  	if fake.killReturnsOnCall == nil {
    83  		fake.killReturnsOnCall = make(map[int]struct {
    84  			result1 error
    85  		})
    86  	}
    87  	fake.killReturnsOnCall[i] = struct {
    88  		result1 error
    89  	}{result1}
    90  }
    91  
    92  func (fake *FakeKiller) Invocations() map[string][][]interface{} {
    93  	fake.invocationsMutex.RLock()
    94  	defer fake.invocationsMutex.RUnlock()
    95  	fake.killMutex.RLock()
    96  	defer fake.killMutex.RUnlock()
    97  	copiedInvocations := map[string][][]interface{}{}
    98  	for key, value := range fake.invocations {
    99  		copiedInvocations[key] = value
   100  	}
   101  	return copiedInvocations
   102  }
   103  
   104  func (fake *FakeKiller) recordInvocation(key string, args []interface{}) {
   105  	fake.invocationsMutex.Lock()
   106  	defer fake.invocationsMutex.Unlock()
   107  	if fake.invocations == nil {
   108  		fake.invocations = map[string][][]interface{}{}
   109  	}
   110  	if fake.invocations[key] == nil {
   111  		fake.invocations[key] = [][]interface{}{}
   112  	}
   113  	fake.invocations[key] = append(fake.invocations[key], args)
   114  }
   115  
   116  var _ runtime.Killer = new(FakeKiller)