github.com/chenbh/concourse/v6@v6.4.2/atc/policy/policyfakes/fake_agent.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package policyfakes
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/chenbh/concourse/v6/atc/policy"
     8  )
     9  
    10  type FakeAgent struct {
    11  	CheckStub        func(policy.PolicyCheckInput) (bool, error)
    12  	checkMutex       sync.RWMutex
    13  	checkArgsForCall []struct {
    14  		arg1 policy.PolicyCheckInput
    15  	}
    16  	checkReturns struct {
    17  		result1 bool
    18  		result2 error
    19  	}
    20  	checkReturnsOnCall map[int]struct {
    21  		result1 bool
    22  		result2 error
    23  	}
    24  	invocations      map[string][][]interface{}
    25  	invocationsMutex sync.RWMutex
    26  }
    27  
    28  func (fake *FakeAgent) Check(arg1 policy.PolicyCheckInput) (bool, error) {
    29  	fake.checkMutex.Lock()
    30  	ret, specificReturn := fake.checkReturnsOnCall[len(fake.checkArgsForCall)]
    31  	fake.checkArgsForCall = append(fake.checkArgsForCall, struct {
    32  		arg1 policy.PolicyCheckInput
    33  	}{arg1})
    34  	fake.recordInvocation("Check", []interface{}{arg1})
    35  	fake.checkMutex.Unlock()
    36  	if fake.CheckStub != nil {
    37  		return fake.CheckStub(arg1)
    38  	}
    39  	if specificReturn {
    40  		return ret.result1, ret.result2
    41  	}
    42  	fakeReturns := fake.checkReturns
    43  	return fakeReturns.result1, fakeReturns.result2
    44  }
    45  
    46  func (fake *FakeAgent) CheckCallCount() int {
    47  	fake.checkMutex.RLock()
    48  	defer fake.checkMutex.RUnlock()
    49  	return len(fake.checkArgsForCall)
    50  }
    51  
    52  func (fake *FakeAgent) CheckCalls(stub func(policy.PolicyCheckInput) (bool, error)) {
    53  	fake.checkMutex.Lock()
    54  	defer fake.checkMutex.Unlock()
    55  	fake.CheckStub = stub
    56  }
    57  
    58  func (fake *FakeAgent) CheckArgsForCall(i int) policy.PolicyCheckInput {
    59  	fake.checkMutex.RLock()
    60  	defer fake.checkMutex.RUnlock()
    61  	argsForCall := fake.checkArgsForCall[i]
    62  	return argsForCall.arg1
    63  }
    64  
    65  func (fake *FakeAgent) CheckReturns(result1 bool, result2 error) {
    66  	fake.checkMutex.Lock()
    67  	defer fake.checkMutex.Unlock()
    68  	fake.CheckStub = nil
    69  	fake.checkReturns = struct {
    70  		result1 bool
    71  		result2 error
    72  	}{result1, result2}
    73  }
    74  
    75  func (fake *FakeAgent) CheckReturnsOnCall(i int, result1 bool, result2 error) {
    76  	fake.checkMutex.Lock()
    77  	defer fake.checkMutex.Unlock()
    78  	fake.CheckStub = nil
    79  	if fake.checkReturnsOnCall == nil {
    80  		fake.checkReturnsOnCall = make(map[int]struct {
    81  			result1 bool
    82  			result2 error
    83  		})
    84  	}
    85  	fake.checkReturnsOnCall[i] = struct {
    86  		result1 bool
    87  		result2 error
    88  	}{result1, result2}
    89  }
    90  
    91  func (fake *FakeAgent) Invocations() map[string][][]interface{} {
    92  	fake.invocationsMutex.RLock()
    93  	defer fake.invocationsMutex.RUnlock()
    94  	fake.checkMutex.RLock()
    95  	defer fake.checkMutex.RUnlock()
    96  	copiedInvocations := map[string][][]interface{}{}
    97  	for key, value := range fake.invocations {
    98  		copiedInvocations[key] = value
    99  	}
   100  	return copiedInvocations
   101  }
   102  
   103  func (fake *FakeAgent) recordInvocation(key string, args []interface{}) {
   104  	fake.invocationsMutex.Lock()
   105  	defer fake.invocationsMutex.Unlock()
   106  	if fake.invocations == nil {
   107  		fake.invocations = map[string][][]interface{}{}
   108  	}
   109  	if fake.invocations[key] == nil {
   110  		fake.invocations[key] = [][]interface{}{}
   111  	}
   112  	fake.invocations[key] = append(fake.invocations[key], args)
   113  }
   114  
   115  var _ policy.Agent = new(FakeAgent)