from collections.abc import Iterator
from unittest import TestCase

from xsdata.models.xsd import (
    MaxExclusive,
    MinExclusive,
    MinInclusive,
    MinLength,
    Restriction,
    SimpleType,
    Union,
)


class UnionTests(TestCase):
    def test_property_bases(self) -> None:
        obj = Union()
        self.assertIsInstance(obj.bases, Iterator)
        self.assertEqual([], list(obj.bases))

        obj.member_types = "foo bar   "
        self.assertEqual(["foo", "bar"], list(obj.bases))

    def test_property_is_property(self) -> None:
        obj = Union()
        self.assertTrue(obj.is_property)

    def test_property_real_name(self) -> None:
        obj = Union()
        self.assertEqual("value", obj.real_name)

    def test_property_attr_types(self) -> None:
        obj = Union()
        obj.member_types = "thug life"
        self.assertEqual(["thug", "life"], list(obj.attr_types))

        obj = Union(
            simple_types=[
                SimpleType(restriction=Restriction(base="foo")),
                SimpleType(restriction=Restriction(base="bar")),
            ]
        )

        self.assertEqual(["foo", "bar"], list(obj.attr_types))

    def test_get_restrictions(self) -> None:
        first = Restriction(
            min_exclusive=MinExclusive(value="1"), min_inclusive=MinInclusive(value="2")
        )
        second = Restriction(
            min_length=MinLength(value="3"), max_exclusive=MaxExclusive(value="4")
        )
        obj = Union(
            simple_types=[
                SimpleType(restriction=first),
                SimpleType(restriction=second),
            ]
        )

        expected = {
            "max_exclusive": "4",
            "min_exclusive": "1",
            "min_inclusive": "2",
            "min_length": "3",
        }
        self.assertEqual(expected, obj.get_restrictions())