/
tests.py
56 lines (40 loc) · 1.59 KB
/
tests.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import unittest
import Employee
class EmployeeTestCase(unittest.TestCase):
def setUp(self):
self.e = Employee.Employee("Test","Name",18,set())
def tearDown(self):
self.e = None
def testFirstName(self):
self.assertEqual(self.e.getFirstName(),"Test")
def testLastName(self):
self.assertEqual(self.e.getLastName(),"Name")
def testFullName(self):
self.assertEqual(self.e.getFullName(),"Test Name")
def testFormattedName(self):
self.assertEqual(self.e.getFormattedName(),"Name, Test")
def testPermantentDaysOff(self):
self.assertEqual(len(self.e.getPermDaysOff()),0,'Constructor failed')
self.e.setPermanentDaysOff({'sun','mon'})
self.assertEqual(self.e.getPermDaysOff(),{'sun','mon'})
def testIs18(self):
self.assertTrue(self.e.is18())
self.e.setAge(17)
self.assertFalse(self.e.is18())
def testRequestedDays(self):
day = Employee.myDateTime(1,1)
self.e.addReqDay(day)
self.assertEqual(self.e.getReqDays(),{day})
def testRemoveDaysAfter(self):
day = Employee.myDateTime(6,6)
day2 = Employee.myDateTime(3,3)
middleDay = Employee.myDateTime(3,3)
self.e.addReqDay(day)
self.e.addReqDay(day2)
self.assertTrue(len(self.e.getReqDays()) == 2)
self.e.removeReqDaysBefore(middleDay)
self.assertTrue(len(self.e.getReqDays()) == 1)
self.assertEqual(self.e.getReqDays(),{day})
if __name__ == '__main__':
try: unittest.main()
except SystemExit: pass