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