from __future__ import print_function

import subprocess

import pytest
import rostest.util as u


@pytest.fixture(scope="module")
def build_and_run(request):
    print('\nBuilding and running RancherOS with custom kernel')
    p = subprocess.Popen(['./scripts/docker-run.sh', '--rm',
                          './tests/integration/assets/test_02/test-custom-kernel.sh'],
                         stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)

    def fin():
        print('\nTerminating docker-run test-custom-kernel')
        p.terminate()
        p.wait()

    request.addfinalizer(fin)
    return p


@pytest.mark.timeout(30)
def test_system_boot(build_and_run):
    version = u.rancheros_version('./tests/integration/assets/test_02/build.conf')
    print('parsed version: ' + version)

    u.flush_out(build_and_run.stdout, 'RancherOS {v} started'.format(v=version))