import unittest
import os
import zipfile
import time
import string
import random

class TestSmucar(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        base, ext = os.path.splitext(os.path.basename(__file__))
        zf = zipfile.ZipFile(__file__ + ".zip", "a", zipfile.ZIP_DEFLATED)
        zf.write(__file__, "{}.{}{}".format(base, int(time.time()), ext))
        zf.close()

    def test_ogrevalna(self):
        dodaj("slalom", 5)
        dodaj("velesmuk", 8)
        dodaj("slalom", 12)
        s = random.choice(string.ascii_letters)
        dodaj(s, 42)
        self.assertEqual(tock("slalom"), 17)
        self.assertEqual(tock("velesmuk"), 8)
        self.assertEqual(tock(s), 42)
        self.assertEqual(tock("izmišljena disciplina"), 0)

    def test_obvezna(self):
        ana = Smucar()
        berta = Smucar()
        ana.dodaj("slalom", 5)
        ana.dodaj("velesmuk", 8)
        berta.dodaj("slalom", 12)
        ana.dodaj("slalom", 12)
        cilka = Smucar()
        s = random.choice(string.ascii_letters)
        cilka.dodaj(s, 42)

        self.assertEqual(ana.tock("slalom"), 17)
        self.assertEqual(ana.tock("velesmuk"), 8)
        self.assertEqual(ana.tock(s), 0)

        self.assertEqual(berta.tock("slalom"), 12)
        self.assertEqual(berta.tock("velesmuk"), 0)
        self.assertEqual(berta.tock(s), 0)

        self.assertEqual(cilka.tock("slalom"), 0)
        self.assertEqual(cilka.tock("velesmuk"), 0)
        self.assertEqual(cilka.tock(s), 42)