File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments