rewrite tests + fix error in schedule generator

programs/tests.py

@@ -1,79 +1,66 @@
 import datetime
+import calendar
+import logging
+from dateutil.relativedelta import relativedelta
 
 from django.test import TestCase
 from django.utils import timezone as tz
 
 from aircox.programs.models import *
 
+logger = logging.getLogger('aircox.test')
+logger.setLevel('INFO')
 
-class Programs (TestCase):
-    def setUp (self):
-        stream = Stream.objects.get_or_create(
-            name = 'diffusions',
-            defaults = { 'type': Stream.Type.schedule }
-        )[0]
-        Program.objects.create(name = 'source', stream = stream)
-        Program.objects.create(name = 'microouvert', stream = stream)
+class ScheduleCheck (TestCase):
+    def setUp(self):
+        self.schedules = [
+            Schedule(
+                date = tz.now(),
+                duration = datetime.time(1,30),
+                frequency = frequency,
+            )
+            for frequency in Schedule.Frequency.__members__.values()
+        ]
 
-        self.schedules = {}
-        self.programs = {}
+    def test_frequencies(self):
+        for schedule in self.schedules:
+            logger.info('- test frequency %s' % schedule.get_frequency_display())
+            date = schedule.date
+            count = 24
+            while count:
+                logger.info('- month %(month)s/%(year)s' % {
+                    'month': date.month,
+                    'year': date.year
+                })
+                count -= 1
+                dates = schedule.dates_of_month(date)
+                if schedule.frequency == schedule.Frequency.one_on_two:
+                    self.check_one_on_two(schedule, date, dates)
+                elif schedule.frequency == schedule.Frequency.last:
+                    self.check_last(schedule, date, dates)
+                else:
+                    pass
+                date += relativedelta(months = 1)
 
-    def test_create_programs_schedules (self):
-        program = Program.objects.get(name = 'source')
+    def check_one_on_two(self, schedule, date, dates):
+        for date in dates:
+            delta = date.date() - schedule.date.date()
+            self.assertEqual(delta.days % 14, 0)
 
-        sched_0 = self.create_schedule(program, 'one on two', [
-                tz.datetime(2015, 10, 2, 18),
-                tz.datetime(2015, 10, 16, 18),
-                tz.datetime(2015, 10, 30, 18),
-            ]
-        )
-        sched_1 = self.create_schedule(program, 'one on two', [
-                tz.datetime(2015, 10, 5, 18),
-                tz.datetime(2015, 10, 19, 18),
-            ],
-            rerun = sched_0
-        )
+    def check_last(self, schedule, date, dates):
+        month_info = calendar.monthrange(date.year, date.month)
+        date = datetime.date(date.year, date.month, month_info[1])
 
-        self.programs[program.pk] = program
+        # end of month before the wanted weekday: move one week back
+        if date.weekday() < schedule.date.weekday():
+            date -= datetime.timedelta(days = 7)
 
-        program = Program.objects.get(name = 'microouvert')
-        # special case with november first week starting on sunday
-        sched_2 = self.create_schedule(program, 'first and third', [
-                tz.datetime(2015, 11, 6, 18),
-                tz.datetime(2015, 11, 20, 18),
-            ],
-            date = tz.datetime(2015, 10, 23, 18),
-        )
+        date -= datetime.timedelta(days = date.weekday())
+        date += datetime.timedelta(days = schedule.date.weekday())
+        self.assertEqual(date, dates[0].date())
+
+    def check_n_of_week(self, schedule, date, dates):
+        pass
 
-    def create_schedule (self, program, frequency, dates, date = None, rerun = None):
-        frequency = Schedule.Frequency[frequency]
-        schedule = Schedule(
-            program = program,
-            frequency = frequency,
-            date = date or dates[0],
-            rerun = rerun,
-            duration = datetime.time(1, 30)
-        )
-        print(schedule.__dict__)
-        schedule.save()
-
-        self.schedules[schedule.pk] = (schedule, dates)
-        return schedule
-
-    def test_check_schedule (self):
-        for schedule, dates in self.schedules:
-            dates = [ tz.make_aware(date) for date in dates ]
-            dates.sort()
-
-            # dates
-            dates_ = schedule.dates_of_month(dates[0])
-            dates_.sort()
-            self.assertEqual(dates_, dates)
-
-            # diffusions
-            dates_ = schedule.diffusions_of_month(dates[0])
-            dates_ = [date_.date for date_ in dates_]
-            dates_.sort()
-            self.assertEqual(dates_, dates)