feat: leave balance calculation (congés/RTT)
This commit is contained in:
41
tests/test_leave_calc.py
Normal file
41
tests/test_leave_calc.py
Normal file
@@ -0,0 +1,41 @@
|
||||
from app.business.leave_calc import compute_leave_used, get_or_create_balance
|
||||
from app.models import WorkEntry, LeaveBalance
|
||||
from app import db
|
||||
from datetime import date
|
||||
import sqlalchemy as sa
|
||||
|
||||
|
||||
def test_compute_leave_used_conges(app):
|
||||
with app.app_context():
|
||||
entries = [
|
||||
WorkEntry(date=date(2025, 1, 6), day_type="CONGE"),
|
||||
WorkEntry(date=date(2025, 1, 7), day_type="CONGE"),
|
||||
WorkEntry(date=date(2025, 1, 8), day_type="RTT"),
|
||||
WorkEntry(date=date(2025, 1, 9), day_type="WORK"),
|
||||
]
|
||||
for e in entries:
|
||||
db.session.add(e)
|
||||
db.session.commit()
|
||||
|
||||
used = compute_leave_used(2025)
|
||||
assert used["conges"] == 2
|
||||
assert used["rtt"] == 1
|
||||
|
||||
|
||||
def test_get_or_create_balance_creates_default(app):
|
||||
with app.app_context():
|
||||
balance = get_or_create_balance(2025)
|
||||
assert balance.year == 2025
|
||||
assert balance.conges_total == 28
|
||||
assert balance.rtt_total == 18
|
||||
|
||||
|
||||
def test_get_or_create_balance_returns_existing(app):
|
||||
with app.app_context():
|
||||
existing = LeaveBalance(year=2025, conges_total=25, rtt_total=15)
|
||||
db.session.add(existing)
|
||||
db.session.commit()
|
||||
|
||||
balance = get_or_create_balance(2025)
|
||||
assert balance.conges_total == 25
|
||||
assert balance.rtt_total == 15
|
||||
Reference in New Issue
Block a user