From 526d58ae930cc797ffdc536f20c4e39c74198e0e Mon Sep 17 00:00:00 2001 From: Milla J <232237854+milla-jovovich@users.noreply.github.com> Date: Tue, 2 Jun 2026 19:56:16 -0700 Subject: [PATCH] Update tests/test_hallways_pagination.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- tests/test_hallways_pagination.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/test_hallways_pagination.py b/tests/test_hallways_pagination.py index 34379d9..0ebd129 100644 --- a/tests/test_hallways_pagination.py +++ b/tests/test_hallways_pagination.py @@ -25,9 +25,13 @@ def _collection_that_rejects_where_get(drawers): col.count.return_value = len(drawers) def _get(limit=None, offset=0, include=None, where=None, ids=None, **kw): - if where is not None: + if where is not None and limit is None: raise RuntimeError("Error executing plan: too many SQL variables") - page = drawers[offset : offset + limit] if limit is not None else drawers + filtered_drawers = drawers + if where and "wing" in where: + target_wing = where["wing"] + filtered_drawers = [d for d in drawers if isinstance(d, dict) and d.get("wing") == target_wing] + page = filtered_drawers[offset : offset + limit] if limit is not None else filtered_drawers return { "ids": [f"d{i}" for i in range(offset, offset + len(page))], "metadatas": page,