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

     1  // This file was generated by counterfeiter
     2  package mocks
     3  
     4  import (
     5  	"io"
     6  	"sync"
     7  
     8  	"github.com/cloudfoundry-incubator/ltc/ssh/sshapi"
     9  	"golang.org/x/crypto/ssh"
    10  )
    11  
    12  type FakeSSHSession struct {
    13  	StdinPipeStub        func() (io.WriteCloser, error)
    14  	stdinPipeMutex       sync.RWMutex
    15  	stdinPipeArgsForCall []struct{}
    16  	stdinPipeReturns     struct {
    17  		result1 io.WriteCloser
    18  		result2 error
    19  	}
    20  	StdoutPipeStub        func() (io.Reader, error)
    21  	stdoutPipeMutex       sync.RWMutex
    22  	stdoutPipeArgsForCall []struct{}
    23  	stdoutPipeReturns     struct {
    24  		result1 io.Reader
    25  		result2 error
    26  	}
    27  	StderrPipeStub        func() (io.Reader, error)
    28  	stderrPipeMutex       sync.RWMutex
    29  	stderrPipeArgsForCall []struct{}
    30  	stderrPipeReturns     struct {
    31  		result1 io.Reader
    32  		result2 error
    33  	}
    34  	SendRequestStub        func(name string, wantReply bool, payload []byte) (bool, error)
    35  	sendRequestMutex       sync.RWMutex
    36  	sendRequestArgsForCall []struct {
    37  		name      string
    38  		wantReply bool
    39  		payload   []byte
    40  	}
    41  	sendRequestReturns struct {
    42  		result1 bool
    43  		result2 error
    44  	}
    45  	RequestPtyStub        func(term string, h, w int, termmodes ssh.TerminalModes) error
    46  	requestPtyMutex       sync.RWMutex
    47  	requestPtyArgsForCall []struct {
    48  		term      string
    49  		h         int
    50  		w         int
    51  		termmodes ssh.TerminalModes
    52  	}
    53  	requestPtyReturns struct {
    54  		result1 error
    55  	}
    56  	ShellStub        func() error
    57  	shellMutex       sync.RWMutex
    58  	shellArgsForCall []struct{}
    59  	shellReturns     struct {
    60  		result1 error
    61  	}
    62  	RunStub        func(string) error
    63  	runMutex       sync.RWMutex
    64  	runArgsForCall []struct {
    65  		arg1 string
    66  	}
    67  	runReturns struct {
    68  		result1 error
    69  	}
    70  	WaitStub        func() error
    71  	waitMutex       sync.RWMutex
    72  	waitArgsForCall []struct{}
    73  	waitReturns     struct {
    74  		result1 error
    75  	}
    76  	CloseStub        func() error
    77  	closeMutex       sync.RWMutex
    78  	closeArgsForCall []struct{}
    79  	closeReturns     struct {
    80  		result1 error
    81  	}
    82  }
    83  
    84  func (fake *FakeSSHSession) StdinPipe() (io.WriteCloser, error) {
    85  	fake.stdinPipeMutex.Lock()
    86  	fake.stdinPipeArgsForCall = append(fake.stdinPipeArgsForCall, struct{}{})
    87  	fake.stdinPipeMutex.Unlock()
    88  	if fake.StdinPipeStub != nil {
    89  		return fake.StdinPipeStub()
    90  	} else {
    91  		return fake.stdinPipeReturns.result1, fake.stdinPipeReturns.result2
    92  	}
    93  }
    94  
    95  func (fake *FakeSSHSession) StdinPipeCallCount() int {
    96  	fake.stdinPipeMutex.RLock()
    97  	defer fake.stdinPipeMutex.RUnlock()
    98  	return len(fake.stdinPipeArgsForCall)
    99  }
   100  
   101  func (fake *FakeSSHSession) StdinPipeReturns(result1 io.WriteCloser, result2 error) {
   102  	fake.StdinPipeStub = nil
   103  	fake.stdinPipeReturns = struct {
   104  		result1 io.WriteCloser
   105  		result2 error
   106  	}{result1, result2}
   107  }
   108  
   109  func (fake *FakeSSHSession) StdoutPipe() (io.Reader, error) {
   110  	fake.stdoutPipeMutex.Lock()
   111  	fake.stdoutPipeArgsForCall = append(fake.stdoutPipeArgsForCall, struct{}{})
   112  	fake.stdoutPipeMutex.Unlock()
   113  	if fake.StdoutPipeStub != nil {
   114  		return fake.StdoutPipeStub()
   115  	} else {
   116  		return fake.stdoutPipeReturns.result1, fake.stdoutPipeReturns.result2
   117  	}
   118  }
   119  
   120  func (fake *FakeSSHSession) StdoutPipeCallCount() int {
   121  	fake.stdoutPipeMutex.RLock()
   122  	defer fake.stdoutPipeMutex.RUnlock()
   123  	return len(fake.stdoutPipeArgsForCall)
   124  }
   125  
   126  func (fake *FakeSSHSession) StdoutPipeReturns(result1 io.Reader, result2 error) {
   127  	fake.StdoutPipeStub = nil
   128  	fake.stdoutPipeReturns = struct {
   129  		result1 io.Reader
   130  		result2 error
   131  	}{result1, result2}
   132  }
   133  
   134  func (fake *FakeSSHSession) StderrPipe() (io.Reader, error) {
   135  	fake.stderrPipeMutex.Lock()
   136  	fake.stderrPipeArgsForCall = append(fake.stderrPipeArgsForCall, struct{}{})
   137  	fake.stderrPipeMutex.Unlock()
   138  	if fake.StderrPipeStub != nil {
   139  		return fake.StderrPipeStub()
   140  	} else {
   141  		return fake.stderrPipeReturns.result1, fake.stderrPipeReturns.result2
   142  	}
   143  }
   144  
   145  func (fake *FakeSSHSession) StderrPipeCallCount() int {
   146  	fake.stderrPipeMutex.RLock()
   147  	defer fake.stderrPipeMutex.RUnlock()
   148  	return len(fake.stderrPipeArgsForCall)
   149  }
   150  
   151  func (fake *FakeSSHSession) StderrPipeReturns(result1 io.Reader, result2 error) {
   152  	fake.StderrPipeStub = nil
   153  	fake.stderrPipeReturns = struct {
   154  		result1 io.Reader
   155  		result2 error
   156  	}{result1, result2}
   157  }
   158  
   159  func (fake *FakeSSHSession) SendRequest(name string, wantReply bool, payload []byte) (bool, error) {
   160  	fake.sendRequestMutex.Lock()
   161  	fake.sendRequestArgsForCall = append(fake.sendRequestArgsForCall, struct {
   162  		name      string
   163  		wantReply bool
   164  		payload   []byte
   165  	}{name, wantReply, payload})
   166  	fake.sendRequestMutex.Unlock()
   167  	if fake.SendRequestStub != nil {
   168  		return fake.SendRequestStub(name, wantReply, payload)
   169  	} else {
   170  		return fake.sendRequestReturns.result1, fake.sendRequestReturns.result2
   171  	}
   172  }
   173  
   174  func (fake *FakeSSHSession) SendRequestCallCount() int {
   175  	fake.sendRequestMutex.RLock()
   176  	defer fake.sendRequestMutex.RUnlock()
   177  	return len(fake.sendRequestArgsForCall)
   178  }
   179  
   180  func (fake *FakeSSHSession) SendRequestArgsForCall(i int) (string, bool, []byte) {
   181  	fake.sendRequestMutex.RLock()
   182  	defer fake.sendRequestMutex.RUnlock()
   183  	return fake.sendRequestArgsForCall[i].name, fake.sendRequestArgsForCall[i].wantReply, fake.sendRequestArgsForCall[i].payload
   184  }
   185  
   186  func (fake *FakeSSHSession) SendRequestReturns(result1 bool, result2 error) {
   187  	fake.SendRequestStub = nil
   188  	fake.sendRequestReturns = struct {
   189  		result1 bool
   190  		result2 error
   191  	}{result1, result2}
   192  }
   193  
   194  func (fake *FakeSSHSession) RequestPty(term string, h int, w int, termmodes ssh.TerminalModes) error {
   195  	fake.requestPtyMutex.Lock()
   196  	fake.requestPtyArgsForCall = append(fake.requestPtyArgsForCall, struct {
   197  		term      string
   198  		h         int
   199  		w         int
   200  		termmodes ssh.TerminalModes
   201  	}{term, h, w, termmodes})
   202  	fake.requestPtyMutex.Unlock()
   203  	if fake.RequestPtyStub != nil {
   204  		return fake.RequestPtyStub(term, h, w, termmodes)
   205  	} else {
   206  		return fake.requestPtyReturns.result1
   207  	}
   208  }
   209  
   210  func (fake *FakeSSHSession) RequestPtyCallCount() int {
   211  	fake.requestPtyMutex.RLock()
   212  	defer fake.requestPtyMutex.RUnlock()
   213  	return len(fake.requestPtyArgsForCall)
   214  }
   215  
   216  func (fake *FakeSSHSession) RequestPtyArgsForCall(i int) (string, int, int, ssh.TerminalModes) {
   217  	fake.requestPtyMutex.RLock()
   218  	defer fake.requestPtyMutex.RUnlock()
   219  	return fake.requestPtyArgsForCall[i].term, fake.requestPtyArgsForCall[i].h, fake.requestPtyArgsForCall[i].w, fake.requestPtyArgsForCall[i].termmodes
   220  }
   221  
   222  func (fake *FakeSSHSession) RequestPtyReturns(result1 error) {
   223  	fake.RequestPtyStub = nil
   224  	fake.requestPtyReturns = struct {
   225  		result1 error
   226  	}{result1}
   227  }
   228  
   229  func (fake *FakeSSHSession) Shell() error {
   230  	fake.shellMutex.Lock()
   231  	fake.shellArgsForCall = append(fake.shellArgsForCall, struct{}{})
   232  	fake.shellMutex.Unlock()
   233  	if fake.ShellStub != nil {
   234  		return fake.ShellStub()
   235  	} else {
   236  		return fake.shellReturns.result1
   237  	}
   238  }
   239  
   240  func (fake *FakeSSHSession) ShellCallCount() int {
   241  	fake.shellMutex.RLock()
   242  	defer fake.shellMutex.RUnlock()
   243  	return len(fake.shellArgsForCall)
   244  }
   245  
   246  func (fake *FakeSSHSession) ShellReturns(result1 error) {
   247  	fake.ShellStub = nil
   248  	fake.shellReturns = struct {
   249  		result1 error
   250  	}{result1}
   251  }
   252  
   253  func (fake *FakeSSHSession) Run(arg1 string) error {
   254  	fake.runMutex.Lock()
   255  	fake.runArgsForCall = append(fake.runArgsForCall, struct {
   256  		arg1 string
   257  	}{arg1})
   258  	fake.runMutex.Unlock()
   259  	if fake.RunStub != nil {
   260  		return fake.RunStub(arg1)
   261  	} else {
   262  		return fake.runReturns.result1
   263  	}
   264  }
   265  
   266  func (fake *FakeSSHSession) RunCallCount() int {
   267  	fake.runMutex.RLock()
   268  	defer fake.runMutex.RUnlock()
   269  	return len(fake.runArgsForCall)
   270  }
   271  
   272  func (fake *FakeSSHSession) RunArgsForCall(i int) string {
   273  	fake.runMutex.RLock()
   274  	defer fake.runMutex.RUnlock()
   275  	return fake.runArgsForCall[i].arg1
   276  }
   277  
   278  func (fake *FakeSSHSession) RunReturns(result1 error) {
   279  	fake.RunStub = nil
   280  	fake.runReturns = struct {
   281  		result1 error
   282  	}{result1}
   283  }
   284  
   285  func (fake *FakeSSHSession) Wait() error {
   286  	fake.waitMutex.Lock()
   287  	fake.waitArgsForCall = append(fake.waitArgsForCall, struct{}{})
   288  	fake.waitMutex.Unlock()
   289  	if fake.WaitStub != nil {
   290  		return fake.WaitStub()
   291  	} else {
   292  		return fake.waitReturns.result1
   293  	}
   294  }
   295  
   296  func (fake *FakeSSHSession) WaitCallCount() int {
   297  	fake.waitMutex.RLock()
   298  	defer fake.waitMutex.RUnlock()
   299  	return len(fake.waitArgsForCall)
   300  }
   301  
   302  func (fake *FakeSSHSession) WaitReturns(result1 error) {
   303  	fake.WaitStub = nil
   304  	fake.waitReturns = struct {
   305  		result1 error
   306  	}{result1}
   307  }
   308  
   309  func (fake *FakeSSHSession) Close() error {
   310  	fake.closeMutex.Lock()
   311  	fake.closeArgsForCall = append(fake.closeArgsForCall, struct{}{})
   312  	fake.closeMutex.Unlock()
   313  	if fake.CloseStub != nil {
   314  		return fake.CloseStub()
   315  	} else {
   316  		return fake.closeReturns.result1
   317  	}
   318  }
   319  
   320  func (fake *FakeSSHSession) CloseCallCount() int {
   321  	fake.closeMutex.RLock()
   322  	defer fake.closeMutex.RUnlock()
   323  	return len(fake.closeArgsForCall)
   324  }
   325  
   326  func (fake *FakeSSHSession) CloseReturns(result1 error) {
   327  	fake.CloseStub = nil
   328  	fake.closeReturns = struct {
   329  		result1 error
   330  	}{result1}
   331  }
   332  
   333  var _ sshapi.SSHSession = new(FakeSSHSession)