import pytest
import subprocess
import rancherostest.util as u


@pytest.fixture(scope="module")
def qemu(request):
    return u.run_qemu(request)


def rancheros_version():
    with open('./scripts/version') as f:
        for ln in f:
            (k, _, v) = ln.partition('=')
            if k == 'VERSION' and v.strip() != '':
                return v.strip()
    raise RuntimeError("Could not parse RancherOS version")


@pytest.mark.timeout(30)
def test_system_boot(qemu):
    version = rancheros_version()
    print('parsed version: ' + version)
    for ln in u.iter_lines(qemu.stdout):
        ros_booted_substr = str.find(ln, 'RancherOS {v} started'.format(v=version))
        print(str.strip(ln))
        if ros_booted_substr > -1:
            assert True
            return
    assert False


@pytest.mark.timeout(60)
def test_run_system_container(qemu):
    assert qemu is not None
    u.wait_for_ssh()

    ssh = subprocess.Popen(
        './scripts/ssh sudo system-docker run --rm busybox /bin/true', shell=True,
        stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)

    for ln in u.iter_lines(ssh.stdout):
        print(str.strip(ln))
    ssh.wait()

    assert ssh.returncode == 0