github.com/geofffranks/garden-linux@v0.0.0-20160715111146-26c893169cfa/linux_container/iptables_manager/fake_chain/fake_chain.go (about)

     1  // This file was generated by counterfeiter
     2  package fake_chain
     3  
     4  import (
     5  	"net"
     6  	"sync"
     7  
     8  	"code.cloudfoundry.org/garden-linux/linux_container/iptables_manager"
     9  )
    10  
    11  type FakeChain struct {
    12  	SetupStub        func(containerID, bridgeName string, ip net.IP, network *net.IPNet) error
    13  	setupMutex       sync.RWMutex
    14  	setupArgsForCall []struct {
    15  		containerID string
    16  		bridgeName  string
    17  		ip          net.IP
    18  		network     *net.IPNet
    19  	}
    20  	setupReturns struct {
    21  		result1 error
    22  	}
    23  	TeardownStub        func(containerID string) error
    24  	teardownMutex       sync.RWMutex
    25  	teardownArgsForCall []struct {
    26  		containerID string
    27  	}
    28  	teardownReturns struct {
    29  		result1 error
    30  	}
    31  }
    32  
    33  func (fake *FakeChain) Setup(containerID string, bridgeName string, ip net.IP, network *net.IPNet) error {
    34  	fake.setupMutex.Lock()
    35  	fake.setupArgsForCall = append(fake.setupArgsForCall, struct {
    36  		containerID string
    37  		bridgeName  string
    38  		ip          net.IP
    39  		network     *net.IPNet
    40  	}{containerID, bridgeName, ip, network})
    41  	fake.setupMutex.Unlock()
    42  	if fake.SetupStub != nil {
    43  		return fake.SetupStub(containerID, bridgeName, ip, network)
    44  	} else {
    45  		return fake.setupReturns.result1
    46  	}
    47  }
    48  
    49  func (fake *FakeChain) SetupCallCount() int {
    50  	fake.setupMutex.RLock()
    51  	defer fake.setupMutex.RUnlock()
    52  	return len(fake.setupArgsForCall)
    53  }
    54  
    55  func (fake *FakeChain) SetupArgsForCall(i int) (string, string, net.IP, *net.IPNet) {
    56  	fake.setupMutex.RLock()
    57  	defer fake.setupMutex.RUnlock()
    58  	return fake.setupArgsForCall[i].containerID, fake.setupArgsForCall[i].bridgeName, fake.setupArgsForCall[i].ip, fake.setupArgsForCall[i].network
    59  }
    60  
    61  func (fake *FakeChain) SetupReturns(result1 error) {
    62  	fake.SetupStub = nil
    63  	fake.setupReturns = struct {
    64  		result1 error
    65  	}{result1}
    66  }
    67  
    68  func (fake *FakeChain) Teardown(containerID string) error {
    69  	fake.teardownMutex.Lock()
    70  	fake.teardownArgsForCall = append(fake.teardownArgsForCall, struct {
    71  		containerID string
    72  	}{containerID})
    73  	fake.teardownMutex.Unlock()
    74  	if fake.TeardownStub != nil {
    75  		return fake.TeardownStub(containerID)
    76  	} else {
    77  		return fake.teardownReturns.result1
    78  	}
    79  }
    80  
    81  func (fake *FakeChain) TeardownCallCount() int {
    82  	fake.teardownMutex.RLock()
    83  	defer fake.teardownMutex.RUnlock()
    84  	return len(fake.teardownArgsForCall)
    85  }
    86  
    87  func (fake *FakeChain) TeardownArgsForCall(i int) string {
    88  	fake.teardownMutex.RLock()
    89  	defer fake.teardownMutex.RUnlock()
    90  	return fake.teardownArgsForCall[i].containerID
    91  }
    92  
    93  func (fake *FakeChain) TeardownReturns(result1 error) {
    94  	fake.TeardownStub = nil
    95  	fake.teardownReturns = struct {
    96  		result1 error
    97  	}{result1}
    98  }
    99  
   100  var _ iptables_manager.Chain = new(FakeChain)