Тесты и инфраструктура
- tests/: 104 теста — lessons (все форматы ячеек), xlsx (мержи/скрытые столбцы/красный текст/даты), parsing (группы/недели/индексы), API (TestClient, TTL/modified/stale-while-error, 404/400/502) - xlsx_builder.py: генерация xlsx в памяти для тестов без openpyxl - docker-compose: healthz-хелскчеки (не блокируют старт), SCHEDULE_LINK - Makefile: target test - requirements: закреплённые версии - .gitignore: __pycache__/, .venv/ и т.д.
This commit is contained in:
111
tests/fixtures.py
Normal file
111
tests/fixtures.py
Normal file
@@ -0,0 +1,111 @@
|
||||
"""Общая тестовая таблица: 2 группы, 2 дня, мержи, скрытый столбец, красный текст."""
|
||||
|
||||
from xlsx_builder import SheetSpec, build_workbook
|
||||
|
||||
MAIN = "Бакалавры и магистры"
|
||||
ROOMS = "аудитории"
|
||||
TEACHERS = "Преподаватели"
|
||||
|
||||
# серии Excel
|
||||
# недели ч: 08.31-09.06, 14.09-20.09, 28.09-04.10, 12.10-18.10
|
||||
# недели з: 07.09-13.09, 21.09-27.09, 05.10-11.10, 19.10-25.10
|
||||
W1 = (46265, 46271, 46272, 46278)
|
||||
W2 = (46279, 46285, 46286, 46292)
|
||||
W3 = (46293, 46299, 46300, 46306)
|
||||
W4 = (46307, 46313, 46314, 46320)
|
||||
|
||||
|
||||
def main_sheet() -> SheetSpec:
|
||||
g1 = "ИВТ-11БО\nсеместр до 11.01.27"
|
||||
g2 = "ИТ-12БО\nсеместр до 29.12.26"
|
||||
hidden = "СКРЫ-11БО\nсеместр до 1.1.27"
|
||||
math_red = [
|
||||
("Математика\nСмирнов А.В.\nауд. 215\n", False),
|
||||
("пара 14.09 не будет", True),
|
||||
]
|
||||
rows = [
|
||||
# 0: заголовок
|
||||
["числитель", "", "знаменатель", "", "день недели", "пара\nвремя",
|
||||
g1, g2, hidden, None],
|
||||
# 1: с/по
|
||||
["с", "по", "с", "по", "день недели", "пара\nвремя", None, None, None, None],
|
||||
# 2: понедельник, пара 1, ч
|
||||
[*W1, "понедельник", "1\n9:00-\n10:35", math_red, "Физика\nИванов И.И.\nауд. 301"],
|
||||
# 3: понедельник, пара 1, з
|
||||
[*W2, "", "", "Математика\nСмирнов А.В.\nауд. 216", None],
|
||||
# 4: понедельник, пара 2, ч
|
||||
[*W3, "", "2\n10:45- 12:20", "к.в. - 2\nФизика\nИванов И.И.\nауд. 301",
|
||||
"Химия\nПетров П.П.\nонлайн"],
|
||||
# 5: понедельник, пара 2, з
|
||||
[*W4, "", "", None, "к.в.\nХимия\nПетров П.П.\nауд. 201"],
|
||||
# 6: вторник, пара 1, ч
|
||||
[*W1, "вторник", "1\n9:00-\n10:35", "Биология\nСидорова С.С.\nауд. 220", None],
|
||||
# 7: вторник, пара 1, з
|
||||
[*W2, "", "", None, None],
|
||||
]
|
||||
return SheetSpec(
|
||||
rows=rows,
|
||||
merges=["F3:F4", "F5:F6", "F7:F8"],
|
||||
hidden_cols=[(8, 8)],
|
||||
)
|
||||
|
||||
|
||||
def rooms_sheet() -> SheetSpec:
|
||||
rows = [
|
||||
[None, None, None, None, None, None, "215", "301", "220"],
|
||||
[None, None, None, None, None, None, "комп.класс, 12 мест",
|
||||
"интерактивная панель, 24 места", "20 мест"],
|
||||
]
|
||||
return SheetSpec(rows=rows)
|
||||
|
||||
|
||||
def teachers_sheet() -> SheetSpec:
|
||||
rows = [
|
||||
["числитель", "числитель", "знаменатель", "знаменатель", "день недели", "пара\nвремя"],
|
||||
[None, None, None, None, None, None,
|
||||
"Смирнов\nАлександр Викторович", "Иванов\nИван Иванович"],
|
||||
[*W1, "понедельник", "1\n9:00-\n10:35",
|
||||
"Математика\nСмирнов А.В.\nауд. 215", "Физика\nИванов И.И.\nауд. 301"],
|
||||
]
|
||||
return SheetSpec(rows=rows)
|
||||
|
||||
|
||||
def fixture_sheets() -> dict[str, SheetSpec]:
|
||||
return {MAIN: main_sheet(), ROOMS: rooms_sheet(), TEACHERS: teachers_sheet()}
|
||||
|
||||
|
||||
def fixture_xlsx() -> bytes:
|
||||
return build_workbook(fixture_sheets())
|
||||
|
||||
|
||||
def make_sheets(sheets: dict[str, SheetSpec]):
|
||||
"""SheetSpec-фикстуры -> те же объекты, что парсит parse_workbook."""
|
||||
from schedule.xlsx import parse_workbook
|
||||
|
||||
return parse_workbook(build_workbook(sheets))
|
||||
|
||||
|
||||
def make_cache(monkeypatch, xlsx: bytes | None, modified: str = "fixed-modified"):
|
||||
"""ScheduleCache с заменёнными сетевыми функциями."""
|
||||
import schedule.cache as cache_mod
|
||||
from schedule import ScheduleCache
|
||||
|
||||
calls = {"download": 0, "modified": 0}
|
||||
fail = {"download": False, "modified": False}
|
||||
|
||||
def fake_download(url, timeout=60.0):
|
||||
calls["download"] += 1
|
||||
if fail["download"]:
|
||||
from schedule import ScheduleError
|
||||
|
||||
raise ScheduleError("сетевой сбой (тест)")
|
||||
return xlsx
|
||||
|
||||
def fake_modified(url, timeout=20.0):
|
||||
calls["modified"] += 1
|
||||
return None if fail["modified"] else modified
|
||||
|
||||
monkeypatch.setattr(cache_mod, "download_schedule", fake_download)
|
||||
monkeypatch.setattr(cache_mod, "fetch_table_modified", fake_modified)
|
||||
cache = ScheduleCache("test-link", ttl_seconds=3600, retry_after=60)
|
||||
return cache, calls, fail
|
||||
Reference in New Issue
Block a user