from maconomy import EmployeeStatusListTemplate, Employee, Timesheet
import unittest

class EmployeeStatusListTemplateTest(unittest.TestCase):

    def setUp(self):
        self.template = EmployeeStatusListTemplate()
        self.employee = Employee.from_result(("MK", "Markus Krogh", "markus@nordu.net"))

    def test_substitute(self):
        timesheets = [Timesheet("11", 0, 0, self.employee, "JK")]
        result = self.template.build(timesheets)
        self.assertIn("[Unsubmitted] Markus Krogh (MK)", result)

    def test_submitted(self):
        timesheets = [Timesheet("11", submitted=1, approved=0, employee=self.employee, approver="JK")]
        result = self.template.build(timesheets)
        self.assertIn("[Not approved] Markus Krogh (MK)", result)

    def test_approved(self):
        timesheets = [Timesheet("11", submitted=1, approved=1, employee=self.employee, approver="JK")]
        result = self.template.build(timesheets)
        self.assertIn("[OK] Markus Krogh (MK)", result)