from maconomy import EmployeeStatusTemplate, Employee, Timesheet import unittest class EmployeeStatusTemplateTest(unittest.TestCase): def setUp(self): self.template = EmployeeStatusTemplate() self.employee = Employee.from_result(("MK", "Markus Krogh", "markus@nordu.net")) def test_substitute(self): timesheet = Timesheet("11", 0, 0, self.employee, "JK") result = self.template.build(timesheet) self.assertIn("Markus Krogh (MK)", result) self.assertIn("not been submitted", result) self.assertIn("not been approved", result) def test_submitted(self): timesheet = Timesheet("11", submitted=1, approved=0, employee=self.employee, approver="JK") result = self.template.build(timesheet) self.assertIn("has been submitted", result) def test_approved(self): timesheet = Timesheet("11", submitted=1, approved=1, employee=self.employee, approver="JK") result = self.template.build(timesheet) self.assertIn("has been approved", result)