import pdb

import pytest
from ..fixtures import nwb_core_fixture
from nwb_schema_language import Dataset, Group
from nwb_linkml.adapters import SchemaAdapter

@pytest.mark.parametrize(
    ['class_name','schema_file','namespace_name'],
    [
        ('DynamicTable', 'table.yaml', 'hdmf-common'),
        ('Container', 'base.yaml', 'hdmf-common'),
        ('TimeSeries', 'nwb.base.yaml', 'core'),
        ('ImageSeries', 'nwb.image.yaml', 'core')
    ]
)
def test_find_type_source(nwb_core_fixture, class_name, schema_file, namespace_name):
    defining_sch = nwb_core_fixture.find_type_source(class_name)
    assert defining_sch.path.name == schema_file
    assert namespace_name == defining_sch.namespace


def test_populate_imports(nwb_core_fixture):
    nwb_core_fixture.populate_imports()
    schema: SchemaAdapter
    assert len(nwb_core_fixture.schemas) > 0
    for schema in nwb_core_fixture.schemas:
        need_imports = [nwb_core_fixture.find_type_source(cls.neurodata_type_def).namespace for cls in schema.created_classes if cls.neurodata_type_inc is not None]
        need_imports = [i for i in need_imports if i != schema.namespace]
        for i in need_imports:
            assert i in schema.imports




def test_build(nwb_core_fixture):
    pass

def test_skip_imports(nwb_core_fixture):
    """
    We can build just the namespace in question without also building the other namespaces that it imports
    """
    res = nwb_core_fixture.build(skip_imports=True)

    # we shouldn't have any of the hdmf-common schema in with us
    namespaces = [sch.annotations['namespace'].value for sch in res.schemas]
    assert all([ns == 'core' for ns in namespaces])