diff --git a/tests/utils/data_extraction/bundesanzeiger_test.py b/tests/utils/data_extraction/bundesanzeiger_test.py
index 6c870ff..49dbb3f 100644
--- a/tests/utils/data_extraction/bundesanzeiger_test.py
+++ b/tests/utils/data_extraction/bundesanzeiger_test.py
@@ -1,11 +1,29 @@
+from unittest.mock import Mock
+
+import pandas as pd
+import pytest
+
+from aki_prj23_transparenzregister.models.company import FinancialKPIEnum
from aki_prj23_transparenzregister.utils.data_extraction.bundesanzeiger import (
Bundesanzeiger,
)
+@pytest.fixture()
+def mock_bundesanzeiger(mocker: Mock) -> Mock:
+ mock = Mock()
+ mocker.patch(
+ "deutschland.bundesanzeiger",
+ return_value=mock,
+ )
+ return mock
+
+
def test_extract_auditor_company_no_hits() -> None:
input_data = """
- Nothing to see here \O_O/
+
+ Nothing to see here
+
"""
ba = Bundesanzeiger()
result = ba.extract_auditor_company(input_data)
@@ -24,3 +42,55 @@ def test_extract_auditor_company() -> None:
ba = Bundesanzeiger()
result = ba.extract_auditor_company(input_data)
assert result == company_name
+
+
+def test_extract_kpis() -> None:
+ input_data = """
+ Die Prj23_Transparenzregister GmbH erwirtschaftete einen Jahresüberschuss 10.000,43 €.
+ Des Weiteren sanken die Gesamtverbindlichkeiten 42,00 €
+ """
+ ba = Bundesanzeiger()
+ result = ba.__extract_kpis__(input_data)
+
+ net_income = 10000.43
+ liabilities = 42.00
+
+ assert result[FinancialKPIEnum.NET_INCOME.value] == net_income
+ assert result[FinancialKPIEnum.LIABILITIES.value] == liabilities
+
+
+def test_extracct_financial_results() -> None:
+ input_data = """
+
+ Die Prj23_Transparenzregister GmbH erwirtschaftete einen Jahresüberschuss 10.000,43 €.
+
+