wip loan update

backend/tests/test_crud.py

@@ -56,7 +56,9 @@ def make_exchange(session: Session, user_id: int, name: str = "NASDAQ") -> int:
     return cast("int", exchange.id)
 
 
-def make_cycle(session: Session, user_id: int, exchange_id: int, friendly_name: str = "Test Cycle") -> int:
+def make_cycle(
+    session: Session, user_id: int, exchange_id: int, friendly_name: str = "Test Cycle"
+) -> int:
     cycle = models.Cycles(
         user_id=user_id,
         friendly_name=friendly_name,
@@ -72,7 +74,9 @@ def make_cycle(session: Session, user_id: int, exchange_id: int, friendly_name:
     return cast("int", cycle.id)
 
 
-def make_trade(session: Session, user_id: int, cycle_id: int, friendly_name: str = "Test Trade") -> int:
+def make_trade(
+    session: Session, user_id: int, cycle_id: int, friendly_name: str = "Test Trade"
+) -> int:
     cycle: models.Cycles | None = session.get(models.Cycles, cycle_id)
     assert cycle is not None
     exchange_id = cycle.exchange_id
@@ -137,13 +141,17 @@ def _ensure_utc_aware(dt: datetime | None) -> datetime | None:
     return dt.astimezone(timezone.utc)
 
 
-def _validate_timestamp(actual: datetime, expected: datetime, tolerance: timedelta) -> None:
+def _validate_timestamp(
+    actual: datetime, expected: datetime, tolerance: timedelta
+) -> None:
     actual_utc = _ensure_utc_aware(actual)
     expected_utc = _ensure_utc_aware(expected)
     assert actual_utc is not None
     assert expected_utc is not None
     delta = abs(actual_utc - expected_utc)
-    assert delta <= tolerance, f"Timestamps differ by {delta}, which exceeds tolerance of {tolerance}"
+    assert delta <= tolerance, (
+        f"Timestamps differ by {delta}, which exceeds tolerance of {tolerance}"
+    )
 
 
 # Trades
@@ -470,7 +478,9 @@ def test_update_trade_friendly_name(session: Session) -> None:
     trade_id = make_trade(session, user_id, cycle_id)
 
     new_friendly_name = "Updated Trade Name"
-    updated_trade = crud.update_trade_friendly_name(session, trade_id, new_friendly_name)
+    updated_trade = crud.update_trade_friendly_name(
+        session, trade_id, new_friendly_name
+    )
     assert updated_trade is not None
     assert updated_trade.id == trade_id
     assert updated_trade.friendly_name == new_friendly_name
@@ -624,7 +634,9 @@ def test_get_cycles_by_user_id(session: Session) -> None:
 def test_update_cycle(session: Session) -> None:
     user_id = make_user(session)
     exchange_id = make_exchange(session, user_id)
-    cycle_id = make_cycle(session, user_id, exchange_id, friendly_name="Initial Cycle Name")
+    cycle_id = make_cycle(
+        session, user_id, exchange_id, friendly_name="Initial Cycle Name"
+    )
 
     update_data = {
         "friendly_name": "Updated Cycle Name",
@@ -646,14 +658,20 @@ def test_update_cycle(session: Session) -> None:
 def test_update_cycle_immutable_fields(session: Session) -> None:
     user_id = make_user(session)
     exchange_id = make_exchange(session, user_id)
-    cycle_id = make_cycle(session, user_id, exchange_id, friendly_name="Initial Cycle Name")
+    cycle_id = make_cycle(
+        session, user_id, exchange_id, friendly_name="Initial Cycle Name"
+    )
 
     # Attempt to update immutable fields
     update_data = {
         "id": cycle_id + 1,  # Trying to change the ID
         "user_id": user_id + 1,  # Trying to change the user_id
-        "start_date": datetime(2020, 1, 1, tzinfo=timezone.utc).date(),  # Trying to change start_date
-        "created_at": datetime(2020, 1, 1, tzinfo=timezone.utc),  # Trying to change created_at
+        "start_date": datetime(
+            2020, 1, 1, tzinfo=timezone.utc
+        ).date(),  # Trying to change start_date
+        "created_at": datetime(
+            2020, 1, 1, tzinfo=timezone.utc
+        ),  # Trying to change created_at
         "friendly_name": "Valid Update",  # Valid field to update
     }
 
@@ -685,7 +703,10 @@ def test_create_cycle_loan_event(session: Session) -> None:
     assert loan_event.id is not None
     assert loan_event.cycle_id == cycle_id
     assert loan_event.loan_amount_cents == loan_data["loan_amount_cents"]
-    assert loan_event.loan_interest_rate_tenth_bps == loan_data["loan_interest_rate_tenth_bps"]
+    assert (
+        loan_event.loan_interest_rate_tenth_bps
+        == loan_data["loan_interest_rate_tenth_bps"]
+    )
     assert loan_event.notes == loan_data["notes"]
     assert loan_event.effective_date == now.date()
     _validate_timestamp(loan_event.created_at, now, timedelta(seconds=1))
@@ -695,12 +716,41 @@ def test_create_cycle_loan_event(session: Session) -> None:
     assert actual_loan_event is not None
     assert actual_loan_event.cycle_id == cycle_id
     assert actual_loan_event.loan_amount_cents == loan_data["loan_amount_cents"]
-    assert actual_loan_event.loan_interest_rate_tenth_bps == loan_data["loan_interest_rate_tenth_bps"]
+    assert (
+        actual_loan_event.loan_interest_rate_tenth_bps
+        == loan_data["loan_interest_rate_tenth_bps"]
+    )
     assert actual_loan_event.notes == loan_data["notes"]
     assert actual_loan_event.effective_date == now.date()
     _validate_timestamp(actual_loan_event.created_at, now, timedelta(seconds=1))
 
 
+def test_create_cycle_loan_event_same_date_error(session: Session) -> None:
+    user_id = make_user(session)
+    exchange_id = make_exchange(session, user_id)
+    cycle_id = make_cycle(session, user_id, exchange_id)
+
+    loan_data_1 = {
+        "cycle_id": cycle_id,
+        "loan_amount_cents": 100000,
+        "loan_interest_rate_tenth_bps": 5000,
+        "effective_date": datetime(2023, 1, 1, tzinfo=timezone.utc).date(),
+        "notes": "First loan event",
+    }
+    loan_data_2 = {
+        "cycle_id": cycle_id,
+        "loan_amount_cents": 150000,
+        "loan_interest_rate_tenth_bps": 4500,
+        "effective_date": datetime(2023, 1, 1, tzinfo=timezone.utc).date(),
+        "notes": "Second loan event same date",
+    }
+
+    crud.create_cycle_loan_event(session, loan_data_1)
+    with pytest.raises(ValueError) as excinfo:
+        crud.create_cycle_loan_event(session, loan_data_2)
+    assert "create_cycle_loan_event integrity error" in str(excinfo.value)
+
+
 def test_get_cycle_loan_events_by_cycle_id(session: Session) -> None:
     user_id = make_user(session)
     exchange_id = make_exchange(session, user_id)
@@ -729,34 +779,77 @@ def test_get_cycle_loan_events_by_cycle_id(session: Session) -> None:
     notes = [event.notes for event in loan_events]
     assert loan_events[0].notes == loan_data_2["notes"]
     assert loan_events[0].effective_date == yesterday
-    assert notes == ["Second loan event", "First loan event"]  # Ordered by effective_date desc
+    assert notes == [
+        "Second loan event",
+        "First loan event",
+    ]  # Ordered by effective_date desc
 
 
-def test_get_cycle_loan_events_by_cycle_id_same_date(session: Session) -> None:
+def test_get_cycle_loan_event_by_cycle_id_and_effective_date(session: Session) -> None:
     user_id = make_user(session)
     exchange_id = make_exchange(session, user_id)
     cycle_id = make_cycle(session, user_id, exchange_id)
 
-    loan_data_1 = {
+    effective_date = datetime(2023, 1, 1, tzinfo=timezone.utc).date()
+    loan_data = {
         "cycle_id": cycle_id,
         "loan_amount_cents": 100000,
         "loan_interest_rate_tenth_bps": 5000,
-        "notes": "First loan event",
+        "effective_date": effective_date,
+        "notes": "Loan event for specific date",
     }
-    loan_data_2 = {
+
+    crud.create_cycle_loan_event(session, loan_data)
+    loan_event = crud.get_loan_event_by_cycle_id_and_effective_date(
+        session, cycle_id, effective_date
+    )
+    assert loan_event is not None
+    assert loan_event.cycle_id == cycle_id
+    assert loan_event.effective_date == effective_date
+    assert loan_event.notes == loan_data["notes"]
+
+
+def test_update_cycle_loan_event(session: Session) -> None:
+    user_id = make_user(session)
+    exchange_id = make_exchange(session, user_id)
+    cycle_id = make_cycle(session, user_id, exchange_id)
+
+    loan_data = {
         "cycle_id": cycle_id,
-        "loan_amount_cents": 150000,
-        "loan_interest_rate_tenth_bps": 4500,
-        "notes": "Second loan event",
+        "loan_amount_cents": 100000,
+        "loan_interest_rate_tenth_bps": 5000,
+        "notes": "Initial loan event",
     }
 
-    crud.create_cycle_loan_event(session, loan_data_1)
-    crud.create_cycle_loan_event(session, loan_data_2)
+    loan_event = crud.create_cycle_loan_event(session, loan_data)
+    assert loan_event is not None
 
-    loan_events = crud.get_loan_events_by_cycle_id(session, cycle_id)
-    assert len(loan_events) == 2
-    notes = [event.notes for event in loan_events]
-    assert notes == ["First loan event", "Second loan event"]  # Ordered by id desc when effective_date is same
+    update_data = {
+        "loan_amount_cents": 120000,
+        "loan_interest_rate_tenth_bps": 4500,
+        "notes": "Updated loan event",
+    }
+    event_id = loan_event.id or 0
+
+    updated_loan_event = crud.update_cycle_loan_event(session, event_id, update_data)
+    assert updated_loan_event is not None
+    assert updated_loan_event.id == loan_event.id
+    assert updated_loan_event.loan_amount_cents == update_data["loan_amount_cents"]
+    assert (
+        updated_loan_event.loan_interest_rate_tenth_bps
+        == update_data["loan_interest_rate_tenth_bps"]
+    )
+    assert updated_loan_event.notes == update_data["notes"]
+
+    session.refresh(updated_loan_event)
+    actual_loan_event = session.get(models.CycleLoanChangeEvents, loan_event.id)
+    assert actual_loan_event is not None
+    assert actual_loan_event.loan_amount_cents == update_data["loan_amount_cents"]
+    assert (
+        actual_loan_event.loan_interest_rate_tenth_bps
+        == update_data["loan_interest_rate_tenth_bps"]
+    )
+    assert actual_loan_event.notes == update_data["notes"]
 
 
 def test_create_cycle_loan_event_single_field(session: Session) -> None:
@@ -802,7 +895,12 @@ def test_create_cycle_daily_accrual(session: Session) -> None:
         "notes": "Daily interest accrual",
     }
 
-    accrual = crud.create_cycle_daily_accrual(session, cycle_id, accrual_data["accrual_date"], accrual_data["accrued_interest_cents"])
+    accrual = crud.create_cycle_daily_accrual(
+        session,
+        cycle_id,
+        accrual_data["accrual_date"],
+        accrual_data["accrued_interest_cents"],
+    )
     assert accrual.id is not None
     assert accrual.cycle_id == cycle_id
     assert accrual.accrual_date == accrual_data["accrual_date"]
@@ -835,8 +933,18 @@ def test_get_cycle_daily_accruals_by_cycle_id(session: Session) -> None:
         "accrued_interest_cents": 150,
     }
 
-    crud.create_cycle_daily_accrual(session, cycle_id, accrual_data_1["accrual_date"], accrual_data_1["accrued_interest_cents"])
-    crud.create_cycle_daily_accrual(session, cycle_id, accrual_data_2["accrual_date"], accrual_data_2["accrued_interest_cents"])
+    crud.create_cycle_daily_accrual(
+        session,
+        cycle_id,
+        accrual_data_1["accrual_date"],
+        accrual_data_1["accrued_interest_cents"],
+    )
+    crud.create_cycle_daily_accrual(
+        session,
+        cycle_id,
+        accrual_data_2["accrual_date"],
+        accrual_data_2["accrued_interest_cents"],
+    )
 
     accruals = crud.get_cycle_daily_accruals_by_cycle_id(session, cycle_id)
     assert len(accruals) == 2
@@ -863,18 +971,37 @@ def test_get_cycle_daily_accruals_by_cycle_id_and_date(session: Session) -> None
         "accrued_interest_cents": 150,
     }
 
-    crud.create_cycle_daily_accrual(session, cycle_id, accrual_data_1["accrual_date"], accrual_data_1["accrued_interest_cents"])
-    crud.create_cycle_daily_accrual(session, cycle_id, accrual_data_2["accrual_date"], accrual_data_2["accrued_interest_cents"])
+    crud.create_cycle_daily_accrual(
+        session,
+        cycle_id,
+        accrual_data_1["accrual_date"],
+        accrual_data_1["accrued_interest_cents"],
+    )
+    crud.create_cycle_daily_accrual(
+        session,
+        cycle_id,
+        accrual_data_2["accrual_date"],
+        accrual_data_2["accrued_interest_cents"],
+    )
 
-    accruals_today = crud.get_cycle_daily_accrual_by_cycle_id_and_date(session, cycle_id, today)
+    accruals_today = crud.get_cycle_daily_accrual_by_cycle_id_and_date(
+        session, cycle_id, today
+    )
     assert accruals_today is not None
     assert accruals_today.accrual_date == today
-    assert accruals_today.accrual_amount_cents == accrual_data_2["accrued_interest_cents"]
+    assert (
+        accruals_today.accrual_amount_cents == accrual_data_2["accrued_interest_cents"]
+    )
 
-    accruals_yesterday = crud.get_cycle_daily_accrual_by_cycle_id_and_date(session, cycle_id, yesterday)
+    accruals_yesterday = crud.get_cycle_daily_accrual_by_cycle_id_and_date(
+        session, cycle_id, yesterday
+    )
     assert accruals_yesterday is not None
     assert accruals_yesterday.accrual_date == yesterday
-    assert accruals_yesterday.accrual_amount_cents == accrual_data_1["accrued_interest_cents"]
+    assert (
+        accruals_yesterday.accrual_amount_cents
+        == accrual_data_1["accrued_interest_cents"]
+    )
 
 
 # Exchanges
@@ -1031,7 +1158,9 @@ def test_update_user_immutable_fields(session: Session) -> None:
     update_data = {
         "id": user_id + 1,  # Trying to change the ID
         "username": "newusername",  # Trying to change the username
-        "created_at": datetime(2020, 1, 1, tzinfo=timezone.utc),  # Trying to change created_at
+        "created_at": datetime(
+            2020, 1, 1, tzinfo=timezone.utc
+        ),  # Trying to change created_at
         "password_hash": "validupdate",  # Valid field to update
     }
 
@@ -1065,7 +1194,9 @@ def test_create_login_session_with_invalid_user(session: Session) -> None:
 def test_get_login_session_by_token_and_user_id(session: Session) -> None:
     now = datetime.now(timezone.utc)
     created_session = make_login_session(session, now)
-    fetched_session = crud.get_login_session_by_token_hash_and_user_id(session, created_session.session_token_hash, created_session.user_id)
+    fetched_session = crud.get_login_session_by_token_hash_and_user_id(
+        session, created_session.session_token_hash, created_session.user_id
+    )
     assert fetched_session is not None
     assert fetched_session.id == created_session.id
     assert fetched_session.user_id == created_session.user_id
@@ -1075,7 +1206,9 @@ def test_get_login_session_by_token_and_user_id(session: Session) -> None:
 def test_get_login_session_by_token(session: Session) -> None:
     now = datetime.now(timezone.utc)
     created_session = make_login_session(session, now)
-    fetched_session = crud.get_login_session_by_token_hash(session, created_session.session_token_hash)
+    fetched_session = crud.get_login_session_by_token_hash(
+        session, created_session.session_token_hash
+    )
     assert fetched_session is not None
     assert fetched_session.id == created_session.id
     assert fetched_session.user_id == created_session.user_id
@@ -1090,9 +1223,13 @@ def test_update_login_session(session: Session) -> None:
         "last_seen_at": now + timedelta(hours=1),
         "last_used_ip": "192.168.1.1",
     }
-    updated_session = crud.update_login_session(session, created_session.session_token_hash, update_data)
+    updated_session = crud.update_login_session(
+        session, created_session.session_token_hash, update_data
+    )
     assert updated_session is not None
-    assert _ensure_utc_aware(updated_session.last_seen_at) == update_data["last_seen_at"]
+    assert (
+        _ensure_utc_aware(updated_session.last_seen_at) == update_data["last_seen_at"]
+    )
     assert updated_session.last_used_ip == update_data["last_used_ip"]
 
 
@@ -1101,5 +1238,7 @@ def test_delete_login_session(session: Session) -> None:
     created_session = make_login_session(session, now)
 
     crud.delete_login_session(session, created_session.session_token_hash)
-    deleted_session = crud.get_login_session_by_token_hash_and_user_id(session, created_session.session_token_hash, created_session.user_id)
+    deleted_session = crud.get_login_session_by_token_hash_and_user_id(
+        session, created_session.session_token_hash, created_session.user_id
+    )
     assert deleted_session is None