test with correctly added persons

CDCgov · Feb 26, 2025 · 1f9774f · 1f9774f
1 parent 625e844
commit 1f9774f
Showing 1 changed file with 11 additions and 11 deletions.
diff --git 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,
+        ]