diff --git a/tests/utils/data_extraction/bundesanzeiger_test.py b/tests/utils/data_extraction/bundesanzeiger_test.py
index d764a52..73bbbd9 100644
--- a/tests/utils/data_extraction/bundesanzeiger_test.py
+++ b/tests/utils/data_extraction/bundesanzeiger_test.py
@@ -12,14 +12,36 @@ from aki_prj23_transparenzregister.utils.enum_types import FinancialKPIEnum
def test_extract_auditor_company_no_hits() -> None:
input_data = """
+ Mega GmbH
Nothing to see here
"""
ba = Bundesanzeiger()
- result = ba.extract_auditor_company(input_data)
+ result = ba.extract_auditor_company(input_data, "Mega GmbH")
assert result is None
+def test_extract_auditor_company_self_referencing() -> None:
+ company = "Mega GmbH"
+ auditor_company = "Super AG"
+
+ input_data = f"""
+
+ {company}
+
+ Nothing to see here
+
+
+ {auditor_company}
+
+ Nothing to see here
+
+ """
+ ba = Bundesanzeiger()
+ result = ba.extract_auditor_company(input_data, company)
+ assert result == auditor_company
+
+
def test_extract_auditor_company() -> None:
company_name = "Korrupte Wirtschaftsprüfer GmbH & Co. KG"
input_data = f"""
@@ -30,7 +52,7 @@ def test_extract_auditor_company() -> None:
"""
ba = Bundesanzeiger()
- result = ba.extract_auditor_company(input_data)
+ result = ba.extract_auditor_company(input_data, "Super AG")
assert result == company_name