from grobid_tei_xml import parse_document_xml


def test_small_xml_csl() -> None:

    with open('tests/files/small.xml', 'r') as f:
        tei_xml = f.read()

    d = parse_document_xml(tei_xml)
    assert d.to_csl_dict() == {
        "type": "article-journal",
        "title": "Dummy Example File",
        "author": [
            {
                "given": "Brewster",
                "family": "Kahle"
            },
            {
                "given": "J",
                "family": "Doe",
            },
        ],
        "container-title": "Dummy Example File. Journal of Fake News. pp. 1-2. ISSN 1234-5678",
        "issued": [[2000]],
    }

    assert d.citations
    assert d.citations[0].to_csl_dict() == {
        "type": "article-journal",
        "title": "Everything is Wonderful",
        "author": [
            {
                "given": "A",
                "family": "Seaperson"
            },
        ],
        "container-title": "Letters in the Alphabet",
        "issued": [[2001]],
        "volume": 20,
        "page": "1-11",
    }