from maconomy import UnsubmittedEmailTemplate, MissingEmailTemplate, ManagerEmailTemplate, Employee import unittest class UnsubmittedEmailTemplateTest(unittest.TestCase): def setUp(self): self.template = UnsubmittedEmailTemplate() def test_substitution(self): result = self.template.build(week=10, maconomyurl="http://localhost/", helpurl="http://example.com") self.assertIn("week 10", result) self.assertIn("href=\"http://localhost/\"", result) self.assertIn("href=\"http://example.com\"", result) class MissingEmailTemplateTest(unittest.TestCase): def setUp(self): self.template = MissingEmailTemplate() def test_substitution(self): result = self.template.build(maconomyurl="http://localhost/", helpurl="http://example.com") self.assertIn("href=\"http://localhost/\"", result) self.assertIn("href=\"http://example.com\"", result) class ManagerEmailTemplateTest(unittest.TestCase): def setUp(self): self.template = ManagerEmailTemplate() self.employee = Employee(("MK", "Markus Krogh", "markus@nordu.net")) def test_substitute(self): result = self.template.build( employee=self.employee, week=11, maconomyurl="http://localhost/", ) self.assertIn("Markus Krogh (MK)", result) self.assertIn("Week 11", result) self.assertIn("not been submitted", result) self.assertIn("not been approved", result) self.assertIn("href=\"http://localhost/\"", result) def test_submitted(self): result = self.template.build( employee=self.employee, week=11, maconomyurl="http://localhost/", submitted = True, ) self.assertIn("has been submitted", result) def test_approved(self): result = self.template.build( employee=self.employee, week=11, maconomyurl="http://localhost/", approved = True, ) self.assertIn("has been approved", result)