Added mortgate stuff

Author: jeeftor

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "simplefin4py"
-version = "0.0.8"
+version = "0.0.9"
 description = ""
 authors = ["Jeef <[email protected]>"]
 readme = "README.md"

simplefin4py/model.py

@@ -15,6 +15,7 @@ class AccountType(Enum):
     CREDIT_CARD = "mdi:credit-card"
     SAVINGS = "mdi:piggy-bank-outline"
     INVESTMENT = "mdi:chart-areaspline"
+    MORTGAGE = "mdi:home-city-outline"
     UNKNOWN = "mdi:cash"
 
 
@@ -102,14 +103,16 @@ def inferred_account_type(self) -> AccountType:
 
         if "savings" in account_name_lower:
             return AccountType.SAVINGS
-
+        if "mortgate" in account_name_lower:
+            return AccountType.MORTGAGE
         if "checking" in account_name_lower:
             return AccountType.CHECKING
 
         if (
             "visa" in account_name_lower
             or "mastercard" in account_name_lower
             or "credit" in account_name_lower
+            or "card" in account_name_lower
         ):
             return AccountType.CREDIT_CARD
 

tests/conftest.py

@@ -2,6 +2,8 @@
 import json
 import os
 from typing import Any
+
+import aiohttp
 import pytest
 from aioresponses import aioresponses
 
@@ -104,5 +106,10 @@ def mock_get_data_generic_error() -> Generator[aioresponses, None, None]:
     """Fixture for mocking a fech process with bad credentials/url."""
     with aioresponses() as m:
         # Mock a failed login attempt due to bad credentials.
-        m.get(MOCK_ACCESS_URL + "/accounts", status=200, body="<,asduB*J(_8asdji>")
+        m.get(
+            MOCK_ACCESS_URL + "/accounts",
+            status=200,
+            body="<,asduB*J(_8asdji>",
+            exception=aiohttp.ClientConnectorError,
+        )
         yield m

tests/fixtures/200-with-errors.json

@@ -158,7 +158,7 @@
         "url": "https://www.randombank2.com/"
       },
       "id": "ACT-7a8b9c0d-1e2f-3g4h-5i6j",
-      "name": "Leprecon Horn",
+      "name": "Castle Mortgage",
       "currency": "USD",
       "balance": "7500.00",
       "available-balance": "5000.00",