from datetime import datetime

from sqlalchemy.orm import Session

from dibbler.models import Transaction, User
from dibbler.models.Transaction import (
    DEFAULT_PENALTY_MULTIPLIER_PERCENT,
    DEFAULT_PENALTY_THRESHOLD,
)
from dibbler.queries import current_penalty
from tests.helpers import assign_times, assert_id_order_similar_to_time_order


def test_current_penalty_no_history(sql_session: Session) -> None:
    assert current_penalty(sql_session) == (
        DEFAULT_PENALTY_THRESHOLD,
        DEFAULT_PENALTY_MULTIPLIER_PERCENT,
    )


def test_current_penalty_with_history(sql_session: Session) -> None:
    user = User("Admin User")
    sql_session.add(user)
    sql_session.commit()

    transactions = [
        Transaction.adjust_penalty(
            penalty_threshold=-200,
            penalty_multiplier_percent=150,
            user_id=user.id,
        ),
        Transaction.adjust_penalty(
            penalty_threshold=-300,
            penalty_multiplier_percent=200,
            user_id=user.id,
        ),
    ]

    assign_times(transactions)

    sql_session.add_all(transactions)
    sql_session.commit()

    assert_id_order_similar_to_time_order(transactions)

    assert current_penalty(sql_session) == (-300, 200)