community.general/tests/unit/plugins/module_utils/test_opennebula.py

# -*- coding: utf-8 -*-
# Copyright (c) 2023, Michal Opala <mopala@opennebula.io>
# 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

from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

import os
import textwrap

import pytest

from ansible_collections.community.general.plugins.module_utils.opennebula import flatten, render


FLATTEN_VALID = [
    (
        [[[1]], [2], 3],
        False,
        [1, 2, 3]
    ),
    (
        [[[1]], [2], 3],
        True,
        [1, 2, 3]
    ),
    (
        [[1]],
        False,
        [1]
    ),
    (
        [[1]],
        True,
        1
    ),
    (
        1,
        False,
        [1]
    ),
    (
        1,
        True,
        1
    ),
]

RENDER_VALID = [
    (
        {
            "NIC": {"NAME": "NIC0", "NETWORK_ID": 0},
            "CPU": 1,
            "MEMORY": 1024,
        },
        textwrap.dedent('''
            CPU="1"
            MEMORY="1024"
            NIC=[NAME="NIC0",NETWORK_ID="0"]
        ''').strip()
    ),
    (
        {
            "NIC": [
                {"NAME": "NIC0", "NETWORK_ID": 0},
                {"NAME": "NIC1", "NETWORK_ID": 1},
            ],
            "CPU": 1,
            "MEMORY": 1024,
        },
        textwrap.dedent('''
            CPU="1"
            MEMORY="1024"
            NIC=[NAME="NIC0",NETWORK_ID="0"]
            NIC=[NAME="NIC1",NETWORK_ID="1"]
        ''').strip()
    ),
]


@pytest.mark.parametrize('to_flatten,extract,expected_result', FLATTEN_VALID)
def test_flatten(to_flatten, extract, expected_result):
    result = flatten(to_flatten, extract)
    assert result == expected_result, repr(result)


@pytest.mark.parametrize('to_render,expected_result', RENDER_VALID)
def test_render(to_render, expected_result):
    result = render(to_render)
    assert result == expected_result, repr(result)