Fix fints account type check (#106082)

2025-12-24 12:59:34 +00:00 · 2024-01-02 09:59:13 +01:00
parent 7396bc61d7
commit b27e830997
4 changed files with 103 additions and 4 deletions
--- a/tests/components/fints/test_client.py
+++ b/tests/components/fints/test_client.py
@@ -0,0 +1,95 @@
+"""Tests for the FinTS client."""
+
+from typing import Optional
+
+from fints.client import BankIdentifier, FinTSOperations
+import pytest
+
+from homeassistant.components.fints.sensor import (
+    BankCredentials,
+    FinTsClient,
+    SEPAAccount,
+)
+
+BANK_INFORMATION = {
+    "bank_identifier": BankIdentifier(country_identifier="280", bank_code="50010517"),
+    "currency": "EUR",
+    "customer_id": "0815",
+    "owner_name": ["SURNAME, FIRSTNAME"],
+    "subaccount_number": None,
+    "supported_operations": {
+        FinTSOperations.GET_BALANCE: True,
+        FinTSOperations.GET_CREDIT_CARD_TRANSACTIONS: False,
+        FinTSOperations.GET_HOLDINGS: False,
+        FinTSOperations.GET_SCHEDULED_DEBITS_MULTIPLE: False,
+        FinTSOperations.GET_SCHEDULED_DEBITS_SINGLE: False,
+        FinTSOperations.GET_SEPA_ACCOUNTS: True,
+        FinTSOperations.GET_STATEMENT: False,
+        FinTSOperations.GET_STATEMENT_PDF: False,
+        FinTSOperations.GET_TRANSACTIONS: True,
+        FinTSOperations.GET_TRANSACTIONS_XML: False,
+    },
+}
+
+
+@pytest.mark.parametrize(
+    (
+        "account_number",
+        "iban",
+        "product_name",
+        "account_type",
+        "expected_balance_result",
+        "expected_holdings_result",
+    ),
+    [
+        ("GIRO1", "GIRO1", "Valid balance account", 5, True, False),
+        (None, None, "Invalid account", None, False, False),
+        ("GIRO2", "GIRO2", "Account without type", None, False, False),
+        ("GIRO3", "GIRO3", "Balance account from fallback", None, True, False),
+        ("DEPOT1", "DEPOT1", "Valid holdings account", 33, False, True),
+        ("DEPOT2", "DEPOT2", "Holdings account from fallback", None, False, True),
+    ],
+)
+async def test_account_type(
+    account_number: Optional[str],
+    iban: Optional[str],
+    product_name: str,
+    account_type: Optional[int],
+    expected_balance_result: bool,
+    expected_holdings_result: bool,
+) -> None:
+    """Check client methods is_balance_account and is_holdings_account."""
+    credentials = BankCredentials(
+        blz=1234, login="test", pin="0000", url="https://example.com"
+    )
+    account_config = {"GIRO3": True}
+    holdings_config = {"DEPOT2": True}
+
+    client = FinTsClient(
+        credentials=credentials,
+        name="test",
+        account_config=account_config,
+        holdings_config=holdings_config,
+    )
+
+    client._account_information_fetched = True
+    client._account_information = {
+        iban: BANK_INFORMATION
+        | {
+            "account_number": account_number,
+            "iban": iban,
+            "product_name": product_name,
+            "type": account_type,
+        }
+    }
+
+    sepa_account = SEPAAccount(
+        iban=iban,
+        bic="BANCODELTEST",
+        accountnumber=account_number,
+        subaccount=None,
+        blz="12345",
+    )
+
+    assert client.is_balance_account(sepa_account) == expected_balance_result
+    assert client.is_holdings_account(sepa_account) == expected_holdings_result