def test_removeActivityPersons(self):
repo = PersonsActivitiesRepo()
repo.add(PersonActivity(1, 1, 1))
repo.add(PersonActivity(2, 2, 1))
repo.add(PersonActivity(3, 3, 2))
repo.add(PersonActivity(5, 1, 3))
repo.add(PersonActivity(6, 2, 2))
self.assertEqual(repo.getActivityPersons(2), [3, 2])
def test_findActivityPersons(self):
repo = PersonsActivitiesRepo()
cont = PersonActivityController(repo, UndoRedoController())
cont.addPersonActivity(1, 1)
cont.addPersonActivity(1, 2)
cont.addPersonActivity(1, 3)
cont.addPersonActivity(2, 1)
cont.addPersonActivity(2, 2)
cont.addPersonActivity(3, 1)
cont.addPersonActivity(4, 2)
self.assertEqual(cont.findActivityPersons(1), repo.getActivityPersons(1))
