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)