import unittest


class Test01Obvezna(unittest.TestCase):
    def test_dodaj_stranko(self):
        banka = []
        self.assertIsNone(dodaj_stranko(banka, "Ana", 42))
        self.assertEqual(banka, [["Ana", 42]])
        dodaj_stranko(banka, "Berta", 13)
        self.assertEqual(banka, [["Ana", 42], ["Berta", 13]])
        dodaj_stranko(banka, "Cilka", 66)
        self.assertEqual(banka, [["Ana", 42], ["Berta", 13], ["Cilka", 66]])

    def test_koliko_ima(self):
        banka = [["Ana", 42], ["Berta", 13], ["Cilka", 66]]
        self.assertEqual(koliko_ima(banka, "Ana"), 42)
        self.assertEqual(koliko_ima(banka, "Berta"), 13)
        self.assertEqual(koliko_ima(banka, "Cilka"), 66)
        self.assertIsNone(koliko_ima(banka, "Dani"))

        self.assertIsNone(koliko_ima([["Ana", 42]], "Dani"))
        self.assertIsNone(koliko_ima([], "Dani"))

    def test_bilanca(self):
        self.assertEqual(bilanca([["Ana", 42], ["Berta", 13], ["Cilka", 66]]), 121)
        self.assertEqual(bilanca([["Cilka", 66]]), 66)
        self.assertEqual(bilanca([]), 0)

    def test_najbogatejsi(self):
        self.assertEqual(najbogatejsi([["Ana", 42], ["Berta", 13], ["Cilka", 66]]), "Cilka")
        self.assertEqual(najbogatejsi([["Ana", 142], ["Berta", 13], ["Cilka", 66]]), "Ana")
        self.assertEqual(najbogatejsi([["Ana", 42], ["Berta", 113], ["Cilka", 66]]), "Berta")

        self.assertEqual(najbogatejsi([["Ana", 42]]), "Ana")
        self.assertEqual(najbogatejsi([["Ana", 0]]), "Ana")
        self.assertIsNone(najbogatejsi([]))

    def test_placilo(self):
        banka = [["Ana", 42], ["Berta", 13], ["Cilka", 66]]
        self.assertIsNone(placilo(banka, "Ana", "Cilka", 10))
        self.assertEqual(koliko_ima(banka, "Ana"), 32)
        self.assertEqual(koliko_ima(banka, "Cilka"), 76)
        self.assertEqual(koliko_ima(banka, "Berta"), 13)
        self.assertEqual(len(banka), 3, "A ima banka zdaj naenkrat več kot tri stranke?!")


class Test02Dodatna(unittest.TestCase):
    def test_dodaj_stranko_t(self):
        banka = []
        self.assertIsNone(dodaj_stranko_t(banka, "Ana", 42))
        self.assertEqual(banka, [("Ana", 42)])
        dodaj_stranko_t(banka, "Berta", 13)
        self.assertEqual(banka, [("Ana", 42), ("Berta", 13)])
        dodaj_stranko_t(banka, "Cilka", 66)
        self.assertEqual(banka, [("Ana", 42), ("Berta", 13), ("Cilka", 66)])

    def test_koliko_ima(self):
        banka = [("Ana", 42), ("Berta", 13), ("Cilka", 66)]
        self.assertEqual(koliko_ima_t(banka, "Ana"), 42)
        self.assertEqual(koliko_ima_t(banka, "Berta"), 13)
        self.assertEqual(koliko_ima_t(banka, "Cilka"), 66)
        self.assertIsNone(koliko_ima_t(banka, "Dani"))

        self.assertIsNone(koliko_ima_t([("Ana", 42)], "Dani"))
        self.assertIsNone(koliko_ima_t([], "Dani"))

    def test_bilanca(self):
        self.assertEqual(bilanca_t([("Ana", 42), ("Berta", 13), ("Cilka", 66)]), 121)
        self.assertEqual(bilanca_t([("Cilka", 66)]), 66)
        self.assertEqual(bilanca_t([]), 0)

    def test_najbogatejsi(self):
        self.assertEqual(najbogatejsi_t([("Ana", 42), ("Berta", 13), ("Cilka", 66)]), "Cilka")
        self.assertEqual(najbogatejsi_t([("Ana", 142), ("Berta", 13), ("Cilka", 66)]), "Ana")
        self.assertEqual(najbogatejsi_t([("Ana", 42), ("Berta", 113), ("Cilka", 66)]), "Berta")

        self.assertEqual(najbogatejsi_t([("Ana", 42)]), "Ana")
        self.assertEqual(najbogatejsi_t([("Ana", 0)]), "Ana")
        self.assertIsNone(najbogatejsi_t([]))

    def test_placilo(self):
        banka = [("Ana", 42), ("Berta", 13), ("Cilka", 66)]
        self.assertIsNone(placilo_t(banka, "Ana", "Cilka", 10))
        self.assertEqual(koliko_ima_t(banka, "Ana"), 32)
        self.assertEqual(koliko_ima_t(banka, "Cilka"), 76)
        self.assertEqual(koliko_ima_t(banka, "Berta"), 13)
        self.assertEqual(len(banka), 3, "A ima banka zdaj naenkrat več kot tri stranke?!")


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