github.com/geofffranks/garden-linux@v0.0.0-20160715111146-26c893169cfa/network/subnets/fakes/fake_ipselector.go (about)

     1  // This file was generated by counterfeiter
     2  package fakes
     3  
     4  import (
     5  	"net"
     6  	"sync"
     7  
     8  	"code.cloudfoundry.org/garden-linux/network/subnets"
     9  )
    10  
    11  type FakeIPSelector struct {
    12  	SelectIPStub        func(subnet *net.IPNet, existing []net.IP) (net.IP, error)
    13  	selectIPMutex       sync.RWMutex
    14  	selectIPArgsForCall []struct {
    15  		subnet   *net.IPNet
    16  		existing []net.IP
    17  	}
    18  	selectIPReturns struct {
    19  		result1 net.IP
    20  		result2 error
    21  	}
    22  }
    23  
    24  func (fake *FakeIPSelector) SelectIP(subnet *net.IPNet, existing []net.IP) (net.IP, error) {
    25  	fake.selectIPMutex.Lock()
    26  	fake.selectIPArgsForCall = append(fake.selectIPArgsForCall, struct {
    27  		subnet   *net.IPNet
    28  		existing []net.IP
    29  	}{subnet, existing})
    30  	fake.selectIPMutex.Unlock()
    31  	if fake.SelectIPStub != nil {
    32  		return fake.SelectIPStub(subnet, existing)
    33  	} else {
    34  		return fake.selectIPReturns.result1, fake.selectIPReturns.result2
    35  	}
    36  }
    37  
    38  func (fake *FakeIPSelector) SelectIPCallCount() int {
    39  	fake.selectIPMutex.RLock()
    40  	defer fake.selectIPMutex.RUnlock()
    41  	return len(fake.selectIPArgsForCall)
    42  }
    43  
    44  func (fake *FakeIPSelector) SelectIPArgsForCall(i int) (*net.IPNet, []net.IP) {
    45  	fake.selectIPMutex.RLock()
    46  	defer fake.selectIPMutex.RUnlock()
    47  	return fake.selectIPArgsForCall[i].subnet, fake.selectIPArgsForCall[i].existing
    48  }
    49  
    50  func (fake *FakeIPSelector) SelectIPReturns(result1 net.IP, result2 error) {
    51  	fake.SelectIPStub = nil
    52  	fake.selectIPReturns = struct {
    53  		result1 net.IP
    54  		result2 error
    55  	}{result1, result2}
    56  }
    57  
    58  var _ subnets.IPSelector = new(FakeIPSelector)