Skip to content

Commit fe190ce

Browse files
committed
fix(tests): update firestore state assertions to expect JSON-serialized string
1 parent 72ebdc5 commit fe190ce

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/unittests/integrations/firestore/test_firestore_session_service.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414

1515
from __future__ import annotations
1616

17+
import json
1718
from unittest import mock
1819

1920
from 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

0 commit comments

Comments
 (0)