From 600039207df8dca9169da248011edfdabebc1f33 Mon Sep 17 00:00:00 2001
From: TrisNol <tristan.nolde@yahoo.de>
Date: Tue, 17 Oct 2023 18:16:44 +0200
Subject: [PATCH] test(data-extraction): Adapt unit tests to new behaviour

---
 .../data_extraction/bundesanzeiger_test.py    | 26 +++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

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 = """
     <b>
+        Mega GmbH
         Nothing to see here
     </b>
     """
     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"""
+    <b>
+        {company}
+        <br>
+        Nothing to see here
+    </b>
+    <b>
+        {auditor_company}
+        <br>
+        Nothing to see here
+    </b>
+    """
+    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:
         </b>
     """
     ba = Bundesanzeiger()
-    result = ba.extract_auditor_company(input_data)
+    result = ba.extract_auditor_company(input_data, "Super AG")
     assert result == company_name