def test_canAssignTasks(self): worker = Worker(workerID="W1", begin=tHour(0, 00), end=tHour(23, 59), multitask=2) self.assertFalse(worker.hasReachedTaskLimit()) worker.assignTask(self.task1) self.assertFalse(worker.hasReachedTaskLimit()) worker.assignTask(self.task2) self.assertTrue(worker.hasReachedTaskLimit())
def test_unnasignTask(self): worker = Worker(workerID="W1", begin=tHour(0, 00), end=tHour(23, 59), multitask=2) worker.assignTask(self.task1) worker.unassignTask(self.task1.taskID) self.assertEqual(len(worker.assignedTasks), 0)
def test_exactTaskLimit(self): worker = Worker(workerID="W1", begin=tHour(0, 00), end=tHour(23, 59), multitask=2) worker.assignTask(self.task1) worker.assignTask(self.task2) self.assertEqual(len(worker.assignedTasks), 2)
def test_maxTasksLimit(self): worker = Worker(workerID="W1", begin=tHour(0, 00), end=tHour(23, 59), multitask=2) worker.assignTask(self.task1) worker.assignTask(self.task2) with self.assertRaises(MaxTaskLimitReachedException): worker.assignTask(self.task3)