from pathlib import Path

from src.models import ComdirectCreditCard, ComdirectDepot, ComdirectGiro
from src.reader import COMDIRECT_ACCOUNTS, reader_comdirect

TEST_DATA_PATH = ".local/data/comdirect/umsaetze_9773063005_20260906-2043.csv"


def test_reader_comdirect():
    test_path = Path(TEST_DATA_PATH)
    assert test_path.exists()
    data = reader_comdirect(test_path)
    assert data
    assert len(data) == len(COMDIRECT_ACCOUNTS)
    assert set(data) == set(COMDIRECT_ACCOUNTS)
    assert all(isinstance(row, ComdirectGiro) for row in data["Girokonto"])
    assert all(
        isinstance(row, ComdirectCreditCard) for row in data["Visa-Karte (Kreditkarte)"]
    )
    assert all(isinstance(row, ComdirectGiro) for row in data["Tagesgeld PLUS-Konto"])
    assert all(isinstance(row, ComdirectDepot) for row in data["Depot"])