def test_addPersonActivity(self): repo = PersonsActivitiesRepo() cont = PersonActivityController(repo, UndoRedoController()) cont.addPersonActivity(1, 1) self.assertIn(repo.findById(1), cont.getAll())
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_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())