ansible/test/units/plugins/lookup/test_env.py

# -*- coding: utf-8 -*-
# Copyright: (c) 2019, Abhay Kadam <abhaykadam88@gmail.com>
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)

from __future__ import annotations

import pytest

from ansible.plugins.loader import lookup_loader


@pytest.mark.parametrize('env_var,exp_value', [
    ('foo', 'bar'),
    ('equation', 'a=b*100')
])
def test_env_var_value(monkeypatch, env_var, exp_value):
    monkeypatch.setattr('os.environ.get', lambda x, y: exp_value)

    env_lookup = lookup_loader.get('env')
    retval = env_lookup.run([env_var], None)
    assert retval == [exp_value]


@pytest.mark.parametrize('env_var,exp_value', [
    ('simple_var', 'alpha-β-gamma'),
    ('the_var', 'ãnˈsiβle')
])
def test_utf8_env_var_value(monkeypatch, env_var, exp_value):
    monkeypatch.setattr('os.environ.get', lambda x, y: exp_value)

    env_lookup = lookup_loader.get('env')
    retval = env_lookup.run([env_var], None)
    assert retval == [exp_value]