baby/tests/test_calendar.py

import datetime

import pytest
from django.contrib.auth.models import User

from baby.app.common.calendar import MonthCalendar
from baby.app.models.regle import Regle


@pytest.mark.django_db
class TestCalendar:

    def test_nominal_case(self):
        user = User.objects.create_user("user")
        ref = datetime.date(2023, 9, 2)

        times = [
            -28 - 25 - 26,
            -26 - 25,
            -25,
            0
        ]
        for i, x in enumerate(times, start=1):
            Regle.objects.create(user=user, date=ref + datetime.timedelta(days=x))


        cal = MonthCalendar(user, 11, 23)
        data = cal.get_calendar_dict()
        js = cal.json
        print(data)