from datetime import date

from django.test import TestCase

from input.models import HonoraryCertificate, Project, Account, Literature


class ModelTestCase(TestCase):

    @classmethod
    def setUpTestData(cls):
        cls.account = Account.objects.create(code='1234', description='blabla')

    def test_set_granted(self):
        """ test if the model function set_granted() works as intended """
        obj = HonoraryCertificate.objects.create(realname='hurzel', email='hurzel@web.de')
        self.assertEqual(obj.granted, None)
        HonoraryCertificate.set_granted(obj.pk, True)
        obj2 = HonoraryCertificate.objects.get(pk=obj.pk)
        self.assertEqual(obj2.granted, True)

    def test_project_of_year(self):
        """ test if the finance id is resettet ad start of year """
        acc = self.account
        startdate = date(2022, 1, 1)
        obj = Project.objects.create(account=acc, name='testproject', start=startdate)
        self.assertEqual(obj.project_of_year, 1)

        obj2 = Project.objects.create(account=acc, name='testproject2', start=startdate)
        self.assertEqual(obj2.project_of_year, 2)

        olddate = date(2021, 12, 31)
        obj4 = Project.objects.create(account=acc, name='testproject2', start=olddate)

        obj3 = Project.objects.create(account=acc, name='testproject2', start=startdate)
        self.assertEqual(obj3.project_of_year, 3)

    def test_finance_id(self):
        """ test if the finance counting is correct """
        acc = self.account
        startdate = date(2022, 1, 1)
        obj = Project.objects.create(account=acc, name='testproject', start=startdate)
        self.assertEqual(obj.finance_id, "1234")

        obj2 = Project.objects.create(account=acc, name='testproject2', start=startdate)
        self.assertEqual(obj2.finance_id, "1234")

        olddate = date(2021, 12, 31)
        obj4 = Project.objects.create(account=acc, name='testproject2', start=olddate)

        obj3 = Project.objects.create(account=acc, name='testproject2', start=startdate)
        self.assertEqual(obj3.finance_id, "1234")

    def test_financed_id_for_subaccounts(self):
        account = Account.objects.create(code='21111', description='has subaccounts')
        obj = Project.objects.create(account=account, name='test', start=date(2025, 1, 1))

        self.assertEqual(obj.finance_id, f'{account.code}-001')

    def test_finance_id_later(self):
        obj = Project.objects.create(name='test', start=date(2025, 1, 1))

        self.assertFalse(obj.finance_id)

        obj.account = self.account
        obj.save()

        self.assertTrue(obj.finance_id)

    def test_pid(self):
        """ test if the pid counting is correct """
        acc = self.account
        startdate = date(2022, 1, 1)
        obj = Project.objects.create(account=acc, name='testproject', start=startdate)
        self.assertEqual(obj.pid, "123400000001")
        self.assertEqual(obj.account.code, "1234")

        obj2 = Project.objects.create(account=acc, name='testproject2', start=startdate)
        self.assertEqual(obj2.pid, "123400000002")

        olddate = date(2021, 12, 31)
        obj4 = Project.objects.create(account=acc, name='testproject2', start=olddate)

        obj3 = Project.objects.create(account=acc, name='testproject2', start=startdate)
        self.assertEqual(obj3.pid, "123400000004")

    def test_literature(self):
        obj = Literature.objects.create(cost='100', notes='jolo', selfbuy_give_data=False)
        self.assertEqual(obj.service_id, f'Literature{obj.id}')