Files
college-infos/tests/unit/test_pronote_lessons.py
T

78 lines
2.3 KiB
Python

"""Tests des règles communes de statut et de consolidation des cours."""
from __future__ import annotations
from datetime import datetime
import pytest
from pronote_sync.models.agenda import Lesson, LessonStatus
from pronote_sync.sources.pronote.lessons import collapse_replaced_lessons, lesson_status
def _lesson(status: LessonStatus, room: str) -> Lesson:
"""Construit un cours minimal pour les tests de consolidation.
:param status: Statut métier du cours.
:param room: Salle du cours.
:return: Cours de test.
:rtype: Lesson
"""
return Lesson(
id=f"lesson-{room}",
start=datetime(2026, 9, 11, 8, 0),
end=datetime(2026, 9, 11, 9, 0),
subject="Mathématiques",
teachers=("Enseignant A",),
rooms=(room,),
group="Classe entière",
status=status,
content=None,
)
@pytest.mark.parametrize(
("raw_status", "canceled", "expected"),
[
("Changement de salle", False, LessonStatus.MOVED),
("Cours déplacé", False, LessonStatus.MOVED),
("Cours annulé", False, LessonStatus.CANCELLED),
(None, True, LessonStatus.CANCELLED),
("Statut inconnu", False, LessonStatus.NORMAL),
],
)
def test_lesson_status_maps_pronotepy_variants(
raw_status: object, canceled: bool, expected: LessonStatus
) -> None:
"""Convertit les statuts Pronote connus sans interpréter les inconnus.
:param raw_status: Statut brut simulé.
:param canceled: Indicateur d'annulation simulé.
:param expected: Statut métier attendu.
:return: None
"""
assert lesson_status(raw_status, canceled) == expected
def test_collapse_replaced_lessons_keeps_effective_lesson() -> None:
"""Retire l'ancien exemplaire annulé d'un changement de salle.
:return: None
"""
cancelled = _lesson(LessonStatus.CANCELLED, "101")
moved = _lesson(LessonStatus.MOVED, "204")
result = collapse_replaced_lessons([cancelled, moved])
assert result == [moved]
def test_collapse_replaced_lessons_keeps_real_cancellation() -> None:
"""Conserve une annulation lorsqu'aucun cours déplacé ne la remplace.
:return: None
"""
cancelled = _lesson(LessonStatus.CANCELLED, "101")
assert collapse_replaced_lessons([cancelled]) == [cancelled]