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]])