from dataclasses import dataclass, field
from omegaconf import OmegaConf


@dataclass
class DeeplyNestedConf:
    an_int: int = 10
    a_string: str = "test"


@dataclass
class StructuredConf:
    single_nest: dict[str, DeeplyNestedConf] = field(default_factory=dict)
    double_nest: dict[str, dict[str, DeeplyNestedConf]] = field(default_factory=dict)


def test_nested_struct():
    yaml = """
    single_nest:
        level_1:
            an_int: 20
    double_nest:
        level_1:
            nested_level_1:
                an_int: 30
                a_string: this is not typed
            nested_level_2:
                an_int: 40
                a_typo: this does not fail
    """

    schema = OmegaConf.structured(StructuredConf())
    untyped = OmegaConf.create(yaml)
    typed = OmegaConf.merge(schema, untyped)
    structured_conf = OmegaConf.to_object(typed)  # type: StructuredConf
    assert isinstance(structured_conf.single_nest["level_1"], DeeplyNestedConf)  # Awesome!
    not_typed = structured_conf.double_nest["level_1"]["nested_level_1"]
    assert not isinstance(not_typed, DeeplyNestedConf)  # why not?
    assert isinstance(not_typed, dict)  # it’s just a dict :-(
    did_not_fail_validation = structured_conf.double_nest["level_1"]["nested_level_2"]
    assert not isinstance(did_not_fail_validation, DeeplyNestedConf)
    assert isinstance(did_not_fail_validation, dict)