github.com/nimakaviani/cli@v6.37.1-0.20180619223813-e734901a73fa+incompatible/util/clissh/clisshfakes/fake_secure_session.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package clisshfakes
     3  
     4  import (
     5  	"io"
     6  	"sync"
     7  
     8  	"code.cloudfoundry.org/cli/util/clissh"
     9  	"golang.org/x/crypto/ssh"
    10  )
    11  
    12  type FakeSecureSession struct {
    13  	RequestPtyStub        func(term string, height, width int, termModes ssh.TerminalModes) error
    14  	requestPtyMutex       sync.RWMutex
    15  	requestPtyArgsForCall []struct {
    16  		term      string
    17  		height    int
    18  		width     int
    19  		termModes ssh.TerminalModes
    20  	}
    21  	requestPtyReturns struct {
    22  		result1 error
    23  	}
    24  	requestPtyReturnsOnCall map[int]struct {
    25  		result1 error
    26  	}
    27  	SendRequestStub        func(name string, wantReply bool, payload []byte) (bool, error)
    28  	sendRequestMutex       sync.RWMutex
    29  	sendRequestArgsForCall []struct {
    30  		name      string
    31  		wantReply bool
    32  		payload   []byte
    33  	}
    34  	sendRequestReturns struct {
    35  		result1 bool
    36  		result2 error
    37  	}
    38  	sendRequestReturnsOnCall map[int]struct {
    39  		result1 bool
    40  		result2 error
    41  	}
    42  	StdinPipeStub        func() (io.WriteCloser, error)
    43  	stdinPipeMutex       sync.RWMutex
    44  	stdinPipeArgsForCall []struct{}
    45  	stdinPipeReturns     struct {
    46  		result1 io.WriteCloser
    47  		result2 error
    48  	}
    49  	stdinPipeReturnsOnCall map[int]struct {
    50  		result1 io.WriteCloser
    51  		result2 error
    52  	}
    53  	StdoutPipeStub        func() (io.Reader, error)
    54  	stdoutPipeMutex       sync.RWMutex
    55  	stdoutPipeArgsForCall []struct{}
    56  	stdoutPipeReturns     struct {
    57  		result1 io.Reader
    58  		result2 error
    59  	}
    60  	stdoutPipeReturnsOnCall map[int]struct {
    61  		result1 io.Reader
    62  		result2 error
    63  	}
    64  	StderrPipeStub        func() (io.Reader, error)
    65  	stderrPipeMutex       sync.RWMutex
    66  	stderrPipeArgsForCall []struct{}
    67  	stderrPipeReturns     struct {
    68  		result1 io.Reader
    69  		result2 error
    70  	}
    71  	stderrPipeReturnsOnCall map[int]struct {
    72  		result1 io.Reader
    73  		result2 error
    74  	}
    75  	StartStub        func(command string) error
    76  	startMutex       sync.RWMutex
    77  	startArgsForCall []struct {
    78  		command string
    79  	}
    80  	startReturns struct {
    81  		result1 error
    82  	}
    83  	startReturnsOnCall map[int]struct {
    84  		result1 error
    85  	}
    86  	ShellStub        func() error
    87  	shellMutex       sync.RWMutex
    88  	shellArgsForCall []struct{}
    89  	shellReturns     struct {
    90  		result1 error
    91  	}
    92  	shellReturnsOnCall map[int]struct {
    93  		result1 error
    94  	}
    95  	WaitStub        func() error
    96  	waitMutex       sync.RWMutex
    97  	waitArgsForCall []struct{}
    98  	waitReturns     struct {
    99  		result1 error
   100  	}
   101  	waitReturnsOnCall map[int]struct {
   102  		result1 error
   103  	}
   104  	CloseStub        func() error
   105  	closeMutex       sync.RWMutex
   106  	closeArgsForCall []struct{}
   107  	closeReturns     struct {
   108  		result1 error
   109  	}
   110  	closeReturnsOnCall map[int]struct {
   111  		result1 error
   112  	}
   113  	invocations      map[string][][]interface{}
   114  	invocationsMutex sync.RWMutex
   115  }
   116  
   117  func (fake *FakeSecureSession) RequestPty(term string, height int, width int, termModes ssh.TerminalModes) error {
   118  	fake.requestPtyMutex.Lock()
   119  	ret, specificReturn := fake.requestPtyReturnsOnCall[len(fake.requestPtyArgsForCall)]
   120  	fake.requestPtyArgsForCall = append(fake.requestPtyArgsForCall, struct {
   121  		term      string
   122  		height    int
   123  		width     int
   124  		termModes ssh.TerminalModes
   125  	}{term, height, width, termModes})
   126  	fake.recordInvocation("RequestPty", []interface{}{term, height, width, termModes})
   127  	fake.requestPtyMutex.Unlock()
   128  	if fake.RequestPtyStub != nil {
   129  		return fake.RequestPtyStub(term, height, width, termModes)
   130  	}
   131  	if specificReturn {
   132  		return ret.result1
   133  	}
   134  	return fake.requestPtyReturns.result1
   135  }
   136  
   137  func (fake *FakeSecureSession) RequestPtyCallCount() int {
   138  	fake.requestPtyMutex.RLock()
   139  	defer fake.requestPtyMutex.RUnlock()
   140  	return len(fake.requestPtyArgsForCall)
   141  }
   142  
   143  func (fake *FakeSecureSession) RequestPtyArgsForCall(i int) (string, int, int, ssh.TerminalModes) {
   144  	fake.requestPtyMutex.RLock()
   145  	defer fake.requestPtyMutex.RUnlock()
   146  	return fake.requestPtyArgsForCall[i].term, fake.requestPtyArgsForCall[i].height, fake.requestPtyArgsForCall[i].width, fake.requestPtyArgsForCall[i].termModes
   147  }
   148  
   149  func (fake *FakeSecureSession) RequestPtyReturns(result1 error) {
   150  	fake.RequestPtyStub = nil
   151  	fake.requestPtyReturns = struct {
   152  		result1 error
   153  	}{result1}
   154  }
   155  
   156  func (fake *FakeSecureSession) RequestPtyReturnsOnCall(i int, result1 error) {
   157  	fake.RequestPtyStub = nil
   158  	if fake.requestPtyReturnsOnCall == nil {
   159  		fake.requestPtyReturnsOnCall = make(map[int]struct {
   160  			result1 error
   161  		})
   162  	}
   163  	fake.requestPtyReturnsOnCall[i] = struct {
   164  		result1 error
   165  	}{result1}
   166  }
   167  
   168  func (fake *FakeSecureSession) SendRequest(name string, wantReply bool, payload []byte) (bool, error) {
   169  	var payloadCopy []byte
   170  	if payload != nil {
   171  		payloadCopy = make([]byte, len(payload))
   172  		copy(payloadCopy, payload)
   173  	}
   174  	fake.sendRequestMutex.Lock()
   175  	ret, specificReturn := fake.sendRequestReturnsOnCall[len(fake.sendRequestArgsForCall)]
   176  	fake.sendRequestArgsForCall = append(fake.sendRequestArgsForCall, struct {
   177  		name      string
   178  		wantReply bool
   179  		payload   []byte
   180  	}{name, wantReply, payloadCopy})
   181  	fake.recordInvocation("SendRequest", []interface{}{name, wantReply, payloadCopy})
   182  	fake.sendRequestMutex.Unlock()
   183  	if fake.SendRequestStub != nil {
   184  		return fake.SendRequestStub(name, wantReply, payload)
   185  	}
   186  	if specificReturn {
   187  		return ret.result1, ret.result2
   188  	}
   189  	return fake.sendRequestReturns.result1, fake.sendRequestReturns.result2
   190  }
   191  
   192  func (fake *FakeSecureSession) SendRequestCallCount() int {
   193  	fake.sendRequestMutex.RLock()
   194  	defer fake.sendRequestMutex.RUnlock()
   195  	return len(fake.sendRequestArgsForCall)
   196  }
   197  
   198  func (fake *FakeSecureSession) SendRequestArgsForCall(i int) (string, bool, []byte) {
   199  	fake.sendRequestMutex.RLock()
   200  	defer fake.sendRequestMutex.RUnlock()
   201  	return fake.sendRequestArgsForCall[i].name, fake.sendRequestArgsForCall[i].wantReply, fake.sendRequestArgsForCall[i].payload
   202  }
   203  
   204  func (fake *FakeSecureSession) SendRequestReturns(result1 bool, result2 error) {
   205  	fake.SendRequestStub = nil
   206  	fake.sendRequestReturns = struct {
   207  		result1 bool
   208  		result2 error
   209  	}{result1, result2}
   210  }
   211  
   212  func (fake *FakeSecureSession) SendRequestReturnsOnCall(i int, result1 bool, result2 error) {
   213  	fake.SendRequestStub = nil
   214  	if fake.sendRequestReturnsOnCall == nil {
   215  		fake.sendRequestReturnsOnCall = make(map[int]struct {
   216  			result1 bool
   217  			result2 error
   218  		})
   219  	}
   220  	fake.sendRequestReturnsOnCall[i] = struct {
   221  		result1 bool
   222  		result2 error
   223  	}{result1, result2}
   224  }
   225  
   226  func (fake *FakeSecureSession) StdinPipe() (io.WriteCloser, error) {
   227  	fake.stdinPipeMutex.Lock()
   228  	ret, specificReturn := fake.stdinPipeReturnsOnCall[len(fake.stdinPipeArgsForCall)]
   229  	fake.stdinPipeArgsForCall = append(fake.stdinPipeArgsForCall, struct{}{})
   230  	fake.recordInvocation("StdinPipe", []interface{}{})
   231  	fake.stdinPipeMutex.Unlock()
   232  	if fake.StdinPipeStub != nil {
   233  		return fake.StdinPipeStub()
   234  	}
   235  	if specificReturn {
   236  		return ret.result1, ret.result2
   237  	}
   238  	return fake.stdinPipeReturns.result1, fake.stdinPipeReturns.result2
   239  }
   240  
   241  func (fake *FakeSecureSession) StdinPipeCallCount() int {
   242  	fake.stdinPipeMutex.RLock()
   243  	defer fake.stdinPipeMutex.RUnlock()
   244  	return len(fake.stdinPipeArgsForCall)
   245  }
   246  
   247  func (fake *FakeSecureSession) StdinPipeReturns(result1 io.WriteCloser, result2 error) {
   248  	fake.StdinPipeStub = nil
   249  	fake.stdinPipeReturns = struct {
   250  		result1 io.WriteCloser
   251  		result2 error
   252  	}{result1, result2}
   253  }
   254  
   255  func (fake *FakeSecureSession) StdinPipeReturnsOnCall(i int, result1 io.WriteCloser, result2 error) {
   256  	fake.StdinPipeStub = nil
   257  	if fake.stdinPipeReturnsOnCall == nil {
   258  		fake.stdinPipeReturnsOnCall = make(map[int]struct {
   259  			result1 io.WriteCloser
   260  			result2 error
   261  		})
   262  	}
   263  	fake.stdinPipeReturnsOnCall[i] = struct {
   264  		result1 io.WriteCloser
   265  		result2 error
   266  	}{result1, result2}
   267  }
   268  
   269  func (fake *FakeSecureSession) StdoutPipe() (io.Reader, error) {
   270  	fake.stdoutPipeMutex.Lock()
   271  	ret, specificReturn := fake.stdoutPipeReturnsOnCall[len(fake.stdoutPipeArgsForCall)]
   272  	fake.stdoutPipeArgsForCall = append(fake.stdoutPipeArgsForCall, struct{}{})
   273  	fake.recordInvocation("StdoutPipe", []interface{}{})
   274  	fake.stdoutPipeMutex.Unlock()
   275  	if fake.StdoutPipeStub != nil {
   276  		return fake.StdoutPipeStub()
   277  	}
   278  	if specificReturn {
   279  		return ret.result1, ret.result2
   280  	}
   281  	return fake.stdoutPipeReturns.result1, fake.stdoutPipeReturns.result2
   282  }
   283  
   284  func (fake *FakeSecureSession) StdoutPipeCallCount() int {
   285  	fake.stdoutPipeMutex.RLock()
   286  	defer fake.stdoutPipeMutex.RUnlock()
   287  	return len(fake.stdoutPipeArgsForCall)
   288  }
   289  
   290  func (fake *FakeSecureSession) StdoutPipeReturns(result1 io.Reader, result2 error) {
   291  	fake.StdoutPipeStub = nil
   292  	fake.stdoutPipeReturns = struct {
   293  		result1 io.Reader
   294  		result2 error
   295  	}{result1, result2}
   296  }
   297  
   298  func (fake *FakeSecureSession) StdoutPipeReturnsOnCall(i int, result1 io.Reader, result2 error) {
   299  	fake.StdoutPipeStub = nil
   300  	if fake.stdoutPipeReturnsOnCall == nil {
   301  		fake.stdoutPipeReturnsOnCall = make(map[int]struct {
   302  			result1 io.Reader
   303  			result2 error
   304  		})
   305  	}
   306  	fake.stdoutPipeReturnsOnCall[i] = struct {
   307  		result1 io.Reader
   308  		result2 error
   309  	}{result1, result2}
   310  }
   311  
   312  func (fake *FakeSecureSession) StderrPipe() (io.Reader, error) {
   313  	fake.stderrPipeMutex.Lock()
   314  	ret, specificReturn := fake.stderrPipeReturnsOnCall[len(fake.stderrPipeArgsForCall)]
   315  	fake.stderrPipeArgsForCall = append(fake.stderrPipeArgsForCall, struct{}{})
   316  	fake.recordInvocation("StderrPipe", []interface{}{})
   317  	fake.stderrPipeMutex.Unlock()
   318  	if fake.StderrPipeStub != nil {
   319  		return fake.StderrPipeStub()
   320  	}
   321  	if specificReturn {
   322  		return ret.result1, ret.result2
   323  	}
   324  	return fake.stderrPipeReturns.result1, fake.stderrPipeReturns.result2
   325  }
   326  
   327  func (fake *FakeSecureSession) StderrPipeCallCount() int {
   328  	fake.stderrPipeMutex.RLock()
   329  	defer fake.stderrPipeMutex.RUnlock()
   330  	return len(fake.stderrPipeArgsForCall)
   331  }
   332  
   333  func (fake *FakeSecureSession) StderrPipeReturns(result1 io.Reader, result2 error) {
   334  	fake.StderrPipeStub = nil
   335  	fake.stderrPipeReturns = struct {
   336  		result1 io.Reader
   337  		result2 error
   338  	}{result1, result2}
   339  }
   340  
   341  func (fake *FakeSecureSession) StderrPipeReturnsOnCall(i int, result1 io.Reader, result2 error) {
   342  	fake.StderrPipeStub = nil
   343  	if fake.stderrPipeReturnsOnCall == nil {
   344  		fake.stderrPipeReturnsOnCall = make(map[int]struct {
   345  			result1 io.Reader
   346  			result2 error
   347  		})
   348  	}
   349  	fake.stderrPipeReturnsOnCall[i] = struct {
   350  		result1 io.Reader
   351  		result2 error
   352  	}{result1, result2}
   353  }
   354  
   355  func (fake *FakeSecureSession) Start(command string) error {
   356  	fake.startMutex.Lock()
   357  	ret, specificReturn := fake.startReturnsOnCall[len(fake.startArgsForCall)]
   358  	fake.startArgsForCall = append(fake.startArgsForCall, struct {
   359  		command string
   360  	}{command})
   361  	fake.recordInvocation("Start", []interface{}{command})
   362  	fake.startMutex.Unlock()
   363  	if fake.StartStub != nil {
   364  		return fake.StartStub(command)
   365  	}
   366  	if specificReturn {
   367  		return ret.result1
   368  	}
   369  	return fake.startReturns.result1
   370  }
   371  
   372  func (fake *FakeSecureSession) StartCallCount() int {
   373  	fake.startMutex.RLock()
   374  	defer fake.startMutex.RUnlock()
   375  	return len(fake.startArgsForCall)
   376  }
   377  
   378  func (fake *FakeSecureSession) StartArgsForCall(i int) string {
   379  	fake.startMutex.RLock()
   380  	defer fake.startMutex.RUnlock()
   381  	return fake.startArgsForCall[i].command
   382  }
   383  
   384  func (fake *FakeSecureSession) StartReturns(result1 error) {
   385  	fake.StartStub = nil
   386  	fake.startReturns = struct {
   387  		result1 error
   388  	}{result1}
   389  }
   390  
   391  func (fake *FakeSecureSession) StartReturnsOnCall(i int, result1 error) {
   392  	fake.StartStub = nil
   393  	if fake.startReturnsOnCall == nil {
   394  		fake.startReturnsOnCall = make(map[int]struct {
   395  			result1 error
   396  		})
   397  	}
   398  	fake.startReturnsOnCall[i] = struct {
   399  		result1 error
   400  	}{result1}
   401  }
   402  
   403  func (fake *FakeSecureSession) Shell() error {
   404  	fake.shellMutex.Lock()
   405  	ret, specificReturn := fake.shellReturnsOnCall[len(fake.shellArgsForCall)]
   406  	fake.shellArgsForCall = append(fake.shellArgsForCall, struct{}{})
   407  	fake.recordInvocation("Shell", []interface{}{})
   408  	fake.shellMutex.Unlock()
   409  	if fake.ShellStub != nil {
   410  		return fake.ShellStub()
   411  	}
   412  	if specificReturn {
   413  		return ret.result1
   414  	}
   415  	return fake.shellReturns.result1
   416  }
   417  
   418  func (fake *FakeSecureSession) ShellCallCount() int {
   419  	fake.shellMutex.RLock()
   420  	defer fake.shellMutex.RUnlock()
   421  	return len(fake.shellArgsForCall)
   422  }
   423  
   424  func (fake *FakeSecureSession) ShellReturns(result1 error) {
   425  	fake.ShellStub = nil
   426  	fake.shellReturns = struct {
   427  		result1 error
   428  	}{result1}
   429  }
   430  
   431  func (fake *FakeSecureSession) ShellReturnsOnCall(i int, result1 error) {
   432  	fake.ShellStub = nil
   433  	if fake.shellReturnsOnCall == nil {
   434  		fake.shellReturnsOnCall = make(map[int]struct {
   435  			result1 error
   436  		})
   437  	}
   438  	fake.shellReturnsOnCall[i] = struct {
   439  		result1 error
   440  	}{result1}
   441  }
   442  
   443  func (fake *FakeSecureSession) Wait() error {
   444  	fake.waitMutex.Lock()
   445  	ret, specificReturn := fake.waitReturnsOnCall[len(fake.waitArgsForCall)]
   446  	fake.waitArgsForCall = append(fake.waitArgsForCall, struct{}{})
   447  	fake.recordInvocation("Wait", []interface{}{})
   448  	fake.waitMutex.Unlock()
   449  	if fake.WaitStub != nil {
   450  		return fake.WaitStub()
   451  	}
   452  	if specificReturn {
   453  		return ret.result1
   454  	}
   455  	return fake.waitReturns.result1
   456  }
   457  
   458  func (fake *FakeSecureSession) WaitCallCount() int {
   459  	fake.waitMutex.RLock()
   460  	defer fake.waitMutex.RUnlock()
   461  	return len(fake.waitArgsForCall)
   462  }
   463  
   464  func (fake *FakeSecureSession) WaitReturns(result1 error) {
   465  	fake.WaitStub = nil
   466  	fake.waitReturns = struct {
   467  		result1 error
   468  	}{result1}
   469  }
   470  
   471  func (fake *FakeSecureSession) WaitReturnsOnCall(i int, result1 error) {
   472  	fake.WaitStub = nil
   473  	if fake.waitReturnsOnCall == nil {
   474  		fake.waitReturnsOnCall = make(map[int]struct {
   475  			result1 error
   476  		})
   477  	}
   478  	fake.waitReturnsOnCall[i] = struct {
   479  		result1 error
   480  	}{result1}
   481  }
   482  
   483  func (fake *FakeSecureSession) Close() error {
   484  	fake.closeMutex.Lock()
   485  	ret, specificReturn := fake.closeReturnsOnCall[len(fake.closeArgsForCall)]
   486  	fake.closeArgsForCall = append(fake.closeArgsForCall, struct{}{})
   487  	fake.recordInvocation("Close", []interface{}{})
   488  	fake.closeMutex.Unlock()
   489  	if fake.CloseStub != nil {
   490  		return fake.CloseStub()
   491  	}
   492  	if specificReturn {
   493  		return ret.result1
   494  	}
   495  	return fake.closeReturns.result1
   496  }
   497  
   498  func (fake *FakeSecureSession) CloseCallCount() int {
   499  	fake.closeMutex.RLock()
   500  	defer fake.closeMutex.RUnlock()
   501  	return len(fake.closeArgsForCall)
   502  }
   503  
   504  func (fake *FakeSecureSession) CloseReturns(result1 error) {
   505  	fake.CloseStub = nil
   506  	fake.closeReturns = struct {
   507  		result1 error
   508  	}{result1}
   509  }
   510  
   511  func (fake *FakeSecureSession) CloseReturnsOnCall(i int, result1 error) {
   512  	fake.CloseStub = nil
   513  	if fake.closeReturnsOnCall == nil {
   514  		fake.closeReturnsOnCall = make(map[int]struct {
   515  			result1 error
   516  		})
   517  	}
   518  	fake.closeReturnsOnCall[i] = struct {
   519  		result1 error
   520  	}{result1}
   521  }
   522  
   523  func (fake *FakeSecureSession) Invocations() map[string][][]interface{} {
   524  	fake.invocationsMutex.RLock()
   525  	defer fake.invocationsMutex.RUnlock()
   526  	fake.requestPtyMutex.RLock()
   527  	defer fake.requestPtyMutex.RUnlock()
   528  	fake.sendRequestMutex.RLock()
   529  	defer fake.sendRequestMutex.RUnlock()
   530  	fake.stdinPipeMutex.RLock()
   531  	defer fake.stdinPipeMutex.RUnlock()
   532  	fake.stdoutPipeMutex.RLock()
   533  	defer fake.stdoutPipeMutex.RUnlock()
   534  	fake.stderrPipeMutex.RLock()
   535  	defer fake.stderrPipeMutex.RUnlock()
   536  	fake.startMutex.RLock()
   537  	defer fake.startMutex.RUnlock()
   538  	fake.shellMutex.RLock()
   539  	defer fake.shellMutex.RUnlock()
   540  	fake.waitMutex.RLock()
   541  	defer fake.waitMutex.RUnlock()
   542  	fake.closeMutex.RLock()
   543  	defer fake.closeMutex.RUnlock()
   544  	copiedInvocations := map[string][][]interface{}{}
   545  	for key, value := range fake.invocations {
   546  		copiedInvocations[key] = value
   547  	}
   548  	return copiedInvocations
   549  }
   550  
   551  func (fake *FakeSecureSession) recordInvocation(key string, args []interface{}) {
   552  	fake.invocationsMutex.Lock()
   553  	defer fake.invocationsMutex.Unlock()
   554  	if fake.invocations == nil {
   555  		fake.invocations = map[string][][]interface{}{}
   556  	}
   557  	if fake.invocations[key] == nil {
   558  		fake.invocations[key] = [][]interface{}{}
   559  	}
   560  	fake.invocations[key] = append(fake.invocations[key], args)
   561  }
   562  
   563  var _ clissh.SecureSession = new(FakeSecureSession)