From eeb731c07ea0498dda138861864929199c86a033 Mon Sep 17 00:00:00 2001 From: Sangamesh Kulkarni <59434228+Sangamesh26@users.noreply.github.com> Date: Sat, 20 Sep 2025 05:23:30 +0530 Subject: [PATCH] test: Add tests for before and limit parameters for list SqliteSaver (#5816) **Description:** Add test for before and limit parameters for the list in SqliteSaver which was marked as TODO. --------- Co-authored-by: Caspar Broekhuizen --- libs/checkpoint-sqlite/tests/test_sqlite.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/libs/checkpoint-sqlite/tests/test_sqlite.py b/libs/checkpoint-sqlite/tests/test_sqlite.py index f242f54f5..062bde23c 100644 --- a/libs/checkpoint-sqlite/tests/test_sqlite.py +++ b/libs/checkpoint-sqlite/tests/test_sqlite.py @@ -116,7 +116,17 @@ class TestSqliteSaver: search_results_5[1].config["configurable"]["checkpoint_ns"], } == {"", "inner"} - # TODO: test before and limit params + # search with before param + search_results_6 = list(saver.list(None, before=search_results_5[1].config)) + assert len(search_results_6) == 1 + assert search_results_6[0].config["configurable"]["thread_id"] == "thread-1" + + # search with limit param + search_results_7 = list( + saver.list({"configurable": {"thread_id": "thread-2"}}, limit=1) + ) + assert len(search_results_7) == 1 + assert search_results_7[0].config["configurable"]["thread_id"] == "thread-2" def test_search_where(self) -> None: # call method / assertions