def _spawn_corenlp(self):
if VERBOSE:
print self.start_corenlp
if use_winpexpect:
self.corenlp = winpexpect.winspawn(self.start_corenlp, maxread=8192,
searchwindowsize=80)
else:
self.corenlp = pexpect.spawn(self.start_corenlp, maxread=8192,
searchwindowsize=80)
# show progress bar while loading the models
if VERBOSE:
widgets = ['Loading Models: ', Fraction()]
# Model timeouts:
# pos tagger model (~5sec)
# NER-all classifier (~33sec)
# NER-muc classifier (~60sec)
# CoNLL classifier (~50sec)
# PCFG (~3sec)
timeouts = [20, 200, 600, 600, 20]
for i in xrange(5):
self.corenlp.expect("done.", timeout=timeouts[i]) # Load model
pbar.update(i + 1)
self.corenlp.expect("Entering interactive shell.")
pbar.finish()
# interactive shell
self.corenlp.expect("\nNLP> ")
def test_expect(self):
ps = winspawn('powershell.exe -command -')
ps.sendline('Get-Location')
ps.expect('---\s+\r\n')
cwd = ps.readline().strip()
assert os.getcwd() == cwd
ps.terminate()
def _spawn_corenlp(self):
if VERBOSE:
print self.start_corenlp
if use_winpexpect:
self.corenlp = winpexpect.winspawn(self.start_corenlp, maxread=8192,
searchwindowsize=80)
else:
self.corenlp = pexpect.spawn(self.start_corenlp, maxread=8192,
searchwindowsize=80, timeout=120)
# show progress bar while loading the models
if VERBOSE:
pass
# widgets = ['Loading Models: ', Fraction()]
# pbar = ProgressBar(widgets=widgets, maxval=5, force_update=True).start()
# Model timeouts:
# pos tagger model (~5sec)
# NER-all classifier (~33sec)
# NER-muc classifier (~60sec)
# CoNLL classifier (~50sec)
# PCFG (~3sec)
# timeouts = [20, 200, 600, 600, 20]
# for i in xrange(5):
# self.corenlp.expect("done.", timeout=timeouts[i]) # Load model
# # pbar.update(i + 1)
# self.corenlp.expect("Entering interactive shell.")
# pbar.finish()
# interactive shell
self.corenlp.expect("\nNLP> ")
print "Done"
def spawn(self, *cmd):
try:
if settings.host_os == 'windows':
self.__ui_str = 'Windows host CMD'
self.__dut = winspawn(*cmd)
elif settings.host_os == 'linux':
self.__ui_str = 'Linux host Bash shell'
self.__dut = pexpect.spawn(*cmd, encoding='utf-8')
cmd = ''.join(*cmd)
UI.log('ACTION', 'Executing command on ' + self.__ui_str,
'Command: ' + cmd)
UI.logTitle('TERMINAL OUTPUT from ' + self.__ui_str)
UI.last_ui = str(self)
last_line = self.__last_buff.split('\n')[-1]
UI.log_stream.write(last_line)
if settings.glb.print_to_stdout:
sys.stdout.write(last_line)
UI.end_msg = False
except Exception as e:
if not settings.glb.print_to_stdout:
sys.stdout.write(
'CRITICAL ALERT: Host spawn session failed to initialize')
sys.stdout.write(str(e))
UI.log('CRITICAL ALERT', 'Host spawn session failed to initialize',
str(e))
self.__dut = None
def test_wait(self):
ps = winspawn('powershell.exe -command -')
ps.sendline('Sleep 2; Exit')
ps.wait()
assert not ps.isalive()
assert ps.exitstatus == 0
ps.terminate()
def test_telnet():
child = winspawn("C:/users/jason.li/Downloads/apps/plink.exe -telnet 127.0.0.1 -P 23")
assert child.isalive()
child.logfile = sys.stdout
# wait until the prompt
index = child.expect(["[A-Z]+>", TIMEOUT, EOF], timeout = 10)
if index != 0:
child.terminate()
exit()
print "The return index is %d" % index
print "child process is alive ?", child.isalive()
child.sendline("znspace \"VISTA\"\r\n")
index = child.expect(["VISTA>", TIMEOUT, EOF], timeout = 10)
if index != 0:
child.terminate()
exit()
child.sendline("D ^XINDEX")
child.expect("All Routines?")
child.sendline("NO")
child.expect("Routine:")
child.sendline()
child.expect("Select BUILD NAME")
child.sendline()
child.expect("Select PACKAGE NAME")
child.sendline()
index = child.expect(["VISTA", TIMEOUT, EOF], timeout=10)
print index
child.sendline("HALT")
child.terminate()
def test_expect(self):
ps = winspawn('powershell.exe -command -')
ps.sendline('Get-Location')
ps.expect('---\s+\r\n')
cwd = ps.readline().strip()
assert os.getcwd() == cwd
ps.terminate()
def test_wait(self):
ps = winspawn('powershell.exe -command -')
ps.sendline('Sleep 2; Exit')
ps.wait()
assert not ps.isalive()
assert ps.exitstatus == 0
ps.terminate()
def test_terminate(self):
ps = winspawn('powershell.exe -command -')
assert ps.isalive()
ps.terminate()
assert not ps.isalive()
assert ps.exitstatus != 0
ps.terminate()
def test_terminate(self):
ps = winspawn('powershell.exe -command -')
assert ps.isalive()
ps.terminate()
assert not ps.isalive()
assert ps.exitstatus != 0
ps.terminate()
def test_dircommand():
logFile = StringIO.StringIO()
child = winspawn("dir", logfile=logFile)
index = child.expect([">>>", TIMEOUT, EOF])
print index
print logFile.getvalue()
child.terminate()
def test_python():
logFile = StringIO.StringIO()
child = winspawn("c:\Python27\python.exe", logfile=logFile)
index = child.expect([">>>", TIMEOUT, EOF])
print index
print logFile.getvalue()
child.terminate()
def createConnection(self, command, username = None, password = None):
if not command:
command = self.DEFAULT_WIN_TELNET_CMD
self._connection = winspawn(command, timeout = DEFAULT_TIME_OUT_VALUE)
assert self._connection.isalive()
if username and password:
self.__signIn__(username, password)
self.__changeNamesapce__()
def initAccess():
global deviceRadio
deviceRadio = []
CRValueName = [
'STA CH',
'STA Signal',
'STA Rx',
'STA Tx',
'STA BSSID',
]
print('{}-ing to {}'.format(deviceAccess, deviceIP))
if deviceAccess == 'ssh':
child = winpexpect.winspawn('plink -{} {}@{} -pw {}'.format(
deviceAccess, deviceUsername, deviceIP, devicePassword))
try:
child.expect('>')
# sleep(1)
except:
print(
"Timeout reached: First time ssh-ing into the device/device is offline"
)
#child.expect('(y/n)', timeout=3)
child.sendline('y')
child.expect('#|$|>', timeout=10)
elif deviceAccess == 'telnet':
child = winpexpect.winspawn('plink -{} {}'.format(
deviceAccess, deviceIP))
sleep(1)
child.expect('ogin:', timeout=10)
child.send('{}\r'.format(deviceUsername))
#Need to add another line for case of password
child.expect('Password:'******'{}\r'.format(devicePassword))
#child.expect('>')
#child.sendline('magic')
#child.expect('MAGIC>')
#child.sendline('!')
#child.sendline('sh')
child.expect('#')
else:
child = ''
print(
'ERROR: Please write deviceAccess label in lower case or appropriately'
)
findStationMAC(child)
return child
def executeScript(inputScript, scriptOptionalArgs, inputFile=None):
global testcaseInfoList
global testcaseInfo
global yamlCont
testcaseInfoList = []
yamlCont = {}
if inputFile != None:
fil = open(inputFile, "r")
yamlCont = yaml.load(fil)
# inputFileType = re.search('(.*).yaml', inputFile).group(1)
inputFileType = 'unknown'
if inputFile != None:
inputFileType = re.search('(.*).yaml', inputFile).group(1)
cmd = "%s %s" % (inputScript, ' '.join(scriptOptionalArgs))
#inputFileType = 'unknown'
print "@@@@@@@@@@@@@@@"
print cmd
print "@@@@@@@@@@@@@@@"
ps = winpexpect.winspawn(cmd)
ps.logfile_read = sys.stdout
## Initialize testcaseInfo if the script is not a flist.
if inputFileType in ['unknown']:
scriptName = inputScript
testcaseInfo = testInfo(scriptName)
testcaseInfoList.append(testcaseInfo)
timeout = -1
ps.interact(output_filter=scriptExecutionMonitor)
if inputFileType is 'unknown':
## creates a nested suite
suiteList = []
for testcaseInfo in testcaseInfoList:
testcaseInfo.flush()
suite = createRobotSuite(testcaseInfo)
suiteList.append(suite)
suite = TestSuite(inputScript)
suite.suites = suiteList
result = suite.run(output='output.xml', loglevel='debug')
for testcaseInfo in testcaseInfoList:
testcaseInfo.variableFile.close()
else:
## creates a single suite
for testcaseInfo in testcaseInfoList:
testcaseInfo.flush()
print testcaseInfo
suite = createRobotSuite(testcaseInfo)
result = suite.run(output='output.xml', loglevel='debug')
testcaseInfo.variableFile.close()
# Generating log files requires processing the earlier generated output XML.
ResultWriter('output.xml').write_results()
pp = postProcess.postProcess(suiteFile=testcaseInfo.scriptName)
pp.close()
def initAccess():
global deviceRadio
deviceRadio = []
print('{}-ing to {}'.format(deviceAccess, deviceIP))
if deviceAccess == 'ssh':
child = winpexpect.winspawn('plink -{} {}@{} -pw {}'.format(
deviceAccess, deviceUsername, deviceIP, devicePassword))
try:
child.expect('$')
#print('got ssh login')
except:
print(
"Timeout reached: First time ssh-ing into the device/device is offline"
)
#child.expect('(y/n)', timeout=3)
child.sendline('y')
child.expect('#|$', timeout=10)
elif deviceAccess == 'telnet':
child = winpexpect.winspawn('plink -{} {}'.format(
deviceAccess, deviceIP))
sleep(1)
child.expect('ogin:', timeout=10)
child.send('{}\r'.format(deviceUsername))
#Need to add another line for case of password
child.expect('Password:'******'{}\r'.format(devicePassword))
#child.expect('>')
#child.sendline('magic')
#child.expect('MAGIC>')
#child.sendline('!')
#child.sendline('sh')
child.expect('#')
elif deviceAccess == 'adb':
child = winpexpect.winspawn('adb connect {}'.format(deviceIP))
child.expect('connected to {}:5555'.format(deviceIP))
print('adb connected')
else:
child = ''
print(
'ERROR: Please write deviceAccess label in lower case or appropriately'
)
#findStationMAC(child)
setHeaders(child)
return child
def executeScript(inputScript, scriptOptionalArgs, inputFile=None):
global testcaseInfoList
global testcaseInfo
global yamlCont
testcaseInfoList = []
yamlCont = {}
if inputFile != None:
fil = open(inputFile, "r")
yamlCont = yaml.load(fil)
# inputFileType = re.search('(.*).yaml', inputFile).group(1)
inputFileType = 'unknown'
if inputFile != None:
inputFileType = re.search('(.*).yaml', inputFile).group(1)
cmd = "%s %s" %(inputScript, ' '.join(scriptOptionalArgs))
#inputFileType = 'unknown'
print "@@@@@@@@@@@@@@@"
print cmd
print "@@@@@@@@@@@@@@@"
ps = winpexpect.winspawn(cmd)
ps.logfile_read = sys.stdout
## Initialize testcaseInfo if the script is not a flist.
if inputFileType in ['unknown']:
scriptName = inputScript
testcaseInfo = testInfo(scriptName)
testcaseInfoList.append(testcaseInfo)
timeout = -1
ps.interact(output_filter=scriptExecutionMonitor)
if inputFileType is 'unknown':
## creates a nested suite
suiteList = []
for testcaseInfo in testcaseInfoList:
testcaseInfo.flush()
suite = createRobotSuite(testcaseInfo)
suiteList.append(suite)
suite = TestSuite(inputScript)
suite.suites = suiteList
result = suite.run(output='output.xml', loglevel='debug')
for testcaseInfo in testcaseInfoList:
testcaseInfo.variableFile.close()
else:
## creates a single suite
for testcaseInfo in testcaseInfoList:
testcaseInfo.flush()
print testcaseInfo
suite = createRobotSuite(testcaseInfo)
result = suite.run(output='output.xml', loglevel='debug')
testcaseInfo.variableFile.close()
# Generating log files requires processing the earlier generated output XML.
ResultWriter('output.xml').write_results()
pp = postProcess.postProcess(suiteFile=testcaseInfo.scriptName)
pp.close()
def createConnection(self, command, instance,
username = None, password = None):
if not command:
command = self.DEFAULT_WIN_TELNET_CMD
self._instance = instance
self._connection = winspawn(command, timeout = DEFAULT_TIME_OUT_VALUE)
assert self._connection.isalive()
if username and password:
self.__signIn__(username, password)
self.__changeNamesapce__()
def __inittelnet__(self, ip, port, user, pwd, prompt):
if port != '':
self.__ui_str = 'Telnet ' + ip + ' on port ' + port
else:
self.__ui_str = 'Telnet ' + ip
UI.log('ACTION', 'Initializing ' + self.__ui_str)
try:
if settings.host_os == 'windows':
if port != '':
spw = winspawn('bin_win32/plink -telnet ' + ip + ' -P ' +
port)
else:
spw = winspawn('bin_win32/plink -telnet ' + ip)
#spw = winspawn('bin_win32/telnet_win.exe '+ip)
else:
spw = pexpect.spawn('telnet ' + ip + ' ' + port,
encoding='utf-8')
except Exception as e:
if not settings.glb.print_to_stdout:
sys.stdout.write(
'CRITICAL ALERT: Telnet spawn session failed to initialize'
)
sys.stdout.write(
'Please check Telnet login credentials, and verify the IP address is configured '
+ 'properly.')
sys.stdout.write(e)
UI.log(
'CRITICAL ALERT', 'Telnet spawn session failed to initialize',
'Please check Telnet login credentials, and verify the IP address is configured properly.',
str(e))
return 'FAIL'
self.__dut = spw
self.__prompt = prompt
return self.__login__(user, pwd)
def __initssh__(self, ip, port, user, pwd, prompt):
if port != '':
self.__ui_str = 'SSH ' + user + '@' + ip + ' on port ' + port
else:
self.__ui_str = 'SSH ' + user + '@' + ip
UI.log('ACTION', 'Initializing ' + self.__ui_str)
try:
if settings.host_os == 'windows':
if port != '':
spw = winspawn('bin_win32/plink -ssh ' + user + '@' + ip +
' -P ' + port)
else:
spw = winspawn('bin_win32/plink -ssh ' + user + '@' + ip)
else:
spw = pexpect.spawn(
'ssh ' + user + '@' + ip + ' -p ' + port +
' -o StrictHostKeyChecking=no -o ServerAliveInterval=60',
encoding='utf-8')
except Exception as e:
if not settings.glb.print_to_stdout:
sys.stdout.write(
'CRITICAL ALERT: SSH spawn session failed to initialize')
sys.stdout.write(
'Please check SSH login credentials, and verify the IP address is configured '
+ 'properly.')
sys.stdout.write(e)
UI.log(
'CRITICAL ALERT', 'SSH spawn session failed to initialize',
'Please check SSH login credentials, and verify the IP address is configured properly.',
str(e))
return 'FAIL'
self.__dut = spw
self.__prompt = prompt
return self.__login__(user, pwd)
def __init__(self,
address,
ttType,
action="pos",
enable="1",
pos_target="0",
vel_target="5000"):
if ttType == 'octoscope':
while True:
try:
self.address = address
self.data = requests.get("http://{}/api/turntable".format(
self.address))
self.action = action
self.enable = enable
self.pos_target = pos_target
self.vel_target = vel_target
self.ttType = ttType
break
except:
print(
'Octoscope: A problem with establishing connection to turntable {}. Retrying again in 1 second.'
.format(self.address))
sleep(1)
elif ttType == 'dot11':
self.ttType = ttType
self.address = address
self.vel_target = 'medium2'
self.fileName = 'turntableControl'
self.position = 0
import platform
if platform.system() == 'Windows':
import winpexpect
self.child = winpexpect.winspawn(
'plink -{} {}@{} -pw {}'.format('ssh', 'pi', self.address,
'raspberry'))
else:
self.child = pexpect.spawn('{} {}@{} -pw {}'.format(
'ssh', 'pi', self.address, 'raspberry'))
try:
self.child.expect('$')
except:
print(
"Timeout reached: First time ssh-ing to raspberry/raspberry is offline"
)
#child.pexpect('(y/n)', timeout=3)
self.child.sendline('y')
self.child.expect('#|$', timeout=10)
else:
print(
"Please enter the proper turntable type (ttType) to either dot11 or octoscope"
)
def createConnection(self, command, instance,
username = None, password = None,
hostname=DEFAULT_HOST, port=DEFAULT_PORT):
if not command:
command = self.DEFAULT_WIN_TELNET_CMD
if (hostname and port):
command += " %s -P %s" % (hostname, port)
self._instance = instance
self._connection = winspawn(command, timeout = DEFAULT_TIME_OUT_VALUE)
assert self._connection.isalive()
if username and password:
self.__signIn__(username, password)
self.__changeNamesapce__()
def test_spawn_with_cwd(self):
# Use a directory that is not our cwd
cwd = os.getcwd()
homedir = os.path.expanduser('~')
if cwd != homedir:
dir = homedir
else:
dir, tail = os.path.splitdrive(homedir)
ps = winspawn('powershell.exe -command -', cwd=dir)
ps.sendline('Get-Location')
ps.expect('---\s+\r\n')
cwd = ps.readline().strip()
assert cwd == dir
ps.terminate()
def test_spawn_with_cwd(self):
# Use a directory that is not our cwd
cwd = os.getcwd()
homedir = os.path.expanduser('~')
if cwd != homedir:
dir = homedir
else:
dir, tail = os.path.splitdrive(homedir)
ps = winspawn('powershell.exe -command -', cwd=dir)
ps.sendline('Get-Location')
ps.expect('---\s+\r\n')
cwd = ps.readline().strip()
assert cwd == dir
ps.terminate()
def start(self, **args):
# On Windows 2008 R2, which is always 64-bit, the PowerShell bindings
# are only installed in the 32-bit "Windows on Windows" environment.
syswow = r'C:\Windows\SysWOW64'
try:
st = os.stat(syswow)
except OSError:
st = None
if st and stat.S_ISDIR(st.st_mode):
newpath = os.path.join(syswow, 'WindowsPowerShell', 'v1.0')
newpath += os.pathsep
if newpath not in os.environ['Path']:
newpath += os.environ['Path']
os.environ['Path'] = newpath
self.child = winspawn('powershell.exe -Command -', **args)
def start(self, **args):
# On Windows 2008 R2, which is always 64-bit, the PowerShell bindings
# are only installed in the 32-bit "Windows on Windows" environment.
syswow = r'C:\Windows\SysWOW64'
try:
st = os.stat(syswow)
except OSError:
st = None
if st and stat.S_ISDIR(st.st_mode):
newpath = os.path.join(syswow, 'WindowsPowerShell', 'v1.0')
newpath += os.pathsep
if newpath not in os.environ['Path']:
newpath += os.environ['Path']
os.environ['Path'] = newpath
self.child = winspawn('powershell.exe -Command -', **args)
def __init__(self, jar_path, grammar_path):
# Check input
if not jar_path.endswith('.jar') or not os.path.isfile(jar_path):
raise Exception("Invalid jar file")
if not grammar_path.endswith('.gr') or \
not os.path.isfile(grammar_path):
raise Exception("Invalid grammar file")
cmd = 'java -jar %s -gr %s' % (jar_path, grammar_path)
if os.name == 'nt':
self.parser = winpexpect.winspawn(cmd)
else:
self.parser = pexpect.spawn(cmd)
# Pass in a dumb sentence to fully initialize
tmp = self.parse('')
def connect(user, host, password):
ssh_newkey = 'Are you sure you want to continue connecting'
connStr = 'ssh ' + user + '@' + host
child = winpexpect.winspawn(connStr)
ret = child.expect([winpexpect.TIMEOUT, ssh_newkey, '[P|p]assword:'])
if ret == 0:
print('[-] Error Connecting')
return
if ret == 1:
child.sendline('yes')
ret = child.expect([winpexpect.TIMEOUT, '[P|p]assword:'])
if ret == 0:
print('[-] Error Connecting')
return
child.sendline(password)
child.expect(PROMPT)
return child
def __init__(self, jar_path, grammar_path):
"""Specify the path to the parser jar file
and the grammar file.
"""
# Check input
if not jar_path.endswith('.jar') or not os.path.isfile(jar_path):
raise Exception("Invalid jar file")
if not grammar_path.endswith('.gr') or \
not os.path.isfile(grammar_path):
raise Exception("Invalid grammar file")
cmd = 'java -jar %s -gr %s' % (jar_path, grammar_path)
if os.name == 'nt':
self.parser = winpexpect.winspawn(cmd)
else:
self.parser = pexpect.spawn(cmd)
# Pass in a dumb sentence to fully initialize
tmp = self.parse('')
def ssh_login(user, NE_IP):
host_user = '******' % (user, NE_IP)
child = winpexpect.winspawn('ssh', [
'-t', '-t', '-o StrictHostKeyChecking=no',
'-o UserKnownHostsFile=d:/null', host_user
])
os.remove('D:\\null')
index = child.expect(
['(?i)~#', 'connect timed out', winpexpect.EOF, winpexpect.TIMEOUT])
if (index != 0):
print 'Connect to %s failed' % NE_IP
child.terminate(force=True)
child.sendline('lsh')
#"(?i)>" 期望具有>çš„å—符出现,就返回Index=0
child.expect(['(?i)>'])
#child.expect('root@mcp1800>')
child.sendline('configure')
#child.expect('root@mcp1800#')
child.sendline('q')
print 'login %s NE succeeded' % NE_IP
return child
def createPackets(self, fileName):
'''
Create PCAP packets. delegate work to PCAP Services
Args:
filename : path to pcap file
'''
# projectPath = " PCAP/PCAPServices.py"
# if platform.system() == 'Darwin':
try:
varSize = os.path.getsize(fileName)
except Exception as e:
print("File doesnt exist! ")
return
#pcap file too big to handle
if (varSize >= 50e7):
print("File size is huge,(" + str(varSize / 1e6) +
" MB) want to proceed?[Y/N]")
answer = input()
if (not answer) or (answer.lower()[0] != 'y'):
return
# projectPath = "src/main/python/Backend/PCAP/PCAPServices.py"
projectPath = "{}/{}.py".format(os.getcwd(), "PCAPServices")
print(projectPath)
#spawn a new process using the right library for the OS
if platform.system() == "Windows":
self.child = winpexpect.winspawn("python " + '"' + projectPath +
'"')
print("created - w ")
else:
self.child = pexpect.spawn("python3.6 " + projectPath,
encoding='utf-8')
print("created")
self.child.expect("loop")
print("Creating")
print(fileName + " file name")
#Send signal to pcapservices
self.child.sendline("create " + fileName)
self.child.expect("Done")
def run_recorder(expInfo):
# parse in currect time and date
currentTime = time.strftime('%m-%d-%y_%H-%M')
# parse in participant ID
subjectID = expInfo['Participant ID']
# parse in experiment mode between "train" and "neurofeedback"
expMode = expInfo['Experiment Mode']
# path to LabRecorder
LabrecorderPath = os.path.join(base_dir_init(), 'lib', 'LabRecorder', 'LabRecorderCLI.exe')
# path to store lsl streams
storePath = os.path.join(osuser_path_init(), subjectID)
# run LabRecorder
recorder_process = winpexpect.winspawn(
'%s %s\subject_%s_%s_%s.xdf \'name="Keyboard"\'' % (LabrecorderPath, storePath,
subjectID, expMode, currentTime))
return recorder_process
def test_multiple_close(self):
ps = winspawn('powershell.exe -command -')
ps.close()
ps.close()
def test_wait_after_close(self):
ps = winspawn('powershell.exe -command -')
ps.close() # closes stdin and should therefore exit powershell
ps.wait()
assert not ps.isalive()
def test_multiple_terminate(self):
ps = winspawn('powershell.exe -command -')
ps.terminate()
ps.terminate()
import sys
timeout = 4200
baseMsg = 'plplplplplp'
messages = 5
setup = None
peer1 = None
peer2 = None
peers = int(sys.argv[1])
try:
start = time.time()
setup = winpexpect.winspawn('py -u setup.py -peers ' + str(peers), timeout=420)
print('------------')
print(str(peers) + ' peers')
setup.expect('>')
end = time.time()
print('setup.py done')
print('time: '+str(end-start))
peer1 = winpexpect.winspawn('py -u interactive_peer.py localhost 8500 peer1 10', timeout=timeout)
peer2 = winpexpect.winspawn('py -u interactive_peer.py localhost 8501 peer2 10', timeout=timeout)
#setup peer1
peer1.expect('>')
peer1.sendline('secret private.pem')
peer1.expect('>')
peer1.sendline('hello')
peer1.expect('>')
def test_wait_timeout(self):
ps = winspawn('powershell.exe -command -')
assert_raises(TIMEOUT, ps.wait, timeout=2)
ps.terminate()
#!/usr/bin/python3
# you must install pexpect for this to work:
# sudo apt-get install python3-pip
# sudo pip3 install pexpect
import time
import winpexpect
import re
import sys
timeout = 900
msg = 'plplplplplp'
try:
print('start: {:.2f}'.format(time.time()))
setup = winpexpect.winspawn('py setup.py -peers 149 -late 0.1', timeout=30)
setup.logfile = sys.stdout
setup.expect('>')
peer = winpexpect.winspawn('py interactive_peer.py localhost 8500 peer1 5', timeout=timeout)
#setup peer1
peer.expect('>')
peer.sendline('hello')
peer.expect('>')
time.sleep(120.0)
peer.sendline('dhcard-all')
peer.expect('Average Diffie-Hellman connection pool cardinality in this network: \d+.\d+\r\n')
avgdhcard = re.compile('\d+.\d+').search(str(peer.after)).group()
totalTime = '{:.2f}'.format(time.time())
print(totalTime + ' ' + avgdhcard)
setup.terminate()
peer.terminate()
channel_format='string', source_id='Keyboard')
# Initialize the keyboard stream.
outlet = StreamOutlet(info)
currentTime = expInfo['date']
subjectID = expInfo['participant']
Labrecorder = '.\lib\LabRecorder\LabRecorderCLI.exe'
Dataset='.\Dataset'
# Open LabRecorder
import subprocess, sys, os, winpexpect, time
#os.system("start /B start cmd.exe @cmd /k .\lib\LabRecorder\LabRecorderCLI.exe .\Dataset\'currentTime'.xdf 'type=EEG'")
#os.system("start /B start cmd.exe @cmd /k %s %s\%s_%s 'type=EEG'" % (Labrecorder, Dataset, currentTime,subjectID))
child = winpexpect.winspawn('%s %s\%s_%s \'name="Keyboard"\'' % (Labrecorder,Dataset,currentTime,subjectID))
# Fix random seed
random.seed()
# Read stimuli excel file
infile = './listStim.xlsx'
# Number of stimuli and number of blocks
num_trials = 10
num_blocks = 8
def test_expect():
ps = winpexpect.winspawn('ping')
ps.sendline('exit')
print ps.before.decode('gbk')
ps.terminate()
wrong_expert_pass = '******'
enter_expert_pass = '******'
sk1 = '\n\n\n---------------------------------------------\n'
def skip():
px.send('\n')
px.expect(']# ')
px.send('\n')
px.expect(']# ')
px.send('\n')
px.expect(']# ')
for host in ip_addr_list:
EXPECTSTUB = 'expectStub.exe'
#plink px = winpexpect.winspawn('c:/python/openssh/bin/plink.exe -t ' + user + '@' + host)
px = winpexpect.winspawn('c:/python/openssh/bin/plink.exe -ssh ' + user + '@' + host, timeout=6, stub=EXPECTSTUB)
fout = open('c:/python/checkpoint/test_debug/{0}.txt'.format(host),"w")
#px.logfile = sys.stdout
px.logfile_read = fout
px.expect('password:'******'1q2w3e\n')
px.expect('>')
px.send('expert\n')
px.expect('password:'******'1q2w3e\n')
px.expect('now.', timeout=10)
px.send('\n')
px.expect('.*Expert.*')
skip()
px.send('cat /etc/hosts\n')
skip()
def getStatistics(siebel_server, hostServer, isFullMetrics=False):
child = None
isError = False
try:
try:
result = dict()
errorPrefix = 'SIEBEL_STAT ERROR| '
successPrefix = 'SIEBEL_STAT OK| '
compNameMap = dict(cgi.parse_qsl(config_props["siebel.mgmt.components.map"]))
metrics_scale = dict(cgi.parse_qsl(config_props["siebel.mgmt.metrics.scale.map"]))
if metrics_scale == None:
metrics_scale = dict()
logger.debug( "Getting stat for enterprise: %s" % (config_props["siebel.enterprise"]) )
svrmgr_call = 'srvrmgr.exe /g ' + config_props["siebel.gateway"] + ' /e ' + config_props["siebel.enterprise"] + ' /s ' + siebel_server + ' /u ' + config_props["siebel.user"] + ' /p ' + config_props["siebel.password"]
svrmgr_prompt = 'srvrmgr:' + siebel_server + '>'
child = winpexpect.winspawn(svrmgr_call)
#time.sleep(15)
logger.debug( "Checking for the prompt: %s" % (svrmgr_prompt) )
child.expect (svrmgr_prompt, timeout=120)
headerMsg = child.before
#Get Server status
logger.info( "Getting the server status" )
child.sendline ('list servers show SBLSRVR_NAME,HOST_NAME,SBLSRVR_STATE,START_TIME,END_TIME')
child.expect (svrmgr_prompt, timeout=120)
stats = child.before
logger.debug( stats )
serverPrefix = "Server_"
oneStats = buildStatDict(stats, serverPrefix,0,2)
logger.debug( oneStats )
if oneStats.get(serverPrefix+siebel_server, None) == None:
return buildNagioOut(errorPrefix + serverPrefix + 'STATE=3 ', result, isError=True)
serverStatus = oneStats[serverPrefix+siebel_server].upper()
del oneStats[serverPrefix+siebel_server]
if serverStatus != 'RUNNING':
return buildNagioOut(errorPrefix + serverPrefix + 'STATE=2 ', result, isError=True)
result.update(oneStats)
result[serverPrefix + 'STATE'] = '0' #serverStatus
components = config_props['siebel.mgmt.components.' + hostServer].split(',')
taskComponents = config_props['siebel.mgmt.interactive.components.' + hostServer].split(',')
#Get Components status
logger.info( "Getting the component status" )
child.sendline ('list component show CC_ALIAS,CG_ALIAS,CP_DISP_RUN_STATE,CP_START_TIME,CP_END_TIME')
child.expect (svrmgr_prompt, timeout=120)
stats = child.before
compPrefix = 'comp_'
oneStats = buildStatDict(stats, compPrefix,0,2)
logger.debug( oneStats )
for aSibelComp in components:
aComp = compNameMap.get(aSibelComp, aSibelComp)
if oneStats.get(compPrefix+aSibelComp, None) == None:
return buildNagioOut(errorPrefix + compPrefix+aComp + '_STATE=3 ', result, isError=True)
compStatus = oneStats[compPrefix+aSibelComp].upper()
#del oneStats[compPrefix+aSibelComp]
if compStatus != 'RUNNING' and compStatus != 'ONLINE':
return buildNagioOut(errorPrefix + compPrefix+aComp + '_STATE=2 ', result, isError=True)
logger.debug( oneStats )
#result.update(oneStats)
result[compPrefix+aComp + '_STATE'] = '0' #compStatus
#Get Running Tasks status for each Component
logger.info( "Getting the component tasks status" )
goodpctStr = config_props["siebel.mgmt.interactive.components.goodPct"]
goodpct = -1
if goodpctStr != None:
goodpct = int(goodpctStr)
for aSibelComp in taskComponents:
aComp = compNameMap.get(aSibelComp, aSibelComp)
taskPrefix = 'task_'+ aComp + '_'
child.sendline ('list tasks for component ' + aSibelComp + ' show SV_NAME, CC_ALIAS, TK_PID, TK_DISP_RUNSTATE')
child.expect (svrmgr_prompt, timeout=120)
stats = child.before
oneStats = buildStatDict(stats, taskPrefix,0,3, inc=1)
logger.debug( oneStats )
if len(oneStats) <= 0:
isError = True
logger.warn( "No active tasks found")
result[taskPrefix + 'STATE'] = '1'
#return buildNagioOut(errorPrefix + taskPrefix + 'STATE=1 ', result, isError=True)
else:
errorTasks = 0
for aTask in oneStats.keys():
taskStatus = oneStats[aTask].upper()
if taskStatus != 'RUNNING' and taskStatus != 'COMPLETED' and taskStatus != 'ONLINE':
errorTasks = errorTasks + 1
#result[taskPrefix + 'STATE'] = taskStatus
#return buildNagioOut(errorPrefix + taskPrefix + 'STATE=' + taskStatus + ' ', result, isError=True)
if errorTasks <= 0:
#result.update(oneStats)
result[taskPrefix + 'STATE'] = '0' #taskStatus
else:
if goodpct < 0 or (((len(oneStats)-errorTasks)/len(oneStats))*100) > goodpct:
logger.warn( "More than allowed number of tasks are down: %s out of %s for task: %s" % (errorTasks,len(oneStats),taskPrefix))
result[taskPrefix + 'STATE'] = '2'
isError = True
result[taskPrefix + 'TOTAL_TASKS'] = str(len(oneStats)-errorTasks)
#Get statistics for each Component
compErrors = 0
logger.info( "Getting the component statistic list" )
for aSibelComp in components:
aComp = compNameMap.get(aSibelComp, aSibelComp)
compPrefix = 'comp_'+ aComp + '_'
logger.debug( "Getting stat for component: %s" % (aSibelComp) )
child.sendline ('list statistics for component ' + aSibelComp + ' show STAT_ALIAS,SD_DATATYPE,SD_SUBSYSTEM,CURR_VAL,STAT_NAME')
child.expect (svrmgr_prompt, timeout=120)
stats = child.before
oneStats = buildStatDict(stats, compPrefix,0,3)
logger.debug( oneStats )
if len(oneStats) <= 0:
isError = True
compErrors = compErrors + 1
result[compPrefix + '_STATE'] = '1'
#return buildNagioOut(errorPrefix + compPrefix + '_STATE=1 ', result, isError=True)
else:
for aMetric in metrics_scale:
metricVal = oneStats.get(aMetric, None)
if metricVal != None:
metricVal = str(int(float(metricVal)*float(metrics_scale[aMetric])))
oneStats[aMetric] = metricVal
result.update(oneStats)
logger.info( "All done; exiting!" )
child.sendline ('exit')
#time.sleep(1)
except Exception:
logger.exception("Failed to get statistics")
return buildNagioOut(errorPrefix + 'General_Error ', result, isError=True)
finally:
try:
child.close()
except Exception:
pass
if isError:
return buildNagioOut(errorPrefix, result, isError=True, isFullMetrics=isFullMetrics)
else:
return buildNagioOut(successPrefix, result, isError=False, isFullMetrics=isFullMetrics)
import winpexpect as winpexpect
winpexpect.run("ssh [email protected] 'mkdir hu'",
events={'(?i)password': '******'})
print 'hhhhhhhhhhhhhh'
import sys, winpexpect
child = winpexpect.winspawn('nslookup')
child.logfile = sys.stdout
child.expect('\n>')
child.sendline('www.google.com')
child.expect('\n>')
result = winpexpect.winspawn('ssh %s@%s' % ('root', '106.15.224.173'))
print result.before, child.after
result.sendline('Huqiang2018')
result.sendline('ls -l')
print result.before, result.after
result.interact()
print 'hu'
result = winpexpect.run("ssh [email protected] 'hostname hu'",
events={'(?i)password': '******'})
print 'hu'
def test_expect():
ps = winpexpect.winspawn('ping')
ps.sendline('exit')
print ps.before.decode('gbk')
cmrb = xlrd.open_workbook('c:/python/checkpoint/commands.xls')
sheet1 = cmrb.sheet_by_name('commands')
num_rows1 = sheet1.nrows
num_cols1 = sheet1.ncols
ip_addr_list = [sheet.row_values(rawnum)[0] for rawnum in range(sheet.nrows)]
commands_list = [sheet1.row_values(rawnum)[0] for rawnum in range(sheet1.nrows)]
num_ips = 0
host = '10.10.0.254'
user = '******'
passwd = '1q2w3e'
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
px = winpexpect.winspawn('c:/python/openssh/bin/ssh.exe ' + user + '@' + host)
fout = open('c:/python/checkpoint/{0}++.txt'.format(host),"w")
px.logfile_read = fout
px.timeout = 35
px.expect('authorized use only.')
px.sendline(passwd)
#create log file with in IP.txt format (10.1.1.1.txt, for example)
fout = open('c:/python/checkpoint/{0}++.txt'.format(host),"wb")
#push pexpect logfile_read output to log file
px.sendline('\n')
px.sendline('\n')
px.sendline('show interfaces all\n')
px.sendline('\n')
#for item in ip_addr_list:
from __future__ import unicode_literals
import winpexpect
import sys
child = winpexpect.winspawn('ftp ftp.openbsd.org')
child.winpexpect('(?i)name .*: ')
child.sendline('anonymous')
child.winexpect('(?i)password')
child.sendline('*****@*****.**')
child.winexpect('ftp> ')
child.sendline('bin')
child.winexpect('ftp> ')
child.sendline('get robots.txt')
child.winexpect('ftp> ')
sys.stdout.write(child.before)
print("Escape character is '^]'.\n")
sys.stdout.write(child.after)
sys.stdout.flush()
child.interact() # Escape character defaults to ^]
child.sendline('bye')
child.close()
# -*- coding: utf-8 -*-
"""
@author: Jack L. Clements
"""
import sys, winpexpect
#Note - I wanted to use winpexpect to run a test script that would, similar to a unit test, attempt to test various aspects of this script
#Either due to compatability or library issues, winpexpect clearly has other ideas, appending extraneous characters to inputs
#It also mangles all non-expect strings into a jumbled half-readable mess
#Due to a lack of a linux testing environment at present
child = winpexpect.winspawn('Python -u PasswordCheck.py')
child.logfile = sys.stdout
prompt = 'test'
child.expect('Please enter your username: \n')
child.sendline(
'User01'
) #appends newline and treats User01, a login attempt, as a whole new
child.expect_exact('User01, please enter your old password.')
child.expect_exact('Please enter your old password: \n')
child.sendline(str('Password1'))
child.expect_exact('Please enter a new password: \n')
child.sendline(str('TestPass3'))
child.expect_exact('Please enter the password again for verification: \n')
child.sendline(str('TestPass3'))
child.expect_exact('User password changed.')
child.close()
#code such as this could theoretically be used for unit testing, but I cannot seem to get any implementation to work properly and as such have left it here
from winpexpect import winspawn
#ssh = winspawn('ssh',['-tty','[email protected]'])
ssh = winspawn('ssh -tt [email protected]')
ssh.logfile = sys.stdout
i = ssh.expect(['word:'], timeout = 5)
print (i)
ssh.sendline('xzK3.............')
def test_multiple_terminate(self):
ps = winspawn('powershell.exe -command -')
ps.terminate()
ps.terminate()
def executeScript(inputScript, scriptOptionalArgs, inputFile=None):
global testcaseInfoList
global testcaseInfo
global yamlCont
testcaseInfoList = []
yamlCont = {}
if inputFile != None:
fil = open(inputFile, "r")
yamlCont = yaml.load(fil)
# inputFileType = re.search('(.*).yaml', inputFile).group(1)
if re.search("\.log\.", inputScript):
cmd = (
"python /work/swtest01_1/sgsubramaniam/sw-test/special_script/demoScript/slowcat.py -d 0.002 %s"
% inputScript
)
else:
cmd = "f10tool %s %s" % (inputScript, " ".join(scriptOptionalArgs))
if re.search("\.flist$", inputScript):
inputFileType = "flist"
cmd = "f10tool %s %s" % (inputScript, " ".join(scriptOptionalArgs))
elif re.search("\.f10$", inputScript):
inputFileType = "f10"
cmd = "f10tool %s %s" % (inputScript, " ".join(scriptOptionalArgs))
else:
inputFileType = "unknown"
if inputFile != None:
inputFileType = re.search("(.*).yaml", inputFile).group(1)
cmd = "%s %s;echo '----------'" % (inputScript, " ".join(scriptOptionalArgs))
ps = winpexpect.winspawn(cmd)
ps.logfile_read = sys.stdout
## Initialize testcaseInfo if the script is not a flist.
if inputFileType in ["f10", "unknown"]:
scriptName = inputScript
testcaseInfo = testInfo(scriptName)
testcaseInfoList.append(testcaseInfo)
timeout = -1
ps.interact(output_filter=scriptExecutionMonitor)
if inputFileType is "flist" or "rspec":
## creates a nested suite
suiteList = []
for testcaseInfo in testcaseInfoList:
testcaseInfo.flush()
suite = createRobotSuite(testcaseInfo)
suiteList.append(suite)
suite = TestSuite(inputScript)
suite.suites = suiteList
result = suite.run(output="output.xml", loglevel="debug")
for testcaseInfo in testcaseInfoList:
testcaseInfo.variableFile.close()
else:
## creates a single suite
for testcaseInfo in testcaseInfoList:
testcaseInfo.flush()
print testcaseInfo
suite = createRobotSuite(testcaseInfo)
result = suite.run(output="output.xml", loglevel="debug")
testcaseInfo.variableFile.close()
# Generating log files requires processing the earlier generated output XML.
ResultWriter("output.xml").write_results()
pp = postProcess.postProcess(suiteFile=testcaseInfo.scriptName)
pp.close()
def test_wait_after_close(self):
ps = winspawn('powershell.exe -command -')
ps.close() # closes stdin and should therefore exit powershell
ps.wait()
assert not ps.isalive()
def main():
cmdpath = r"C:\Windows\System32\cmd.exe"
child = winpexpect.winspawn(cmdpath)
child.expect(["C:\Users\cchhsu>"])
child.sendline("Hello World")
def test_wait_timeout(self):
ps = winspawn('powershell.exe -command -')
assert_raises(TIMEOUT, ps.wait, timeout=2)
ps.terminate()
def getCustomReport(child, i):
iteration = int(i / originalLabelLength)
i = i % originalLabelLength
#print(iteration, i, originalLabelLength, len(CRValueName))
devRadio = deviceRadio[iteration]
s = 'adb -s 192.168.1.31 shell \"dumpsys wifi | grep \'mWifiInfo\'\"'
try:
if deviceAccess == 'adb':
if i == 0:
#print("--------- TESTING ------------")
child = winpexpect.winspawn(
'adb -s {} shell \"dumpsys wifi | grep \'mWifiInfo:\'\"'.
format(deviceIP))
child.expect('Frequency: ', timeout=10)
child.expect('\d+')
data = child.after
#child.sendline('wl -i {} channel'.format(devRadio))
#child.expect('current mac channel')
#child.expect('\d+')
#data = int(child.after)
print('{}: '.format(CRValueName[i]), data)
return data
elif i == 1:
child = winpexpect.winspawn(
'adb -s {} shell \"dumpsys wifi | grep \'mWifiInfo:\'\"'.
format(deviceIP))
child.expect('RSSI: ', timeout=10)
child.expect('-\d+')
data = child.after
# child.sendline('wl -i {} bssid'.format(devRadio))
# child.expect('bssid')
# child.expect('[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]')
# data = (child.after)
print('{}: '.format(CRValueName[i]), data)
return data
elif i == 2:
child = winpexpect.winspawn(
'adb -s {} shell \"dumpsys wifi | grep \'mWifiInfo:\'\"'.
format(deviceIP))
child.expect('speed: ', timeout=10)
child.expect('\d+')
data = child.after
print('{}: '.format(CRValueName[i]), data)
return data
elif i == 3:
child = winpexpect.winspawn(
'adb -s {} shell \"dumpsys wifi | grep \'mWifiInfo:\'\"'.
format(deviceIP))
child.expect('BSSID: ', timeout=10)
child.expect(
'[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]:[0-9a-fA-F][0-9a-fA-F]'
)
data = child.after
print('{}: '.format(CRValueName[i]), data)
return data
elif deviceAccess == 'ssh':
if i == 0:
child.sendline(
'cd \/System\/Library\/PrivateFrameworks\/Apple80211.framework\/Versions\/Current\/Resources'
)
child.expect('$')
child.sendline('./airport -I') # was 'wl -i eth2 rssi'
child.expect('channel:')
child.expect('\d+')
data = int(child.after)
print('{}: '.format(CRValueName[i]), data)
return data
elif i == 1:
child.sendline('./airport -I') # was 'wl -i eth2 rssi'
child.expect('agrCtlRSSI:')
child.expect('-\d\d|\d')
data = int(child.after)
print('{}: '.format(CRValueName[i]), data)
return data
elif i == 2:
child.sendline('./airport -I') # was 'wl -i eth2 rssi'
child.expect('lastTxRate:')
child.expect('\d+')
data = int(child.after)
print('{}: '.format(CRValueName[i]), data)
return data
elif i == 3:
child.sendline('./airport -I') # was 'wl -i eth2 rssi'
child.expect('MCS:')
child.expect('\d+|-\d')
data = int(child.after)
print('{}: '.format(CRValueName[i]), data)
return data
return "#N/A"
except:
print("ERROR - Timeout reached")
print("Error on child.before:", child.before, "& child.after:",
child.after)
data = "#ERROR"
print('{}: '.format(CRValueName[i]), data)
return data
def test_expect_stderr(self):
ps = winspawn('powershell.exe -command -')
ps.sendline('Write-Error "my-error"')
ps.expect('my-error')
ps.terminate()
import sys
import winpexpect
from pexpect import TIMEOUT
from pip._vendor.webencodings.tests import assert_raises
ps = winpexpect.winspawn('powershell.exe -command -')
assert_raises(TIMEOUT, ps.wait, timeout=2)
ps.terminate()
child = winpexpect.winspawn('ssh', ['-tty','[email protected]'])
child.logfile = sys.stdout
child.sendline('Huqiang2018')
child .sendline('ls -l /')
print('Now enter the FTP interactive mode')
child.interact()
def test_timeout(self):
ps = winspawn('powershell.exe -command -')
ps.sendline('Sleep 10; Write "done sleeping"')
assert_raises(TIMEOUT, ps.expect, 'done sleeping', timeout=2)
ps.terminate()
