from aki_prj23_transparenzregister.utils.data_extraction.bundesanzeiger import (
    Bundesanzeiger,
)


def test_extract_auditor_company_no_hits() -> None:
    input_data = """
        Nothing to see here \O_O/
    """
    ba = Bundesanzeiger()
    result = ba.extract_auditor_company(input_data)
    assert result is None


def test_extract_auditor_company() -> None:
    company_name = "Korrupte Wirtschaftsprüfer GmbH & Co. KG"
    input_data = f"""
        <b>
            {company_name}
            <br>
            Max Mustermann
        </b>
    """
    ba = Bundesanzeiger()
    result = ba.extract_auditor_company(input_data)
    assert result == company_name