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)