github.com/mook-as/cf-cli@v7.0.0-beta.28.0.20200120190804-b91c115fae48+incompatible/cf/ssh/sshfakes/fake_secure_client.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package sshfakes
     3  
     4  import (
     5  	"net"
     6  	"sync"
     7  
     8  	sshCmd "code.cloudfoundry.org/cli/cf/ssh"
     9  	"golang.org/x/crypto/ssh"
    10  )
    11  
    12  type FakeSecureClient struct {
    13  	CloseStub        func() error
    14  	closeMutex       sync.RWMutex
    15  	closeArgsForCall []struct {
    16  	}
    17  	closeReturns struct {
    18  		result1 error
    19  	}
    20  	closeReturnsOnCall map[int]struct {
    21  		result1 error
    22  	}
    23  	ConnStub        func() ssh.Conn
    24  	connMutex       sync.RWMutex
    25  	connArgsForCall []struct {
    26  	}
    27  	connReturns struct {
    28  		result1 ssh.Conn
    29  	}
    30  	connReturnsOnCall map[int]struct {
    31  		result1 ssh.Conn
    32  	}
    33  	DialStub        func(string, string) (net.Conn, error)
    34  	dialMutex       sync.RWMutex
    35  	dialArgsForCall []struct {
    36  		arg1 string
    37  		arg2 string
    38  	}
    39  	dialReturns struct {
    40  		result1 net.Conn
    41  		result2 error
    42  	}
    43  	dialReturnsOnCall map[int]struct {
    44  		result1 net.Conn
    45  		result2 error
    46  	}
    47  	NewSessionStub        func() (sshCmd.SecureSession, error)
    48  	newSessionMutex       sync.RWMutex
    49  	newSessionArgsForCall []struct {
    50  	}
    51  	newSessionReturns struct {
    52  		result1 sshCmd.SecureSession
    53  		result2 error
    54  	}
    55  	newSessionReturnsOnCall map[int]struct {
    56  		result1 sshCmd.SecureSession
    57  		result2 error
    58  	}
    59  	WaitStub        func() error
    60  	waitMutex       sync.RWMutex
    61  	waitArgsForCall []struct {
    62  	}
    63  	waitReturns struct {
    64  		result1 error
    65  	}
    66  	waitReturnsOnCall map[int]struct {
    67  		result1 error
    68  	}
    69  	invocations      map[string][][]interface{}
    70  	invocationsMutex sync.RWMutex
    71  }
    72  
    73  func (fake *FakeSecureClient) Close() error {
    74  	fake.closeMutex.Lock()
    75  	ret, specificReturn := fake.closeReturnsOnCall[len(fake.closeArgsForCall)]
    76  	fake.closeArgsForCall = append(fake.closeArgsForCall, struct {
    77  	}{})
    78  	fake.recordInvocation("Close", []interface{}{})
    79  	fake.closeMutex.Unlock()
    80  	if fake.CloseStub != nil {
    81  		return fake.CloseStub()
    82  	}
    83  	if specificReturn {
    84  		return ret.result1
    85  	}
    86  	fakeReturns := fake.closeReturns
    87  	return fakeReturns.result1
    88  }
    89  
    90  func (fake *FakeSecureClient) CloseCallCount() int {
    91  	fake.closeMutex.RLock()
    92  	defer fake.closeMutex.RUnlock()
    93  	return len(fake.closeArgsForCall)
    94  }
    95  
    96  func (fake *FakeSecureClient) CloseCalls(stub func() error) {
    97  	fake.closeMutex.Lock()
    98  	defer fake.closeMutex.Unlock()
    99  	fake.CloseStub = stub
   100  }
   101  
   102  func (fake *FakeSecureClient) CloseReturns(result1 error) {
   103  	fake.closeMutex.Lock()
   104  	defer fake.closeMutex.Unlock()
   105  	fake.CloseStub = nil
   106  	fake.closeReturns = struct {
   107  		result1 error
   108  	}{result1}
   109  }
   110  
   111  func (fake *FakeSecureClient) CloseReturnsOnCall(i int, result1 error) {
   112  	fake.closeMutex.Lock()
   113  	defer fake.closeMutex.Unlock()
   114  	fake.CloseStub = nil
   115  	if fake.closeReturnsOnCall == nil {
   116  		fake.closeReturnsOnCall = make(map[int]struct {
   117  			result1 error
   118  		})
   119  	}
   120  	fake.closeReturnsOnCall[i] = struct {
   121  		result1 error
   122  	}{result1}
   123  }
   124  
   125  func (fake *FakeSecureClient) Conn() ssh.Conn {
   126  	fake.connMutex.Lock()
   127  	ret, specificReturn := fake.connReturnsOnCall[len(fake.connArgsForCall)]
   128  	fake.connArgsForCall = append(fake.connArgsForCall, struct {
   129  	}{})
   130  	fake.recordInvocation("Conn", []interface{}{})
   131  	fake.connMutex.Unlock()
   132  	if fake.ConnStub != nil {
   133  		return fake.ConnStub()
   134  	}
   135  	if specificReturn {
   136  		return ret.result1
   137  	}
   138  	fakeReturns := fake.connReturns
   139  	return fakeReturns.result1
   140  }
   141  
   142  func (fake *FakeSecureClient) ConnCallCount() int {
   143  	fake.connMutex.RLock()
   144  	defer fake.connMutex.RUnlock()
   145  	return len(fake.connArgsForCall)
   146  }
   147  
   148  func (fake *FakeSecureClient) ConnCalls(stub func() ssh.Conn) {
   149  	fake.connMutex.Lock()
   150  	defer fake.connMutex.Unlock()
   151  	fake.ConnStub = stub
   152  }
   153  
   154  func (fake *FakeSecureClient) ConnReturns(result1 ssh.Conn) {
   155  	fake.connMutex.Lock()
   156  	defer fake.connMutex.Unlock()
   157  	fake.ConnStub = nil
   158  	fake.connReturns = struct {
   159  		result1 ssh.Conn
   160  	}{result1}
   161  }
   162  
   163  func (fake *FakeSecureClient) ConnReturnsOnCall(i int, result1 ssh.Conn) {
   164  	fake.connMutex.Lock()
   165  	defer fake.connMutex.Unlock()
   166  	fake.ConnStub = nil
   167  	if fake.connReturnsOnCall == nil {
   168  		fake.connReturnsOnCall = make(map[int]struct {
   169  			result1 ssh.Conn
   170  		})
   171  	}
   172  	fake.connReturnsOnCall[i] = struct {
   173  		result1 ssh.Conn
   174  	}{result1}
   175  }
   176  
   177  func (fake *FakeSecureClient) Dial(arg1 string, arg2 string) (net.Conn, error) {
   178  	fake.dialMutex.Lock()
   179  	ret, specificReturn := fake.dialReturnsOnCall[len(fake.dialArgsForCall)]
   180  	fake.dialArgsForCall = append(fake.dialArgsForCall, struct {
   181  		arg1 string
   182  		arg2 string
   183  	}{arg1, arg2})
   184  	fake.recordInvocation("Dial", []interface{}{arg1, arg2})
   185  	fake.dialMutex.Unlock()
   186  	if fake.DialStub != nil {
   187  		return fake.DialStub(arg1, arg2)
   188  	}
   189  	if specificReturn {
   190  		return ret.result1, ret.result2
   191  	}
   192  	fakeReturns := fake.dialReturns
   193  	return fakeReturns.result1, fakeReturns.result2
   194  }
   195  
   196  func (fake *FakeSecureClient) DialCallCount() int {
   197  	fake.dialMutex.RLock()
   198  	defer fake.dialMutex.RUnlock()
   199  	return len(fake.dialArgsForCall)
   200  }
   201  
   202  func (fake *FakeSecureClient) DialCalls(stub func(string, string) (net.Conn, error)) {
   203  	fake.dialMutex.Lock()
   204  	defer fake.dialMutex.Unlock()
   205  	fake.DialStub = stub
   206  }
   207  
   208  func (fake *FakeSecureClient) DialArgsForCall(i int) (string, string) {
   209  	fake.dialMutex.RLock()
   210  	defer fake.dialMutex.RUnlock()
   211  	argsForCall := fake.dialArgsForCall[i]
   212  	return argsForCall.arg1, argsForCall.arg2
   213  }
   214  
   215  func (fake *FakeSecureClient) DialReturns(result1 net.Conn, result2 error) {
   216  	fake.dialMutex.Lock()
   217  	defer fake.dialMutex.Unlock()
   218  	fake.DialStub = nil
   219  	fake.dialReturns = struct {
   220  		result1 net.Conn
   221  		result2 error
   222  	}{result1, result2}
   223  }
   224  
   225  func (fake *FakeSecureClient) DialReturnsOnCall(i int, result1 net.Conn, result2 error) {
   226  	fake.dialMutex.Lock()
   227  	defer fake.dialMutex.Unlock()
   228  	fake.DialStub = nil
   229  	if fake.dialReturnsOnCall == nil {
   230  		fake.dialReturnsOnCall = make(map[int]struct {
   231  			result1 net.Conn
   232  			result2 error
   233  		})
   234  	}
   235  	fake.dialReturnsOnCall[i] = struct {
   236  		result1 net.Conn
   237  		result2 error
   238  	}{result1, result2}
   239  }
   240  
   241  func (fake *FakeSecureClient) NewSession() (sshCmd.SecureSession, error) {
   242  	fake.newSessionMutex.Lock()
   243  	ret, specificReturn := fake.newSessionReturnsOnCall[len(fake.newSessionArgsForCall)]
   244  	fake.newSessionArgsForCall = append(fake.newSessionArgsForCall, struct {
   245  	}{})
   246  	fake.recordInvocation("NewSession", []interface{}{})
   247  	fake.newSessionMutex.Unlock()
   248  	if fake.NewSessionStub != nil {
   249  		return fake.NewSessionStub()
   250  	}
   251  	if specificReturn {
   252  		return ret.result1, ret.result2
   253  	}
   254  	fakeReturns := fake.newSessionReturns
   255  	return fakeReturns.result1, fakeReturns.result2
   256  }
   257  
   258  func (fake *FakeSecureClient) NewSessionCallCount() int {
   259  	fake.newSessionMutex.RLock()
   260  	defer fake.newSessionMutex.RUnlock()
   261  	return len(fake.newSessionArgsForCall)
   262  }
   263  
   264  func (fake *FakeSecureClient) NewSessionCalls(stub func() (sshCmd.SecureSession, error)) {
   265  	fake.newSessionMutex.Lock()
   266  	defer fake.newSessionMutex.Unlock()
   267  	fake.NewSessionStub = stub
   268  }
   269  
   270  func (fake *FakeSecureClient) NewSessionReturns(result1 sshCmd.SecureSession, result2 error) {
   271  	fake.newSessionMutex.Lock()
   272  	defer fake.newSessionMutex.Unlock()
   273  	fake.NewSessionStub = nil
   274  	fake.newSessionReturns = struct {
   275  		result1 sshCmd.SecureSession
   276  		result2 error
   277  	}{result1, result2}
   278  }
   279  
   280  func (fake *FakeSecureClient) NewSessionReturnsOnCall(i int, result1 sshCmd.SecureSession, result2 error) {
   281  	fake.newSessionMutex.Lock()
   282  	defer fake.newSessionMutex.Unlock()
   283  	fake.NewSessionStub = nil
   284  	if fake.newSessionReturnsOnCall == nil {
   285  		fake.newSessionReturnsOnCall = make(map[int]struct {
   286  			result1 sshCmd.SecureSession
   287  			result2 error
   288  		})
   289  	}
   290  	fake.newSessionReturnsOnCall[i] = struct {
   291  		result1 sshCmd.SecureSession
   292  		result2 error
   293  	}{result1, result2}
   294  }
   295  
   296  func (fake *FakeSecureClient) Wait() error {
   297  	fake.waitMutex.Lock()
   298  	ret, specificReturn := fake.waitReturnsOnCall[len(fake.waitArgsForCall)]
   299  	fake.waitArgsForCall = append(fake.waitArgsForCall, struct {
   300  	}{})
   301  	fake.recordInvocation("Wait", []interface{}{})
   302  	fake.waitMutex.Unlock()
   303  	if fake.WaitStub != nil {
   304  		return fake.WaitStub()
   305  	}
   306  	if specificReturn {
   307  		return ret.result1
   308  	}
   309  	fakeReturns := fake.waitReturns
   310  	return fakeReturns.result1
   311  }
   312  
   313  func (fake *FakeSecureClient) WaitCallCount() int {
   314  	fake.waitMutex.RLock()
   315  	defer fake.waitMutex.RUnlock()
   316  	return len(fake.waitArgsForCall)
   317  }
   318  
   319  func (fake *FakeSecureClient) WaitCalls(stub func() error) {
   320  	fake.waitMutex.Lock()
   321  	defer fake.waitMutex.Unlock()
   322  	fake.WaitStub = stub
   323  }
   324  
   325  func (fake *FakeSecureClient) WaitReturns(result1 error) {
   326  	fake.waitMutex.Lock()
   327  	defer fake.waitMutex.Unlock()
   328  	fake.WaitStub = nil
   329  	fake.waitReturns = struct {
   330  		result1 error
   331  	}{result1}
   332  }
   333  
   334  func (fake *FakeSecureClient) WaitReturnsOnCall(i int, result1 error) {
   335  	fake.waitMutex.Lock()
   336  	defer fake.waitMutex.Unlock()
   337  	fake.WaitStub = nil
   338  	if fake.waitReturnsOnCall == nil {
   339  		fake.waitReturnsOnCall = make(map[int]struct {
   340  			result1 error
   341  		})
   342  	}
   343  	fake.waitReturnsOnCall[i] = struct {
   344  		result1 error
   345  	}{result1}
   346  }
   347  
   348  func (fake *FakeSecureClient) Invocations() map[string][][]interface{} {
   349  	fake.invocationsMutex.RLock()
   350  	defer fake.invocationsMutex.RUnlock()
   351  	fake.closeMutex.RLock()
   352  	defer fake.closeMutex.RUnlock()
   353  	fake.connMutex.RLock()
   354  	defer fake.connMutex.RUnlock()
   355  	fake.dialMutex.RLock()
   356  	defer fake.dialMutex.RUnlock()
   357  	fake.newSessionMutex.RLock()
   358  	defer fake.newSessionMutex.RUnlock()
   359  	fake.waitMutex.RLock()
   360  	defer fake.waitMutex.RUnlock()
   361  	copiedInvocations := map[string][][]interface{}{}
   362  	for key, value := range fake.invocations {
   363  		copiedInvocations[key] = value
   364  	}
   365  	return copiedInvocations
   366  }
   367  
   368  func (fake *FakeSecureClient) recordInvocation(key string, args []interface{}) {
   369  	fake.invocationsMutex.Lock()
   370  	defer fake.invocationsMutex.Unlock()
   371  	if fake.invocations == nil {
   372  		fake.invocations = map[string][][]interface{}{}
   373  	}
   374  	if fake.invocations[key] == nil {
   375  		fake.invocations[key] = [][]interface{}{}
   376  	}
   377  	fake.invocations[key] = append(fake.invocations[key], args)
   378  }
   379  
   380  var _ sshCmd.SecureClient = new(FakeSecureClient)