forked from rc/aircox
118 lines
3.7 KiB
Python
118 lines
3.7 KiB
Python
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
|