fix(backend): add test for get_many with pagination

Signed-off-by: Fatih Acar <[email protected]>

Author: fatih-acar

backend/tests/unit/core/test_manager_node.py

@@ -3,6 +3,8 @@
 import pytest
 from infrahub_sdk.uuidt import UUIDT
 
+from infrahub import config
+from infrahub.constants.database import Neo4jRuntime
 from infrahub.core.branch import Branch
 from infrahub.core.initialization import create_branch
 from infrahub.core.manager import NodeManager, identify_node_class
@@ -277,6 +279,39 @@ async def test_get_many(db: InfrahubDatabase, default_branch: Branch, criticalit
     assert len(nodes) == 2
 
 
+async def test_get_many_with_pagination(
+    db: InfrahubDatabase,
+    default_branch: Branch,
+    criticality_schema: NodeSchema,
+    criticality_low: Node,
+    criticality_medium: Node,
+    criticality_high: Node,
+):
+    new_node_ids: list[str] = []
+    for index in range(10):
+        bulk_node = await Node.init(db=db, schema=criticality_schema, branch=default_branch)
+        await bulk_node.new(db=db, name=f"bulk-{index}", level=index)
+        await bulk_node.save(db=db)
+        new_node_ids.append(bulk_node.id)
+
+    original_query_size_limit = config.SETTINGS.database.query_size_limit
+    config.SETTINGS.database.query_size_limit = 1
+    original_neo4j_runtime = db.default_neo4j_runtime
+    db.default_neo4j_runtime = Neo4jRuntime.PARALLEL
+
+    try:
+        target_ids = [criticality_low.id, criticality_medium.id, criticality_high.id, *new_node_ids]
+        nodes = await NodeManager.get_many(db=db, ids=target_ids)
+    finally:
+        config.SETTINGS.database.query_size_limit = original_query_size_limit
+        db.default_neo4j_runtime = original_neo4j_runtime
+
+    expected_ids = set(target_ids)
+    assert set(nodes) == expected_ids
+    assert len(nodes) == len(expected_ids)
+    assert all(isinstance(nodes[node_id], Node) for node_id in expected_ids)
+
+
 async def test_get_many_prefetch(
     db: InfrahubDatabase, person_jack_tags_main, tag_blue_main, tag_red_main, branch: Branch
 ):