from maconomy import CEOEmailTemplate, Timesheet import unittest class CEOEmailTemplateTest(unittest.TestCase): def setUp(self): self.template = CEOEmailTemplate() self.not_approved = Timesheet.from_result(("MK", "Markus Krogh", "markus@nordu.net", "11", 1, 0, "JK")) self.not_submitted = Timesheet.from_result(("MKR", "Markus Krogh", "markusk@nordu.net", "11", 0, 0, "JKL")) def test_ceo_email(self): result = self.template.build(timesheets=[self.not_approved, self.not_submitted]) self.assertIn("[Not approved] Markus Krogh (MK)", result) self.assertIn("JK", result) self.assertIn("[Unsubmitted] Markus Krogh (MKR)", result) self.assertIn("JKL", result)