mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-22 00:11:28 +00:00
d7ae2038fb
* Rename TypeRoutedAgent to RoutedAgent; log on unhandled message. * format * Deprecation warning * add test for routed agent * add TypeRoutedAgent import * fix import
25 lines
820 B
Python
25 lines
820 B
Python
from dataclasses import dataclass
|
|
import pytest
|
|
import logging
|
|
|
|
from agnext.application import SingleThreadedAgentRuntime
|
|
from agnext.components import TypeSubscription
|
|
from agnext.core import TopicId
|
|
|
|
from test_utils import LoopbackAgent
|
|
|
|
|
|
@dataclass
|
|
class UnhandledMessageType: ...
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_routed_agent(caplog: pytest.LogCaptureFixture) -> None:
|
|
runtime = SingleThreadedAgentRuntime()
|
|
with caplog.at_level(logging.INFO):
|
|
await runtime.register("loopback", lambda: LoopbackAgent(), lambda: [TypeSubscription("default", "loopback")])
|
|
runtime.start()
|
|
await runtime.publish_message(UnhandledMessageType(), topic_id=TopicId("default", "default"))
|
|
await runtime.stop_when_idle()
|
|
assert any("Unhandled message: " in e.message for e in caplog.records)
|