import pytest from django.test import TestCase from engagements.utils import populate_database class TestModels(TestCase): @classmethod def setUpTestData(cls): cls.data = populate_database() @pytest.mark.django_db def test_check_all_dcs(self): dscs = self.data.get("sub_instruments") self.assertEqual(dscs[0].title, "DSC 1 - Title 1") @pytest.mark.django_db def test_effort_by_type(self): e = self.data["engagements"][0] total_planning = sum([x.effort_total_planned_hours() for x in e.effort.filter(effort_type="PLANNING")]) total_travel = sum([x.effort_total_planned_hours() for x in e.effort.filter(effort_type="TRAVEL")]) total_regulation = sum([x.effort_total_planned_hours() for x in e.effort.filter(effort_type="REGULATION")]) assert total_planning == 4.25 assert total_regulation == 0 assert total_travel == 1 # TODO finish this test! def test_total_effort_for_engagement(self): e = self.data["engagements"][0] assert e.total_effort() == 5.25