File tree Expand file tree Collapse file tree
tests/unittests/integrations/firestore Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1414
1515from __future__ import annotations
1616
17+ import json
1718from unittest import mock
1819
1920from google .adk .events .event import Event
@@ -117,7 +118,7 @@ async def test_create_session(mock_firestore_client):
117118 assert args [1 ]["id" ] == session .id
118119 assert args [1 ]["appName" ] == app_name
119120 assert args [1 ]["userId" ] == user_id
120- assert args [1 ]["state" ] == {}
121+ assert json . loads ( args [1 ]["state" ]) == {}
121122 assert args [1 ]["createTime" ] == firestore .SERVER_TIMESTAMP
122123 assert args [1 ]["updateTime" ] == firestore .SERVER_TIMESTAMP
123124
@@ -324,8 +325,7 @@ async def test_append_event_with_state_delta(mock_firestore_client):
324325
325326 transaction .update .assert_called_once ()
326327 args , kwargs = transaction .update .call_args
327- # In modular Firestore configurations alignments, updating variables mock assertions core setups
328- assert args [1 ]["state" ] == session .state
328+ assert json .loads (args [1 ]["state" ]) == session .state
329329 assert args [1 ]["updateTime" ] == firestore .SERVER_TIMESTAMP
330330
331331
You can’t perform that action at this time.
0 commit comments