summaryrefslogtreecommitdiffstats
path: root/test/units/plugins/lookup/test_env.py
blob: 5d9713fe9cf0a7a1984d0269aec5b2894974dbf2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# -*- 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)

# Make coding more python3-ish
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

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('ansible.utils.py3compat.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('ansible.utils.py3compat.environ.get', lambda x, y: exp_value)

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