From aba9dab2c89dc1a3010e0473e61502d233c080a4 Mon Sep 17 00:00:00 2001 From: pampletousse Date: Mon, 28 Aug 2023 15:45:23 +0200 Subject: [PATCH] Update test with new working new_year calculation --- pod/video/tests/test_obsolescence.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pod/video/tests/test_obsolescence.py b/pod/video/tests/test_obsolescence.py index d0b7fa0fe9..101cbe0c42 100644 --- a/pod/video/tests/test_obsolescence.py +++ b/pod/video/tests/test_obsolescence.py @@ -127,9 +127,15 @@ def test_check_video_date_delete(self): self.assertEqual(video.date_delete, date1) video2 = Video.objects.get(id=2) + affiliation_years = [ + settings.ACCOMMODATION_YEARS.get(key) + for key in video2.owner.owner.affiliation + if key in settings.ACCOMMODATION_YEARS + ] + new_year = max(affiliation_years) if len(affiliation_years) else DEFAULT_YEAR_DATE_DELETE date2 = date( date.today().year - + settings.ACCOMMODATION_YEARS[video2.owner.owner.affiliation], + + new_year, date.today().month, date.today().day, )