#93: reorganise Rerun, Diffusion, Schedule module (#95)

#93

Co-authored-by: bkfox <thomas bkfox net>
Reviewed-on: rc/aircox#95

aircox/tests/models/test_rerun.py

@@ -0,0 +1,117 @@
+from datetime import timedelta
+
+import pytest
+
+from django.core.exceptions import ValidationError
+
+# we use Schedule as concrete class (Rerun is abstract)
+from aircox.models import Schedule
+
+
+class TestRerunQuerySet:
+    @pytest.mark.django_db
+    def test_station_by_obj(self, stations, schedules):
+        for station in stations:
+            queryset = (
+                Schedule.objects.station(station)
+                .distinct()
+                .values_list("program__station", flat=True)
+            )
+            assert queryset.count() == 1
+            assert queryset.first() == station.pk
+
+    @pytest.mark.django_db
+    def test_station_by_id(self, stations, schedules):
+        for station in stations:
+            queryset = (
+                Schedule.objects.station(id=station.pk)
+                .distinct()
+                .values_list("program__station", flat=True)
+            )
+            assert queryset.count() == 1
+            assert queryset.first() == station.pk
+
+    @pytest.mark.django_db
+    def test_program_by_obj(self, programs, schedules):
+        for program in programs:
+            queryset = (
+                Schedule.objects.program(program)
+                .distinct()
+                .values_list("program", flat=True)
+            )
+            assert queryset.count() == 1
+            assert queryset.first() == program.pk
+
+    @pytest.mark.django_db
+    def test_program_by_id(self, programs, schedules):
+        for program in programs:
+            queryset = (
+                Schedule.objects.program(id=program.pk)
+                .distinct()
+                .values_list("program", flat=True)
+            )
+            assert queryset.count() == 1
+            assert queryset.first() == program.pk
+
+    @pytest.mark.django_db
+    def test_rerun(self, schedules):
+        queryset = Schedule.objects.rerun().values_list("initial", flat=True)
+        assert None not in queryset
+
+    @pytest.mark.django_db
+    def test_initial(self, schedules):
+        queryset = (
+            Schedule.objects.initial()
+            .distinct()
+            .values_list("initial", flat=True)
+        )
+        assert queryset.count() == 1
+        assert queryset.first() is None
+
+
+class TestRerun:
+    @pytest.mark.django_db
+    def test_is_initial_true(self, sched_initials):
+        assert all(r.is_initial for r in sched_initials)
+
+    @pytest.mark.django_db
+    def test_is_initial_false(self, sched_reruns):
+        assert all(not r.is_initial for r in sched_reruns)
+
+    @pytest.mark.django_db
+    def test_is_rerun_true(self, sched_reruns):
+        assert all(r.is_rerun for r in sched_reruns)
+
+    @pytest.mark.django_db
+    def test_is_rerun_false(self, sched_initials):
+        assert all(not r.is_rerun for r in sched_initials)
+
+    @pytest.mark.django_db
+    def test_get_initial_of_initials(self, sched_initials):
+        assert all(r.get_initial() is r for r in sched_initials)
+
+    @pytest.mark.django_db
+    def test_get_initial_of_reruns(self, sched_reruns):
+        assert all(r.get_initial() is r.initial for r in sched_reruns)
+
+    @pytest.mark.django_db
+    def test_clean_success(self, sched_reruns):
+        for rerun in sched_reruns:
+            rerun.clean()
+
+    @pytest.mark.django_db
+    def test_clean_fails(self, sched_reruns):
+        for rerun in sched_reruns:
+            rerun.time = (rerun.initial.start - timedelta(hours=2)).time()
+            with pytest.raises(ValidationError):
+                rerun.clean()
+
+    @pytest.mark.django_db
+    def test_save_rerun(self, sched_reruns):
+        for rerun in sched_reruns:
+            rerun.program = None
+            rerun.save_rerun()
+            assert rerun.program == rerun.initial.program
+
+    # TODO: save()
+    # save_initial is empty, thus not tested