from fatcat_openapi_client import ReleaseEntity

from fatcat_scholar.schema import *
from fatcat_scholar.api_entities import *
from fatcat_scholar.transform import *


def test_es_release_from_release() -> None:

    with open("tests/files/release_hsmo6p4smrganpb3fndaj2lon4.json", "r") as f:
        release = entity_from_json(f.read(), ReleaseEntity)

    obj = es_release_from_release(release)
    d = json.loads(obj.json())

    assert obj.ident == release.ident == d["ident"] == "hsmo6p4smrganpb3fndaj2lon4"
    assert obj.doi_registrar == "crossref"
    assert obj.doi_prefix == "10.7717"


def test_es_biblio_from_release() -> None:

    with open("tests/files/release_hsmo6p4smrganpb3fndaj2lon4.json", "r") as f:
        release = entity_from_json(f.read(), ReleaseEntity)

    obj = es_biblio_from_release(release)
    d = json.loads(obj.json())

    assert (
        obj.release_ident
        == release.ident
        == d["release_ident"]
        == "hsmo6p4smrganpb3fndaj2lon4"
    )


def test_run_refs() -> None:

    with open("tests/files/work_iarm6swodra2bcrzhxrfaah7py_bundle.json", "r") as f:
        run_refs(f.readlines())


def test_run_transform() -> None:

    with open("tests/files/work_iarm6swodra2bcrzhxrfaah7py_bundle.json", "r") as f:
        run_transform(f.readlines())

    with open("tests/files/sim_page_bundle.json", "r") as f:
        run_transform(f.readlines())


def test_biblio_metadata_hacks() -> None:
    biblio = ScholarBiblio(
        title="some example paper",
        contrib_names=["able seaperson"],
        release_year=2000,
        volume="10",
        issue="5",
        publisher="some university",
        issns=[],
        affiliations=[],
    )
    out = biblio_metadata_hacks(biblio)

    biblio.release_year = 2030
    out = biblio_metadata_hacks(biblio)
    assert out.release_year == None

    biblio.doi_prefix = "10.1101"
    biblio.container_name = None
    biblio.release_stage = None
    biblio.release_type = "post"
    out = biblio_metadata_hacks(biblio)
    assert out.container_name
    assert out.release_stage == "submitted"
    assert out.release_type == "article"