diff --git a/chrono/manager/templates/chrono/manager_shared_custody_agenda_month_view.html b/chrono/manager/templates/chrono/manager_shared_custody_agenda_month_view.html index 69bf1714..da686fb8 100644 --- a/chrono/manager/templates/chrono/manager_shared_custody_agenda_month_view.html +++ b/chrono/manager/templates/chrono/manager_shared_custody_agenda_month_view.html @@ -18,7 +18,7 @@ {% trans "Week" %} {{ week }} {% for slot in slots %} - {% if slot.date < agenda.date_start %} + {% if slot.date < agenda.date_start or agenda.date_end and slot.date > agenda.date_end %} {% else %} {{ slot }} diff --git a/tests/manager/test_shared_custody_agenda.py b/tests/manager/test_shared_custody_agenda.py index 24bc80eb..5d831c4d 100644 --- a/tests/manager/test_shared_custody_agenda.py +++ b/tests/manager/test_shared_custody_agenda.py @@ -216,7 +216,7 @@ def test_shared_custody_agenda_month_view(app, admin_user): @pytest.mark.freeze_time('2022-02-01 14:00') # Tuesday -def test_shared_custody_agenda_month_view_date_start(app, admin_user): +def test_shared_custody_agenda_month_view_dates(app, admin_user): father = Person.objects.create(user_external_id='father_id', first_name='John', last_name='Doe') mother = Person.objects.create(user_external_id='mother_id', first_name='Jane', last_name='Doe') child = Person.objects.create(user_external_id='child_id', first_name='James', last_name='Doe') @@ -248,11 +248,24 @@ def test_shared_custody_agenda_month_view_date_start(app, admin_user): new_days = [x.text for x in resp.pyquery('tbody tr th span')] assert days == new_days + # custody for only one day + agenda.date_end = datetime.date(2022, 2, 14) + agenda.save() + resp = app.get('/manage/shared-custody/%s/' % agenda.pk).follow() + assert resp.text.count('') == 34 + assert len([x.text for x in resp.pyquery('tbody tr td.guardian')]) == 1 + # month before date_start resp = resp.click('←') assert resp.text.count('') == 42 assert len([x.text for x in resp.pyquery('tbody tr td.guardian')]) == 0 + # month after date_end + resp = app.get('/manage/shared-custody/%s/' % agenda.pk).follow() + resp = resp.click('→') + assert resp.text.count('') == 35 + assert len([x.text for x in resp.pyquery('tbody tr td.guardian')]) == 0 + def test_shared_custody_agenda_month_view_queries(app, admin_user): father = Person.objects.create(user_external_id='father_id', first_name='John', last_name='Doe')