import os
import pytest

import testinfra.utils.ansible_runner

testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
    os.environ['MOLECULE_INVENTORY_FILE']).get_hosts(
      'kubernetes-control-plane')


@pytest.mark.parametrize('container_name', [
  'kube-apiserver', 'kube-controller-manager', 'kube-scheduler', 'kubelet'])
def test_container_running(host, container_name):
    assert host.docker(container_name).is_running