Category overhaul (#20)

Reviewed-on: #20
Co-authored-by: Mathias <mathias2784@web.de>

tests/test_models_init.py

@@ -0,0 +1,110 @@
+"""Tests for src.models.__init__.py lazy loading and __dir__."""
+
+import pytest
+
+from src import models
+
+
+class TestLazyLoading:
+    """Tests for lazy attribute loading via __getattr__."""
+
+    def test_load_b4wexport(self):
+        """Test loading B4wExport model."""
+        from src.models.b4w import B4wExport as DirectB4wExport
+
+        b4w = models.B4wExport
+        assert b4w is DirectB4wExport
+
+    def test_load_category(self):
+        """Test loading Category model."""
+        from src.models.category import Category as DirectCategory
+
+        cat = models.Category
+        assert cat is DirectCategory
+
+    def test_load_categorybudgetdb(self):
+        """Test loading CategoryBudgetDb model."""
+        from src.models.category import CategoryBudgetDb as DirectCategoryBudgetDb
+
+        cbdb = models.CategoryBudgetDb
+        assert cbdb is DirectCategoryBudgetDb
+
+    def test_load_comdirectgiro(self):
+        """Test loading ComdirectGiro model."""
+        from src.models.comdirect import ComdirectGiro as DirectComdirectGiro
+
+        giro = models.ComdirectGiro
+        assert giro is DirectComdirectGiro
+
+    def test_load_transaction(self):
+        """Test loading Transaction model."""
+        from src.models.transactions import Transaction as DirectTransaction
+
+        tx = models.Transaction
+        assert tx is DirectTransaction
+
+    def test_load_account(self):
+        """Test loading Account model."""
+        from src.models.settings import Account as DirectAccount
+
+        acc = models.Account
+        assert acc is DirectAccount
+
+    def test_load_source(self):
+        """Test loading Source model."""
+        from src.models.settings import Source as DirectSource
+
+        src = models.Source
+        assert src is DirectSource
+
+    def test_nonexistent_attribute_raises(self):
+        """Test that non-existent attributes raise AttributeError."""
+        with pytest.raises(AttributeError) as exc_info:
+            # This is intentional: testing that accessing a non-existent attribute raises
+            _ = models.NonExistentModel  # type: ignore[name-defined]
+
+        expected_msg = "module 'src.models' has no attribute 'NonExistentModel'"
+        assert expected_msg in str(exc_info.value)
+
+
+class TestDirMethod:
+    """Tests for __dir__() method."""
+
+    def test_dir_returns_sorted_list(self):
+        """Test that __dir__() returns a sorted list."""
+        result = models.__dir__()
+        assert isinstance(result, list)
+        # Should be alphabetically sorted
+        assert result == sorted(result)
+
+    def test_dir_includes_all_exports(self):
+        """Test that __dir__() includes all exported models."""
+        result = models.__dir__()
+
+        expected_exports = {
+            "B4wExport",
+            "Category",
+            "CategoryBudgetDb",
+            "ComdirectCreditCard",
+            "ComdirectDepot",
+            "ComdirectGiro",
+            "Account",
+            "Source",
+            "Transaction",
+            "TransactionDb",
+        }
+
+        # All exports should be in dir()
+        for export_name in expected_exports:
+            assert export_name in result
+
+    def test_dir_includes_globals(self):
+        """Test that __dir__() includes module-level globals."""
+        result = models.__dir__()
+        assert "_EXPORTS" in result
+
+    def test_dir_is_alphabetically_sorted(self):
+        """Verify the list is truly sorted alphabetically."""
+        result = models.__dir__()
+        # Double-check: if we sort it again, it should be identical
+        assert result == sorted(result)