Skip to content

Commit 8e78a74

Browse files
fix: add test coverage for warning log
1 parent 594a3ef commit 8e78a74

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

tests/test_tools/test_inbuilt_tools.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -609,3 +609,21 @@ def test_speak_to_skips_non_llm_recipient(mocker):
609609
assert call_kwargs["content"]["message"] == "Hello both"
610610

611611
assert "2" in ret and "3" in ret
612+
613+
614+
def test_speak_to_warns_for_non_llm_recipient(mocker, caplog):
615+
import logging
616+
617+
model = DummyModel()
618+
sender = DummyAgent(unique_id=10, model=model)
619+
rule_recipient = DummyAgent(unique_id=11, model=model) # no .memory
620+
621+
model.agents = [sender, rule_recipient]
622+
623+
with caplog.at_level(logging.WARNING, logger="mesa_llm.tools.inbuilt_tools"):
624+
speak_to(sender, [11], "Test message")
625+
626+
assert any(
627+
"11" in record.message and "memory" in record.message
628+
for record in caplog.records
629+
)

0 commit comments

Comments
 (0)