import os

import testinfra.utils.ansible_runner

testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
    os.environ['MOLECULE_INVENTORY_FILE']).get_hosts('all')


def test_run(host):
    youkiruntime = "/usr/local/bin/youki"
    with host.sudo():
        cmd = host.command(youkiruntime + " --version")
    assert cmd.rc == 0
    assert "youki" in cmd.stdout


def test_run_pod(host):
    runtime = "youki"

    run_command = "/usr/local/bin/crictl run --with-pull --runtime {} /tmp/container.json /tmp/sandbox.json".format(runtime)
    with host.sudo():
        cmd = host.command(run_command)
    assert cmd.rc == 0

    with host.sudo():
      log_f = host.file("/tmp/youki1.0.log")

      assert log_f.exists
      assert b"Hello from Docker" in log_f.content