github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/db/lock/lockfakes/fake_lock_db.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package lockfakes
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/pf-qiu/concourse/v6/atc/db/lock"
     8  )
     9  
    10  type FakeLockDB struct {
    11  	AcquireStub        func(lock.LockID) (bool, error)
    12  	acquireMutex       sync.RWMutex
    13  	acquireArgsForCall []struct {
    14  		arg1 lock.LockID
    15  	}
    16  	acquireReturns struct {
    17  		result1 bool
    18  		result2 error
    19  	}
    20  	acquireReturnsOnCall map[int]struct {
    21  		result1 bool
    22  		result2 error
    23  	}
    24  	ReleaseStub        func(lock.LockID) (bool, error)
    25  	releaseMutex       sync.RWMutex
    26  	releaseArgsForCall []struct {
    27  		arg1 lock.LockID
    28  	}
    29  	releaseReturns struct {
    30  		result1 bool
    31  		result2 error
    32  	}
    33  	releaseReturnsOnCall map[int]struct {
    34  		result1 bool
    35  		result2 error
    36  	}
    37  	invocations      map[string][][]interface{}
    38  	invocationsMutex sync.RWMutex
    39  }
    40  
    41  func (fake *FakeLockDB) Acquire(arg1 lock.LockID) (bool, error) {
    42  	fake.acquireMutex.Lock()
    43  	ret, specificReturn := fake.acquireReturnsOnCall[len(fake.acquireArgsForCall)]
    44  	fake.acquireArgsForCall = append(fake.acquireArgsForCall, struct {
    45  		arg1 lock.LockID
    46  	}{arg1})
    47  	fake.recordInvocation("Acquire", []interface{}{arg1})
    48  	fake.acquireMutex.Unlock()
    49  	if fake.AcquireStub != nil {
    50  		return fake.AcquireStub(arg1)
    51  	}
    52  	if specificReturn {
    53  		return ret.result1, ret.result2
    54  	}
    55  	fakeReturns := fake.acquireReturns
    56  	return fakeReturns.result1, fakeReturns.result2
    57  }
    58  
    59  func (fake *FakeLockDB) AcquireCallCount() int {
    60  	fake.acquireMutex.RLock()
    61  	defer fake.acquireMutex.RUnlock()
    62  	return len(fake.acquireArgsForCall)
    63  }
    64  
    65  func (fake *FakeLockDB) AcquireCalls(stub func(lock.LockID) (bool, error)) {
    66  	fake.acquireMutex.Lock()
    67  	defer fake.acquireMutex.Unlock()
    68  	fake.AcquireStub = stub
    69  }
    70  
    71  func (fake *FakeLockDB) AcquireArgsForCall(i int) lock.LockID {
    72  	fake.acquireMutex.RLock()
    73  	defer fake.acquireMutex.RUnlock()
    74  	argsForCall := fake.acquireArgsForCall[i]
    75  	return argsForCall.arg1
    76  }
    77  
    78  func (fake *FakeLockDB) AcquireReturns(result1 bool, result2 error) {
    79  	fake.acquireMutex.Lock()
    80  	defer fake.acquireMutex.Unlock()
    81  	fake.AcquireStub = nil
    82  	fake.acquireReturns = struct {
    83  		result1 bool
    84  		result2 error
    85  	}{result1, result2}
    86  }
    87  
    88  func (fake *FakeLockDB) AcquireReturnsOnCall(i int, result1 bool, result2 error) {
    89  	fake.acquireMutex.Lock()
    90  	defer fake.acquireMutex.Unlock()
    91  	fake.AcquireStub = nil
    92  	if fake.acquireReturnsOnCall == nil {
    93  		fake.acquireReturnsOnCall = make(map[int]struct {
    94  			result1 bool
    95  			result2 error
    96  		})
    97  	}
    98  	fake.acquireReturnsOnCall[i] = struct {
    99  		result1 bool
   100  		result2 error
   101  	}{result1, result2}
   102  }
   103  
   104  func (fake *FakeLockDB) Release(arg1 lock.LockID) (bool, error) {
   105  	fake.releaseMutex.Lock()
   106  	ret, specificReturn := fake.releaseReturnsOnCall[len(fake.releaseArgsForCall)]
   107  	fake.releaseArgsForCall = append(fake.releaseArgsForCall, struct {
   108  		arg1 lock.LockID
   109  	}{arg1})
   110  	fake.recordInvocation("Release", []interface{}{arg1})
   111  	fake.releaseMutex.Unlock()
   112  	if fake.ReleaseStub != nil {
   113  		return fake.ReleaseStub(arg1)
   114  	}
   115  	if specificReturn {
   116  		return ret.result1, ret.result2
   117  	}
   118  	fakeReturns := fake.releaseReturns
   119  	return fakeReturns.result1, fakeReturns.result2
   120  }
   121  
   122  func (fake *FakeLockDB) ReleaseCallCount() int {
   123  	fake.releaseMutex.RLock()
   124  	defer fake.releaseMutex.RUnlock()
   125  	return len(fake.releaseArgsForCall)
   126  }
   127  
   128  func (fake *FakeLockDB) ReleaseCalls(stub func(lock.LockID) (bool, error)) {
   129  	fake.releaseMutex.Lock()
   130  	defer fake.releaseMutex.Unlock()
   131  	fake.ReleaseStub = stub
   132  }
   133  
   134  func (fake *FakeLockDB) ReleaseArgsForCall(i int) lock.LockID {
   135  	fake.releaseMutex.RLock()
   136  	defer fake.releaseMutex.RUnlock()
   137  	argsForCall := fake.releaseArgsForCall[i]
   138  	return argsForCall.arg1
   139  }
   140  
   141  func (fake *FakeLockDB) ReleaseReturns(result1 bool, result2 error) {
   142  	fake.releaseMutex.Lock()
   143  	defer fake.releaseMutex.Unlock()
   144  	fake.ReleaseStub = nil
   145  	fake.releaseReturns = struct {
   146  		result1 bool
   147  		result2 error
   148  	}{result1, result2}
   149  }
   150  
   151  func (fake *FakeLockDB) ReleaseReturnsOnCall(i int, result1 bool, result2 error) {
   152  	fake.releaseMutex.Lock()
   153  	defer fake.releaseMutex.Unlock()
   154  	fake.ReleaseStub = nil
   155  	if fake.releaseReturnsOnCall == nil {
   156  		fake.releaseReturnsOnCall = make(map[int]struct {
   157  			result1 bool
   158  			result2 error
   159  		})
   160  	}
   161  	fake.releaseReturnsOnCall[i] = struct {
   162  		result1 bool
   163  		result2 error
   164  	}{result1, result2}
   165  }
   166  
   167  func (fake *FakeLockDB) Invocations() map[string][][]interface{} {
   168  	fake.invocationsMutex.RLock()
   169  	defer fake.invocationsMutex.RUnlock()
   170  	fake.acquireMutex.RLock()
   171  	defer fake.acquireMutex.RUnlock()
   172  	fake.releaseMutex.RLock()
   173  	defer fake.releaseMutex.RUnlock()
   174  	copiedInvocations := map[string][][]interface{}{}
   175  	for key, value := range fake.invocations {
   176  		copiedInvocations[key] = value
   177  	}
   178  	return copiedInvocations
   179  }
   180  
   181  func (fake *FakeLockDB) recordInvocation(key string, args []interface{}) {
   182  	fake.invocationsMutex.Lock()
   183  	defer fake.invocationsMutex.Unlock()
   184  	if fake.invocations == nil {
   185  		fake.invocations = map[string][][]interface{}{}
   186  	}
   187  	if fake.invocations[key] == nil {
   188  		fake.invocations[key] = [][]interface{}{}
   189  	}
   190  	fake.invocations[key] = append(fake.invocations[key], args)
   191  }
   192  
   193  var _ lock.LockDB = new(FakeLockDB)