Skip to content

Commit 89dceaf

Browse files
fix mypy TypedDict errors in session tests
1 parent fd440c5 commit 89dceaf

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/test_session.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -444,10 +444,15 @@ async def test_sqlite_session_special_characters_and_sql_injection():
444444
{"role": "user", "content": "Normal message"},
445445
]
446446
await session.add_items(items)
447+
448+
# Retrieve all items and verify they are stored correctly
447449
retrieved = await session.get_items()
448450
assert len(retrieved) == len(items)
449-
for i, item in enumerate(items):
450-
assert retrieved[i].get("content") == item["content"]
451+
assert retrieved[0].get("content") == "O'Reilly"
452+
assert retrieved[1].get("content") == "DROP TABLE sessions;"
453+
assert retrieved[2].get("content") == '"SELECT * FROM users WHERE name = \"admin\";"'
454+
assert retrieved[3].get("content") == "Robert'); DROP TABLE students;--"
455+
assert retrieved[4].get("content") == "Normal message"
451456
session.close()
452457

453458
@pytest.mark.asyncio

0 commit comments

Comments
 (0)