From 03636ab33f6bc53f2bbe0ceda8c9bce03277fe66 Mon Sep 17 00:00:00 2001 From: Tuck Date: Sun, 9 Aug 2026 20:55:44 -0400 Subject: [PATCH] test: add message_metadata unit tests --- tests/agent/test_message_metadata.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 tests/agent/test_message_metadata.py diff --git a/tests/agent/test_message_metadata.py b/tests/agent/test_message_metadata.py new file mode 100644 index 0000000000..d8e257a327 --- /dev/null +++ b/tests/agent/test_message_metadata.py @@ -0,0 +1,23 @@ +from agent.message_metadata import append_message, stamp_message_timestamp + + +def test_stamp_preserves_source_timestamp(monkeypatch): + monkeypatch.setattr("agent.message_metadata.wall_time", lambda: 999.0) + message = {"role": "user", "content": "hello", "timestamp": 123.0} + + result = stamp_message_timestamp(message) + + assert result is message + assert message["timestamp"] == 123.0 + + +def test_append_stamps_same_mapping_at_append_time(monkeypatch): + monkeypatch.setattr("agent.message_metadata.wall_time", lambda: 456.0) + messages = [] + message = {"role": "tool", "content": "ok"} + + result = append_message(messages, message) + + assert result is message + assert messages == [message] + assert message["timestamp"] == 456.0