def test_findById(self):
repo = PersonsActivitiesRepo()
repo.add(PersonActivity(1, 1, 1))
repo.add(PersonActivity(2, 2, 1))
self.assertEqual(repo.findById(1), PersonActivity(1, 1, 1))
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_removePersonActivities(self):
repo = PersonsActivitiesRepo()
repo.add(PersonActivity(1, 1, 1))
repo.add(PersonActivity(3, 3, 1))
repo.add(PersonActivity(4, 1, 2))
repo.add(PersonActivity(5, 1, 3))
repo.add(PersonActivity(6, 2, 2))
self.assertEqual(repo.getPersonActivities(1), [1, 2, 3])
def test_add(self):
repo = PersonsActivitiesRepo()
repo.add(PersonActivity(1, 1, 1))
repo.add(PersonActivity(2, 2, 1))
repo.add(PersonActivity(3, 3, 1))
repo.add(PersonActivity(4, 1, 2))
repo.add(PersonActivity(5, 1, 3))
repo.add(PersonActivity(6, 2, 2))
self.assertIn(PersonActivity(1, 1, 1), repo.getAll())
def test_getAll(self):
repo = PersonsActivitiesRepo()
repo.add(PersonActivity(1, 1, 1))
repo.add(PersonActivity(2, 2, 1))
repo.add(PersonActivity(3, 3, 1))
repo.add(PersonActivity(4, 1, 2))
repo.add(PersonActivity(5, 1, 3))
repo.add(PersonActivity(6, 2, 2))
self.assertEqual(repo.getAll(), [PersonActivity(1, 1, 1),
PersonActivity(2, 2, 1),
PersonActivity(3, 3, 1),
PersonActivity(4, 1, 2),
PersonActivity(5, 1, 3),
PersonActivity(6, 2, 2)])
def test_remove(self):
repo = PersonsActivitiesRepo()
repo.add(PersonActivity(1, 1, 1))
repo.add(PersonActivity(2, 2, 1))
self.assertEqual(repo.remove(2), PersonActivity(2, 2, 1))
