def test_run_with_timeout(self, target_with_fakeconsole, mocker):
t = target_with_fakeconsole
d = ShellDriver(t,
"shell",
prompt='dummy',
login_prompt='dummy',
username='******')
d.on_activate = mocker.MagicMock()
d = t.get_driver('ShellDriver')
d.run = mocker.MagicMock(return_value=[['success'], [], 0])
res = d.run_check("test", timeout=30.0)
assert res == ['success']
def test_run_error(self, target_with_fakeconsole, mocker):
t = target_with_fakeconsole
d = ShellDriver(t,
"shell",
prompt='dummy',
login_prompt='dummy',
username='******')
d.on_activate = mocker.MagicMock()
d = t.get_driver('ShellDriver')
d.run = mocker.MagicMock(return_value=[['error'], [], 1])
with pytest.raises(ExecutionError):
res = d.run_check("test")
def test_create_barebox(target):
console = FakeConsoleDriver(target, "console")
power = FakePowerDriver(target, "power")
barebox = BareboxDriver(target, "barebox")
shell = ShellDriver(target, "shell", prompt='root@dummy', login_prompt='login:'******'root')
s = BareboxStrategy(target, "strategy")
assert isinstance(s, Strategy)
assert target.get_driver(BareboxStrategy) is s
assert target.get_driver(Strategy) is s
assert s.state is BindingState.bound
def test_create_uboot(target):
console = FakeConsoleDriver(target)
power = FakePowerDriver(target)
barebox = UBootDriver(target)
shell = ShellDriver(target,
prompt='root@dummy',
login_prompt='login:'******'root')
s = UBootStrategy(target)
assert isinstance(s, Strategy)
assert target.get_driver(UBootStrategy) is s
assert target.get_driver(Strategy) is s
assert s.state is BindingState.bound
def test_instance(self, target, serial_driver):
s = ShellDriver(target, "", "", "")
assert (isinstance(s, ShellDriver))
def test_no_driver(self, target):
with pytest.raises(NoDriverFoundError):
ShellDriver(target, "", "", "")
