OpenHands/tests/unit/io/test_json.py
Xingyao Wang ca424ec15d
[agent] Add LLM risk analyzer (#9349)
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: openhands <openhands@all-hands.dev>
Co-authored-by: Graham Neubig <neubig@gmail.com>
Co-authored-by: llamantino <213239228+llamantino@users.noreply.github.com>
Co-authored-by: mamoodi <mamoodiha@gmail.com>
Co-authored-by: Tim O'Farrell <tofarr@gmail.com>
Co-authored-by: Hiep Le <69354317+hieptl@users.noreply.github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Ryan H. Tran <descience.thh10@gmail.com>
Co-authored-by: Neeraj Panwar <49247372+npneeraj@users.noreply.github.com>
Co-authored-by: sp.wack <83104063+amanape@users.noreply.github.com>
Co-authored-by: Insop <1240382+insop@users.noreply.github.com>
Co-authored-by: test <test@test.com>
Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
Co-authored-by: Zhonghao Jiang <zhonghao.J@outlook.com>
Co-authored-by: Ray Myers <ray.myers@gmail.com>
2025-08-22 14:02:36 +00:00

51 lines
1.5 KiB
Python

from datetime import datetime
from openhands.events.action import MessageAction
from openhands.io import json
def test_event_serialization_deserialization():
message = MessageAction(content='This is a test.', wait_for_response=False)
message._id = 42
message._timestamp = datetime(2020, 1, 1, 23, 59, 58)
serialized = json.dumps(message)
deserialized = json.loads(serialized)
expected = {
'id': 42,
'timestamp': '2020-01-01T23:59:58',
'action': 'message',
'message': 'This is a test.',
'args': {
'content': 'This is a test.',
'image_urls': None,
'file_urls': None,
'wait_for_response': False,
'security_risk': -1,
},
}
assert deserialized == expected
def test_array_serialization_deserialization():
message = MessageAction(content='This is a test.', wait_for_response=False)
message._id = 42
message._timestamp = datetime(2020, 1, 1, 0, 0, 0)
serialized = json.dumps([message])
deserialized = json.loads(serialized)
expected = [
{
'id': 42,
'timestamp': '2020-01-01T00:00:00',
'action': 'message',
'message': 'This is a test.',
'args': {
'content': 'This is a test.',
'image_urls': None,
'file_urls': None,
'wait_for_response': False,
'security_risk': -1,
},
}
]
assert deserialized == expected