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')