github.com/cloudfoundry-attic/ltc@v0.0.0-20151123212628-098adc7919fc/ssh/sshapi/mocks/fake_dialer.go (about)

     1  // This file was generated by counterfeiter
     2  package mocks
     3  
     4  import (
     5  	"net"
     6  	"sync"
     7  
     8  	"github.com/cloudfoundry-incubator/ltc/ssh/sshapi"
     9  )
    10  
    11  type FakeDialer struct {
    12  	DialStub        func(n, addr string) (net.Conn, error)
    13  	dialMutex       sync.RWMutex
    14  	dialArgsForCall []struct {
    15  		n    string
    16  		addr string
    17  	}
    18  	dialReturns struct {
    19  		result1 net.Conn
    20  		result2 error
    21  	}
    22  }
    23  
    24  func (fake *FakeDialer) Dial(n string, addr string) (net.Conn, error) {
    25  	fake.dialMutex.Lock()
    26  	fake.dialArgsForCall = append(fake.dialArgsForCall, struct {
    27  		n    string
    28  		addr string
    29  	}{n, addr})
    30  	fake.dialMutex.Unlock()
    31  	if fake.DialStub != nil {
    32  		return fake.DialStub(n, addr)
    33  	} else {
    34  		return fake.dialReturns.result1, fake.dialReturns.result2
    35  	}
    36  }
    37  
    38  func (fake *FakeDialer) DialCallCount() int {
    39  	fake.dialMutex.RLock()
    40  	defer fake.dialMutex.RUnlock()
    41  	return len(fake.dialArgsForCall)
    42  }
    43  
    44  func (fake *FakeDialer) DialArgsForCall(i int) (string, string) {
    45  	fake.dialMutex.RLock()
    46  	defer fake.dialMutex.RUnlock()
    47  	return fake.dialArgsForCall[i].n, fake.dialArgsForCall[i].addr
    48  }
    49  
    50  func (fake *FakeDialer) DialReturns(result1 net.Conn, result2 error) {
    51  	fake.DialStub = nil
    52  	fake.dialReturns = struct {
    53  		result1 net.Conn
    54  		result2 error
    55  	}{result1, result2}
    56  }
    57  
    58  var _ sshapi.Dialer = new(FakeDialer)