riotctrl_shell.tests: rewrite Mock to detect missing check_term
`check_term` calls `run_term` of the `ctrl` if necessary.
This commit is contained in:
parent
b110f9e38c
commit
637c673e6a
31
dist/pythonlibs/riotctrl_shell/tests/common.py
vendored
31
dist/pythonlibs/riotctrl_shell/tests/common.py
vendored
@ -4,25 +4,27 @@
|
|||||||
# General Public License v2.1. See the file LICENSE in the top level
|
# General Public License v2.1. See the file LICENSE in the top level
|
||||||
# directory for more details.
|
# directory for more details.
|
||||||
|
|
||||||
|
import contextlib
|
||||||
|
|
||||||
|
|
||||||
class MockSpawn():
|
class MockSpawn():
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, ctrl, *args, **kwargs):
|
||||||
|
self.ctrl = ctrl
|
||||||
|
self.last_command = None
|
||||||
# set some expected attributes
|
# set some expected attributes
|
||||||
self.before = None
|
self.before = None
|
||||||
self.echo = False
|
self.echo = False
|
||||||
self.output = None
|
|
||||||
self.last_command = None
|
|
||||||
|
|
||||||
def sendline(self, line, *args, **kwargs):
|
def sendline(self, line, *args, **kwargs):
|
||||||
self.last_command = line
|
self.last_command = line
|
||||||
if self.output is None:
|
if self.ctrl.output is None:
|
||||||
# just echo last input for before (what replwrap is assembling
|
# just echo last input for before (what replwrap is assembling
|
||||||
# output from)
|
# output from)
|
||||||
self.before = line
|
self.before = line
|
||||||
else:
|
else:
|
||||||
# use pre-configured output in case command expects a specific
|
# use pre-configured output in case command expects a specific
|
||||||
# output
|
# output
|
||||||
self.before = self.output
|
self.before = self.ctrl.output
|
||||||
|
|
||||||
def expect_exact(self, *args, **kwargs):
|
def expect_exact(self, *args, **kwargs):
|
||||||
# always match on prompt with replwrap
|
# always match on prompt with replwrap
|
||||||
@ -34,11 +36,26 @@ class MockRIOTCtrl():
|
|||||||
Mock RIOT ctrl
|
Mock RIOT ctrl
|
||||||
"""
|
"""
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
self.term = MockSpawn()
|
self.term = None
|
||||||
|
self.output = None
|
||||||
self.last_command = None
|
self.last_command = None
|
||||||
|
|
||||||
|
@contextlib.contextmanager
|
||||||
|
def run_term(self, reset=True, **startkwargs):
|
||||||
|
try:
|
||||||
|
self.start_term(**startkwargs)
|
||||||
|
yield self.term
|
||||||
|
finally:
|
||||||
|
self.stop_term()
|
||||||
|
|
||||||
|
def start_term(self, **spawnkwargs):
|
||||||
|
self.term = MockSpawn(self)
|
||||||
|
|
||||||
|
def stop_term(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
def init_ctrl(output=None):
|
def init_ctrl(output=None):
|
||||||
rc = MockRIOTCtrl("foobar", env={"BOARD": "native"})
|
rc = MockRIOTCtrl("foobar", env={"BOARD": "native"})
|
||||||
rc.term.output = output
|
rc.output = output
|
||||||
return rc
|
return rc
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user