mamweb/tests.py

48 lines
1.5 KiB
Python

from django.test import TestCase
import seminar.models as m
from vysledkovky.utils import VysledkovkaRocniku, VysledkovkaCisla
from .testutils import create_resitel
class VysledkovkaCislaTestCase(TestCase):
@classmethod
def setUpTestData(cls):
cls.rocnik1 = m.Rocnik.objects.create(prvni_rok=1, rocnik=1)
cls.rocnik25 = m.Rocnik.objects.create(prvni_rok=25, rocnik=25)
cls.rocnik26 = m.Rocnik.objects.create(prvni_rok=21, rocnik=26)
cls.cisla1 = [
m.Cislo.objects.create(rocnik=cls.rocnik1, poradi=i,)
for i in range(5)
]
m.Deadline.objects.create(cislo=cls.cisla1[0], verejna_vysledkovka=True)
m.Deadline.objects.create(cislo=cls.cisla1[0], verejna_vysledkovka=True)
cls.resitele = [create_resitel() for _ in range(10)]
cls.tema = m.Tema.objects.create(
rocnik=cls.rocnik1,
stav=m.Problem.STAV_ZADANY,
)
cls.uloha_tematu = m.Uloha.objects.create(
nadproblem=cls.tema,
stav=m.Problem.STAV_ZADANY,
)
cls.uloha = m.Uloha.objects.create(
stav=m.Problem.STAV_ZADANY,
)
cls.reseni = m.Reseni.objects.create()
# for resitel in cls.resitele[:5]:
# m.Reseni_Resitele(resitel=resitel, reseni=cls.reseni)
def test_cisla_rocniku(self):
vysledkovka_rocniku = VysledkovkaRocniku(self.rocnik1)
self.assertListEqual(vysledkovka_rocniku.cisla_rocniku, [self.cisla1[0]])