podman-compose/tests/unit/test_normalize_depends_on.py

from __future__ import annotations

import copy
import unittest
from typing import Any

from parameterized import parameterized

from podman_compose import normalize_service


class TestNormalizeServicesSimple(unittest.TestCase):
    @parameterized.expand([
        (
            {"depends_on": "my_service"},
            {"depends_on": {"my_service": {"condition": "service_started"}}},
        ),
        (
            {"depends_on": ["my_service"]},
            {"depends_on": {"my_service": {"condition": "service_started"}}},
        ),
        (
            {"depends_on": ["my_service1", "my_service2"]},
            {
                "depends_on": {
                    "my_service1": {"condition": "service_started"},
                    "my_service2": {"condition": "service_started"},
                },
            },
        ),
        (
            {"depends_on": {"my_service": {"condition": "service_started"}}},
            {"depends_on": {"my_service": {"condition": "service_started"}}},
        ),
        (
            {"depends_on": {"my_service": {"condition": "service_healthy"}}},
            {"depends_on": {"my_service": {"condition": "service_healthy"}}},
        ),
    ])
    def test_normalize_service_simple(
        self, test_case: dict[str, Any], expected: dict[str, Any]
    ) -> None:
        copy.deepcopy(test_case)
        result = normalize_service(test_case)
        self.assertEqual(result, expected)