import os
import tempfile
from pathlib import Path

import yaml
from yaml import CDumper as Dumper

from nwb_linkml.io.yaml import load_yaml


def test_preload_maps():
    hdmf_style_naming = {
        "groups": [
            {
                "data_type_def": "Container",
                "data_type_inc": "MainClass",
                "doc": "Demo group",
                "datasets": [{"data_type_inc": "Data"}],
            }
        ]
    }

    temp, temp_name = tempfile.mkstemp(suffix=".yaml")

    with open(temp_name, "w") as temp_f:
        yaml.dump(hdmf_style_naming, temp_f, Dumper=Dumper)
    loaded = load_yaml(Path(temp_name))

    assert "neurodata_type_def" in loaded["groups"][0]
    assert "data_type_def" not in loaded["groups"][0]
    assert "neurodata_type_inc" in loaded["groups"][0]
    assert "data_type_inc" not in loaded["groups"][0]
    assert "neurodata_type_inc" in loaded["groups"][0]["datasets"][0]
    assert "data_type_inc" not in loaded["groups"][0]["datasets"][0]

    os.remove(temp_name)