def testServiceVisitWithOneTaskThatDoesNotHaveTheSpecifiedId(self):
task1 = Task("task a")
task2 = Task("task b")
visit = ServiceVisit("name", "address", "customer", [task1])
with self.assertRaises(ValueError):
visit.getTaskById(task2.getTaskId())
def testServiceVisitWithOneTaskWithSpecifiedId(self):
task1 = Task("task a")
visit = ServiceVisit("name", "address", "customer", [task1])
result = visit.getTaskById(task1.getTaskId())
self.assertEqual(task1, result, "checking result")
def testServiceVisitWithMultipleTasksAndLastHasSpecifiedId(self):
task1 = Task("task a")
task2 = Task("task b")
task3 = Task("task c")
visit = ServiceVisit("name", "address", "customer",
[task1, task2, task3])
result = visit.getTaskById(task3.getTaskId())
self.assertEqual(task3, result, "checking result")
