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)