"""Test for yaml loader

    A simple pytest script to test loading environment variables in an yaml file

    requirement: pytest
    run: pytest -s
"""
import os

import pytest

from loader import load_yaml


def test_loader():
    """Test fails if loader fails to parse environment variables"""
    os.environ.update({"VAR1": "var1", "VAR2": "var2"})
    data = load_yaml("sample.yaml")
    assert "var1" in data["node1"][0]["field"]
    assert "var2" in data["node1"][1]["field"]
    assert all(x in data["node1"][2]["field"] for x in ["var1", "var2"])
    assert "123" in data["node1"][3]["field"]
    assert "" == data["node1"][4]["field"]