mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
2ff60c1485
Add test for unsafe plugin util (#8345)
Add test for unsafe plugin util.
(cherry picked from commit 7aa118b957)
Co-authored-by: Felix Fontein <felix@fontein.de>
133 lines
2.8 KiB
Python
133 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) 2024, Felix Fontein <felix@fontein.de>
|
|
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
# Make coding more python3-ish
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
__metaclass__ = type
|
|
|
|
|
|
import pytest
|
|
|
|
from ansible.utils.unsafe_proxy import AnsibleUnsafe
|
|
|
|
from ansible_collections.community.general.plugins.plugin_utils.unsafe import (
|
|
make_unsafe,
|
|
)
|
|
|
|
|
|
TEST_MAKE_UNSAFE = [
|
|
(
|
|
u'text',
|
|
[],
|
|
[
|
|
(),
|
|
],
|
|
),
|
|
(
|
|
u'{{text}}',
|
|
[
|
|
(),
|
|
],
|
|
[],
|
|
),
|
|
(
|
|
b'text',
|
|
[],
|
|
[
|
|
(),
|
|
],
|
|
),
|
|
(
|
|
b'{{text}}',
|
|
[
|
|
(),
|
|
],
|
|
[],
|
|
),
|
|
(
|
|
{
|
|
'skey': 'value',
|
|
'ukey': '{{value}}',
|
|
1: [
|
|
'value',
|
|
'{{value}}',
|
|
{
|
|
1.0: '{{value}}',
|
|
2.0: 'value',
|
|
},
|
|
],
|
|
},
|
|
[
|
|
('ukey', ),
|
|
(1, 1),
|
|
(1, 2, 1.0),
|
|
],
|
|
[
|
|
('skey', ),
|
|
(1, 0),
|
|
(1, 2, 2.0),
|
|
],
|
|
),
|
|
(
|
|
['value', '{{value}}'],
|
|
[
|
|
(1, ),
|
|
],
|
|
[
|
|
(0, ),
|
|
],
|
|
),
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize("value, check_unsafe_paths, check_safe_paths", TEST_MAKE_UNSAFE)
|
|
def test_make_unsafe(value, check_unsafe_paths, check_safe_paths):
|
|
unsafe_value = make_unsafe(value)
|
|
assert unsafe_value == value
|
|
for check_path in check_unsafe_paths:
|
|
obj = unsafe_value
|
|
for elt in check_path:
|
|
obj = obj[elt]
|
|
assert isinstance(obj, AnsibleUnsafe)
|
|
for check_path in check_safe_paths:
|
|
obj = unsafe_value
|
|
for elt in check_path:
|
|
obj = obj[elt]
|
|
assert not isinstance(obj, AnsibleUnsafe)
|
|
|
|
|
|
def test_make_unsafe_dict_key():
|
|
value = {
|
|
b'test': 1,
|
|
u'test': 2,
|
|
}
|
|
unsafe_value = make_unsafe(value)
|
|
assert unsafe_value == value
|
|
for obj in unsafe_value:
|
|
assert not isinstance(obj, AnsibleUnsafe)
|
|
|
|
value = {
|
|
b'{{test}}': 1,
|
|
u'{{test}}': 2,
|
|
}
|
|
unsafe_value = make_unsafe(value)
|
|
assert unsafe_value == value
|
|
for obj in unsafe_value:
|
|
assert isinstance(obj, AnsibleUnsafe)
|
|
|
|
|
|
def test_make_unsafe_set():
|
|
value = set([b'test', u'test'])
|
|
unsafe_value = make_unsafe(value)
|
|
assert unsafe_value == value
|
|
for obj in unsafe_value:
|
|
assert not isinstance(obj, AnsibleUnsafe)
|
|
|
|
value = set([b'{{test}}', u'{{test}}'])
|
|
unsafe_value = make_unsafe(value)
|
|
assert unsafe_value == value
|
|
for obj in unsafe_value:
|
|
assert isinstance(obj, AnsibleUnsafe)
|