from __future__ import print_function
import itertools as it
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('./build.conf') as f:
        for v in it.ifilter(u.non_empty,
                            it.imap(u.parse_value('VERSION'),
                                    it.ifilter(u.non_empty,
                                               it.imap(u.strip_comment('#'), u.iter_lines(f))))):
            return v
    raise RuntimeError("Could not parse RancherOS version")


@pytest.mark.timeout(30)
def test_system_boot(qemu):
    version = rancheros_version()
    print('parsed version: ' + version)

    def has_ros_started_substr(s):
        return str.find(s, 'RancherOS {v} started'.format(v=version)) > -1

    for _ in it.ifilter(has_ros_started_substr,
                        it.imap(u.with_effect(print), u.iter_lines(qemu.stdout))):
        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(ln)
    ssh.wait()

    assert ssh.returncode == 0