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_addPersonActivity(self): repo = PersonsActivitiesRepo() cont = PersonActivityController(repo, UndoRedoController()) cont.addPersonActivity(1, 1) self.assertIn(repo.findById(1), cont.getAll())
def test_removePersonActivity(self): repo = PersonsActivitiesRepo() cont = PersonActivityController(repo, UndoRedoController()) cont.addPersonActivity(1, 1) pa = repo.findById(1) cont.removePersonActivity(1, 1) self.assertNotIn(pa, cont.getAll())
def test_getAll(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.getAll(), repo.getAll())
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))