NetCube-Systems-Austria/kumquat-buildroot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support/testing: use pexpect in emulator

Browse Source 
Replace subprocess + telnetlib with pexpect.

Use the telnet installed on the host machine instead of telnetlib, while
the serial from qemu is not yet redirected to stdio.

Signed-off-by: Ricardo Martincoski <ricardo.martincoski@gmail.com>
Signed-off-by: Thomas Petazzoni <thomas.petazzoni@free-electrons.com>
This commit is contained in:
Ricardo Martincoski 2017-06-28 23:45:41 -03:00 committed by Thomas Petazzoni
parent d498aa4a34
commit eb0fab80f6
1 changed files with 13 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
support/testing/infra/emulator.py
View File

@ -1,6 +1,4 @@
import socket import pexpect
import subprocess
import telnetlib
import infra import infra
import infra.basetest import infra.basetest
@ -71,25 +69,24 @@ class Emulator(object):
qemu_cmd += ["-append", " ".join(kernel_cmdline)] qemu_cmd += ["-append", " ".join(kernel_cmdline)]
self.logfile.write("> starting qemu with '%s'\n" % " ".join(qemu_cmd)) self.logfile.write("> starting qemu with '%s'\n" % " ".join(qemu_cmd))
self.qemu = subprocess.Popen(qemu_cmd, stdout=self.logfile, stderr=self.logfile) self.qemu = pexpect.spawn(qemu_cmd[0], qemu_cmd[1:])
# Wait for the telnet port to appear and connect to it. # Wait for the telnet port to appear and connect to it.
while True: self.qemu.expect("waiting for connection")
try: telnet_cmd = ["telnet", "localhost", "1234"]
self.__tn = telnetlib.Telnet("localhost", 1234) self.__tn = pexpect.spawn(telnet_cmd[0], telnet_cmd[1:])
if self.__tn:
break
except socket.error:
continue
def __read_until(self, waitstr, timeout=5): def __read_until(self, waitstr, timeout=5):
data = self.__tn.read_until(waitstr, timeout) index = self.__tn.expect([waitstr, pexpect.TIMEOUT], timeout=timeout)
data = self.__tn.before
if index == 0:
data += self.__tn.after
self.log += data self.log += data
self.logfile.write(data) self.logfile.write(data)
return data return data
def __write(self, wstr): def __write(self, wstr):
self.__tn.write(wstr) self.__tn.send(wstr)
# Wait for the login prompt to appear, and then login as root with # Wait for the login prompt to appear, and then login as root with
# the provided password, or no password if not specified. # the provided password, or no password if not specified.
@ -124,7 +121,8 @@ class Emulator(object):
return output, exit_code return output, exit_code
def stop(self): def stop(self):
if self.__tn:
self.__tn.terminate(force=True)
if self.qemu is None: if self.qemu is None:
return return
self.qemu.terminate() self.qemu.terminate(force=True)
self.qemu.kill()