import pytest
import rostest.util as u
from rostest.util import SSH

ssh_command = ['./scripts/ssh', '--qemu', '--key', './tests/integration/assets/test.key']
cloud_config_path = {'amd64': './tests/integration/assets/test_11/cloud-config.yml',
                     'arm': './tests/integration/assets/test_11/cloud-config_arm.yml',
                     'arm64': './tests/integration/assets/test_11/cloud-config_arm64.yml'}


@pytest.fixture(scope="module")
def qemu(request):
    q = u.run_qemu(request, ['--cloud-config', cloud_config_path[u.arch]])
    u.flush_out(q.stdout)
    return q


@pytest.mark.timeout(40)
def test_rancher_environment_in_system_service(qemu):
    SSH(qemu, ssh_command).check_call('''
sudo system-docker logs env | grep A=A
sudo system-docker logs env | grep BB=BB
sudo system-docker logs env | grep BC=BC
    ''')