| #!/usr/bin/env python |
| from __future__ import print_function |
| |
| import getpass |
| import sys |
| PY3 = (sys.version_info[0] >= 3) |
| if not PY3: |
| input = raw_input |
| |
| print("Mock SSH client for tests. Do not enter real security info.") |
| |
| pw = getpass.getpass('password:') |
| if pw != 's3cret': |
| print('Permission denied!') |
| sys.exit(1) |
| |
| prompt = "$" |
| while True: |
| cmd = input(prompt) |
| if cmd.startswith('PS1='): |
| prompt = eval(cmd[4:]).replace('\$', '$') |
| elif cmd == 'ping': |
| print('pong') |
| elif cmd.startswith('ls'): |
| print('file1.py', 'file2.html', sep='\t') |
| elif cmd == 'echo $?': |
| print(0) |
| elif cmd in ('exit', 'logout'): |
| break |
