ClickHouse/tests/queries/0_stateless/helpers/shell.py

35 lines
988 B
Python
Raw Normal View History

import os
import sys
import time
CURDIR = os.path.dirname(os.path.realpath(__file__))
sys.path.insert(0, os.path.join(CURDIR))
import uexpect
class shell(object):
def __init__(self, command=None, name='', log=None, prompt='[#\$] '):
if command is None:
2020-05-03 01:46:06 +00:00
command = ['/bin/bash', '--noediting']
self.prompt = prompt
self.client = uexpect.spawn(command)
self.client.eol('\r')
self.client.logger(log, prefix=name)
self.client.timeout(20)
self.client.expect(prompt, timeout=60)
def __enter__(self):
io = self.client.__enter__()
io.prompt = self.prompt
return io
def __exit__(self, type, value, traceback):
self.client.reader['kill_event'].set()
# send Ctrl-C
self.client.send('\x03', eol='')
time.sleep(0.3)
self.client.send('exit', eol='\r')
self.client.send('\x03', eol='')
return self.client.__exit__(type, value, traceback)