import pytest

from labgrid.exceptions import NoDriverFoundError


@pytest.fixture()
def strategy(target):
    try:
        return target.get_driver('ExampleStrategy')
    except NoDriverFoundError:
        pytest.skip("strategy not found")


@pytest.fixture(scope="function")
def bootloader(target, strategy, capsys):
    with capsys.disabled():
        strategy.transition("barebox")
    return target['CommandProtocol'] # this will return the BareboxDriver


@pytest.fixture(scope="function")
def shell(target, strategy, capsys):
    with capsys.disabled():
        strategy.transition("shell")
    return target['CommandProtocol'] # this will return the ShellDriver


def test_barebox(bootloader):
    stdout, stderr, returncode = bootloader.run('version')
    assert returncode == 0
    assert stdout
    assert not stderr
    assert 'barebox' in '\n'.join(stdout)


def test_shell(shell):
    stdout, stderr, returncode = shell.run('cat /proc/version')
    assert returncode == 0
    assert stdout
    assert not stderr
    assert 'Linux' in stdout[0]


def test_barebox_2(bootloader):
    bootloader.run_check('true')