def test_addPersonActivity(self):
    repo = PersonsActivitiesRepo()
    cont = PersonActivityController(repo, UndoRedoController())

    cont.addPersonActivity(1, 1)

    self.assertIn(repo.findById(1), cont.getAll())
예제 #2
0
  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())