diff --git a/tests/unit/database/test_mpi_service.py b/tests/unit/database/test_mpi_service.py index ca9d278b..c4dfbae6 100644 --- a/tests/unit/database/test_mpi_service.py +++ b/tests/unit/database/test_mpi_service.py @@ -986,17 +986,17 @@ def test_get_orphaned_persons_cursor(self, session: Session): person3 = models.Person(id=3) person4 = models.Person(id=4) patient = models.Patient(person=person4, data={}) - session.add_all([patient, person2, person3, person4]) + session.add_all([patient, person1, person2, person3]) session.flush() assert mpi_service.get_orphaned_persons(session, limit=1, cursor=person1.id) == [person2] - # assert mpi_service.get_orphaned_persons(session, limit=1, cursor=person2.id) == [person3] - # assert mpi_service.get_orphaned_persons(session, limit=2, cursor=person2.id) == [person3] - # assert mpi_service.get_orphaned_persons(session, limit=2, cursor=person1.id) == [ - # person2, - # person3, - # ] - # assert mpi_service.get_orphaned_persons(session, limit=5, cursor=person1.id) == [ - # person2, - # person3, - # ] + assert mpi_service.get_orphaned_persons(session, limit=1, cursor=person2.id) == [person3] + assert mpi_service.get_orphaned_persons(session, limit=2, cursor=person2.id) == [person3] + assert mpi_service.get_orphaned_persons(session, limit=2, cursor=person1.id) == [ + person2, + person3, + ] + assert mpi_service.get_orphaned_persons(session, limit=5, cursor=person1.id) == [ + person2, + person3, + ]