github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/worker/workerfakes/fake_fetcher.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package workerfakes
     3  
     4  import (
     5  	"context"
     6  	"sync"
     7  
     8  	"code.cloudfoundry.org/lager"
     9  	"github.com/pf-qiu/concourse/v6/atc/db"
    10  	"github.com/pf-qiu/concourse/v6/atc/resource"
    11  	"github.com/pf-qiu/concourse/v6/atc/runtime"
    12  	"github.com/pf-qiu/concourse/v6/atc/worker"
    13  )
    14  
    15  type FakeFetcher struct {
    16  	FetchStub        func(context.Context, lager.Logger, db.ContainerMetadata, worker.Worker, worker.ContainerSpec, runtime.ProcessSpec, resource.Resource, db.ContainerOwner, db.UsedResourceCache, string) (worker.GetResult, worker.Volume, error)
    17  	fetchMutex       sync.RWMutex
    18  	fetchArgsForCall []struct {
    19  		arg1  context.Context
    20  		arg2  lager.Logger
    21  		arg3  db.ContainerMetadata
    22  		arg4  worker.Worker
    23  		arg5  worker.ContainerSpec
    24  		arg6  runtime.ProcessSpec
    25  		arg7  resource.Resource
    26  		arg8  db.ContainerOwner
    27  		arg9  db.UsedResourceCache
    28  		arg10 string
    29  	}
    30  	fetchReturns struct {
    31  		result1 worker.GetResult
    32  		result2 worker.Volume
    33  		result3 error
    34  	}
    35  	fetchReturnsOnCall map[int]struct {
    36  		result1 worker.GetResult
    37  		result2 worker.Volume
    38  		result3 error
    39  	}
    40  	invocations      map[string][][]interface{}
    41  	invocationsMutex sync.RWMutex
    42  }
    43  
    44  func (fake *FakeFetcher) Fetch(arg1 context.Context, arg2 lager.Logger, arg3 db.ContainerMetadata, arg4 worker.Worker, arg5 worker.ContainerSpec, arg6 runtime.ProcessSpec, arg7 resource.Resource, arg8 db.ContainerOwner, arg9 db.UsedResourceCache, arg10 string) (worker.GetResult, worker.Volume, error) {
    45  	fake.fetchMutex.Lock()
    46  	ret, specificReturn := fake.fetchReturnsOnCall[len(fake.fetchArgsForCall)]
    47  	fake.fetchArgsForCall = append(fake.fetchArgsForCall, struct {
    48  		arg1  context.Context
    49  		arg2  lager.Logger
    50  		arg3  db.ContainerMetadata
    51  		arg4  worker.Worker
    52  		arg5  worker.ContainerSpec
    53  		arg6  runtime.ProcessSpec
    54  		arg7  resource.Resource
    55  		arg8  db.ContainerOwner
    56  		arg9  db.UsedResourceCache
    57  		arg10 string
    58  	}{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10})
    59  	fake.recordInvocation("Fetch", []interface{}{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10})
    60  	fake.fetchMutex.Unlock()
    61  	if fake.FetchStub != nil {
    62  		return fake.FetchStub(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10)
    63  	}
    64  	if specificReturn {
    65  		return ret.result1, ret.result2, ret.result3
    66  	}
    67  	fakeReturns := fake.fetchReturns
    68  	return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3
    69  }
    70  
    71  func (fake *FakeFetcher) FetchCallCount() int {
    72  	fake.fetchMutex.RLock()
    73  	defer fake.fetchMutex.RUnlock()
    74  	return len(fake.fetchArgsForCall)
    75  }
    76  
    77  func (fake *FakeFetcher) FetchCalls(stub func(context.Context, lager.Logger, db.ContainerMetadata, worker.Worker, worker.ContainerSpec, runtime.ProcessSpec, resource.Resource, db.ContainerOwner, db.UsedResourceCache, string) (worker.GetResult, worker.Volume, error)) {
    78  	fake.fetchMutex.Lock()
    79  	defer fake.fetchMutex.Unlock()
    80  	fake.FetchStub = stub
    81  }
    82  
    83  func (fake *FakeFetcher) FetchArgsForCall(i int) (context.Context, lager.Logger, db.ContainerMetadata, worker.Worker, worker.ContainerSpec, runtime.ProcessSpec, resource.Resource, db.ContainerOwner, db.UsedResourceCache, string) {
    84  	fake.fetchMutex.RLock()
    85  	defer fake.fetchMutex.RUnlock()
    86  	argsForCall := fake.fetchArgsForCall[i]
    87  	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4, argsForCall.arg5, argsForCall.arg6, argsForCall.arg7, argsForCall.arg8, argsForCall.arg9, argsForCall.arg10
    88  }
    89  
    90  func (fake *FakeFetcher) FetchReturns(result1 worker.GetResult, result2 worker.Volume, result3 error) {
    91  	fake.fetchMutex.Lock()
    92  	defer fake.fetchMutex.Unlock()
    93  	fake.FetchStub = nil
    94  	fake.fetchReturns = struct {
    95  		result1 worker.GetResult
    96  		result2 worker.Volume
    97  		result3 error
    98  	}{result1, result2, result3}
    99  }
   100  
   101  func (fake *FakeFetcher) FetchReturnsOnCall(i int, result1 worker.GetResult, result2 worker.Volume, result3 error) {
   102  	fake.fetchMutex.Lock()
   103  	defer fake.fetchMutex.Unlock()
   104  	fake.FetchStub = nil
   105  	if fake.fetchReturnsOnCall == nil {
   106  		fake.fetchReturnsOnCall = make(map[int]struct {
   107  			result1 worker.GetResult
   108  			result2 worker.Volume
   109  			result3 error
   110  		})
   111  	}
   112  	fake.fetchReturnsOnCall[i] = struct {
   113  		result1 worker.GetResult
   114  		result2 worker.Volume
   115  		result3 error
   116  	}{result1, result2, result3}
   117  }
   118  
   119  func (fake *FakeFetcher) Invocations() map[string][][]interface{} {
   120  	fake.invocationsMutex.RLock()
   121  	defer fake.invocationsMutex.RUnlock()
   122  	fake.fetchMutex.RLock()
   123  	defer fake.fetchMutex.RUnlock()
   124  	copiedInvocations := map[string][][]interface{}{}
   125  	for key, value := range fake.invocations {
   126  		copiedInvocations[key] = value
   127  	}
   128  	return copiedInvocations
   129  }
   130  
   131  func (fake *FakeFetcher) recordInvocation(key string, args []interface{}) {
   132  	fake.invocationsMutex.Lock()
   133  	defer fake.invocationsMutex.Unlock()
   134  	if fake.invocations == nil {
   135  		fake.invocations = map[string][][]interface{}{}
   136  	}
   137  	if fake.invocations[key] == nil {
   138  		fake.invocations[key] = [][]interface{}{}
   139  	}
   140  	fake.invocations[key] = append(fake.invocations[key], args)
   141  }
   142  
   143  var _ worker.Fetcher = new(FakeFetcher)