import unittest
from random import randint

class Test(unittest.TestCase):
    def setUp(self):
        self.odredbe = [
            "Prepovedano: gostilne, fakultete",
            "Prepovedano: prehajanje med regijami, nočni sprehodi, javni promet",
            "Dovoljeno: fakultete",
            "Prepovedano: tek na prostem",
            "Dovoljeno: fakultete",
            "Dovoljeno: nočni sprehodi, tek na prostem",
            "Prepovedano: fakultete, nočni sprehodi, frizer",
            "Dovoljeno: frizer",
            "Prepovedano: fakultete, zobozdravnik, tek na prostem",
            "Dovoljeno: tek na prostem, javni promet"]

        self.d1 = str(randint(1000, 2000))
        self.d2 = str(randint(1000, 2000))
        self.nakljucne = [
            f"Prepovedano: {self.d1}, {self.d2}",
            f"Prepovedano: {self.d1}",
            f"Dovoljeno: {self.d2}"
        ]

    def test_je_dovoljeno(self):
        self.assertTrue(je_dovoljeno(self.odredbe, "tek na prostem"))
        self.assertTrue(je_dovoljeno(self.odredbe, "frizer"))
        self.assertFalse(je_dovoljeno(self.odredbe, "fakultete"))
        self.assertFalse(je_dovoljeno(self.odredbe, "zobozdravnik"))
        self.assertFalse(je_dovoljeno(self.odredbe, "nočni sprehodi"))
        self.assertTrue(je_dovoljeno(self.odredbe, "pitje čaja"))

        self.assertTrue(je_dovoljeno(self.nakljucne, self.d2))
        self.assertFalse(je_dovoljeno(self.nakljucne, self.d1))
        self.assertTrue(je_dovoljeno(self.nakljucne, "pitje čaja"))

    def test_najnevarnejse(self):
        self.assertEqual("fakultete", najnevarnejse(self.odredbe))
        self.assertEqual(self.d1, najnevarnejse(self.nakljucne))

    def test_tezki_casi(self):
        self.assertEqual(7, tezki_casi(self.odredbe))
        self.assertEqual(2, tezki_casi(self.nakljucne))


if __name__ == "__main__":
    unittest.main()