pubmed-bibtex/tests/test_bibtex_entry_from_pmid.py

import unittest.mock
import urllib.error

import pytest

import pubmed_bibtex

# pylint: disable=wrong-import-order; false positive
from conftest import TEST_BIBTEX_ENTRY, TEST_PMID


def test_bibtex_entry_from_pmid() -> None:
    assert pubmed_bibtex.bibtex_entry_from_pmid(pmid=TEST_PMID) == TEST_BIBTEX_ENTRY


@unittest.mock.patch.object(
    pubmed_bibtex,
    "_TEXMED_URL_PATTERN",
    "https://www.bioinformatics.org/texmed/cgi-bin/42.cgi",
)
def test_bibtex_entry_from_pmid_not_found() -> None:
    with pytest.raises(urllib.error.HTTPError, match=r"^HTTP Error 404: Not Found$"):
        pubmed_bibtex.bibtex_entry_from_pmid(pmid=TEST_PMID)