github.com/geofffranks/garden-linux@v0.0.0-20160715111146-26c893169cfa/container_daemon/fake_connector/FakeConnector.go (about) 1 // This file was generated by counterfeiter 2 package fake_connector 3 4 import ( 5 "sync" 6 7 "code.cloudfoundry.org/garden-linux/container_daemon" 8 ) 9 10 type FakeConnector struct { 11 ConnectStub func(msg *container_daemon.RequestMessage) (*container_daemon.ResponseMessage, error) 12 connectMutex sync.RWMutex 13 connectArgsForCall []struct { 14 msg *container_daemon.RequestMessage 15 } 16 connectReturns struct { 17 result1 *container_daemon.ResponseMessage 18 result2 error 19 } 20 } 21 22 func (fake *FakeConnector) Connect(msg *container_daemon.RequestMessage) (*container_daemon.ResponseMessage, error) { 23 fake.connectMutex.Lock() 24 fake.connectArgsForCall = append(fake.connectArgsForCall, struct { 25 msg *container_daemon.RequestMessage 26 }{msg}) 27 fake.connectMutex.Unlock() 28 if fake.ConnectStub != nil { 29 return fake.ConnectStub(msg) 30 } else { 31 return fake.connectReturns.result1, fake.connectReturns.result2 32 } 33 } 34 35 func (fake *FakeConnector) ConnectCallCount() int { 36 fake.connectMutex.RLock() 37 defer fake.connectMutex.RUnlock() 38 return len(fake.connectArgsForCall) 39 } 40 41 func (fake *FakeConnector) ConnectArgsForCall(i int) *container_daemon.RequestMessage { 42 fake.connectMutex.RLock() 43 defer fake.connectMutex.RUnlock() 44 return fake.connectArgsForCall[i].msg 45 } 46 47 func (fake *FakeConnector) ConnectReturns(result1 *container_daemon.ResponseMessage, result2 error) { 48 fake.ConnectStub = nil 49 fake.connectReturns = struct { 50 result1 *container_daemon.ResponseMessage 51 result2 error 52 }{result1, result2} 53 } 54 55 var _ container_daemon.Connector = new(FakeConnector)