github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/engine/enginefakes/fake_engine.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package enginefakes
     3  
     4  import (
     5  	"context"
     6  	"sync"
     7  
     8  	"github.com/pf-qiu/concourse/v6/atc/db"
     9  	"github.com/pf-qiu/concourse/v6/atc/engine"
    10  )
    11  
    12  type FakeEngine struct {
    13  	DrainStub        func(context.Context)
    14  	drainMutex       sync.RWMutex
    15  	drainArgsForCall []struct {
    16  		arg1 context.Context
    17  	}
    18  	NewBuildStub        func(db.Build) engine.Runnable
    19  	newBuildMutex       sync.RWMutex
    20  	newBuildArgsForCall []struct {
    21  		arg1 db.Build
    22  	}
    23  	newBuildReturns struct {
    24  		result1 engine.Runnable
    25  	}
    26  	newBuildReturnsOnCall map[int]struct {
    27  		result1 engine.Runnable
    28  	}
    29  	invocations      map[string][][]interface{}
    30  	invocationsMutex sync.RWMutex
    31  }
    32  
    33  func (fake *FakeEngine) Drain(arg1 context.Context) {
    34  	fake.drainMutex.Lock()
    35  	fake.drainArgsForCall = append(fake.drainArgsForCall, struct {
    36  		arg1 context.Context
    37  	}{arg1})
    38  	fake.recordInvocation("Drain", []interface{}{arg1})
    39  	fake.drainMutex.Unlock()
    40  	if fake.DrainStub != nil {
    41  		fake.DrainStub(arg1)
    42  	}
    43  }
    44  
    45  func (fake *FakeEngine) DrainCallCount() int {
    46  	fake.drainMutex.RLock()
    47  	defer fake.drainMutex.RUnlock()
    48  	return len(fake.drainArgsForCall)
    49  }
    50  
    51  func (fake *FakeEngine) DrainCalls(stub func(context.Context)) {
    52  	fake.drainMutex.Lock()
    53  	defer fake.drainMutex.Unlock()
    54  	fake.DrainStub = stub
    55  }
    56  
    57  func (fake *FakeEngine) DrainArgsForCall(i int) context.Context {
    58  	fake.drainMutex.RLock()
    59  	defer fake.drainMutex.RUnlock()
    60  	argsForCall := fake.drainArgsForCall[i]
    61  	return argsForCall.arg1
    62  }
    63  
    64  func (fake *FakeEngine) NewBuild(arg1 db.Build) engine.Runnable {
    65  	fake.newBuildMutex.Lock()
    66  	ret, specificReturn := fake.newBuildReturnsOnCall[len(fake.newBuildArgsForCall)]
    67  	fake.newBuildArgsForCall = append(fake.newBuildArgsForCall, struct {
    68  		arg1 db.Build
    69  	}{arg1})
    70  	fake.recordInvocation("NewBuild", []interface{}{arg1})
    71  	fake.newBuildMutex.Unlock()
    72  	if fake.NewBuildStub != nil {
    73  		return fake.NewBuildStub(arg1)
    74  	}
    75  	if specificReturn {
    76  		return ret.result1
    77  	}
    78  	fakeReturns := fake.newBuildReturns
    79  	return fakeReturns.result1
    80  }
    81  
    82  func (fake *FakeEngine) NewBuildCallCount() int {
    83  	fake.newBuildMutex.RLock()
    84  	defer fake.newBuildMutex.RUnlock()
    85  	return len(fake.newBuildArgsForCall)
    86  }
    87  
    88  func (fake *FakeEngine) NewBuildCalls(stub func(db.Build) engine.Runnable) {
    89  	fake.newBuildMutex.Lock()
    90  	defer fake.newBuildMutex.Unlock()
    91  	fake.NewBuildStub = stub
    92  }
    93  
    94  func (fake *FakeEngine) NewBuildArgsForCall(i int) db.Build {
    95  	fake.newBuildMutex.RLock()
    96  	defer fake.newBuildMutex.RUnlock()
    97  	argsForCall := fake.newBuildArgsForCall[i]
    98  	return argsForCall.arg1
    99  }
   100  
   101  func (fake *FakeEngine) NewBuildReturns(result1 engine.Runnable) {
   102  	fake.newBuildMutex.Lock()
   103  	defer fake.newBuildMutex.Unlock()
   104  	fake.NewBuildStub = nil
   105  	fake.newBuildReturns = struct {
   106  		result1 engine.Runnable
   107  	}{result1}
   108  }
   109  
   110  func (fake *FakeEngine) NewBuildReturnsOnCall(i int, result1 engine.Runnable) {
   111  	fake.newBuildMutex.Lock()
   112  	defer fake.newBuildMutex.Unlock()
   113  	fake.NewBuildStub = nil
   114  	if fake.newBuildReturnsOnCall == nil {
   115  		fake.newBuildReturnsOnCall = make(map[int]struct {
   116  			result1 engine.Runnable
   117  		})
   118  	}
   119  	fake.newBuildReturnsOnCall[i] = struct {
   120  		result1 engine.Runnable
   121  	}{result1}
   122  }
   123  
   124  func (fake *FakeEngine) Invocations() map[string][][]interface{} {
   125  	fake.invocationsMutex.RLock()
   126  	defer fake.invocationsMutex.RUnlock()
   127  	fake.drainMutex.RLock()
   128  	defer fake.drainMutex.RUnlock()
   129  	fake.newBuildMutex.RLock()
   130  	defer fake.newBuildMutex.RUnlock()
   131  	copiedInvocations := map[string][][]interface{}{}
   132  	for key, value := range fake.invocations {
   133  		copiedInvocations[key] = value
   134  	}
   135  	return copiedInvocations
   136  }
   137  
   138  func (fake *FakeEngine) recordInvocation(key string, args []interface{}) {
   139  	fake.invocationsMutex.Lock()
   140  	defer fake.invocationsMutex.Unlock()
   141  	if fake.invocations == nil {
   142  		fake.invocations = map[string][][]interface{}{}
   143  	}
   144  	if fake.invocations[key] == nil {
   145  		fake.invocations[key] = [][]interface{}{}
   146  	}
   147  	fake.invocations[key] = append(fake.invocations[key], args)
   148  }
   149  
   150  var _ engine.Engine = new(FakeEngine)