mirror of
https://github.com/microsoft/autogen.git
synced 2025-06-26 22:30:10 +00:00
23 lines
795 B
Python
23 lines
795 B
Python
from autogen_core.components import TypeSubscription
|
|
from autogen_core.base import TopicId, AgentId
|
|
|
|
import pytest
|
|
|
|
from autogen_core.base.exceptions import CantHandleException
|
|
|
|
def test_type_subscription_match() -> None:
|
|
sub = TypeSubscription(topic_type="t1", agent_type="a1")
|
|
|
|
assert sub.is_match(TopicId(type="t0", source="s1")) == False
|
|
assert sub.is_match(TopicId(type="t1", source="s1")) == True
|
|
assert sub.is_match(TopicId(type="t1", source="s2")) == True
|
|
|
|
|
|
def test_type_subscription_map() -> None:
|
|
sub = TypeSubscription(topic_type="t1", agent_type="a1")
|
|
|
|
assert sub.map_to_agent(TopicId(type="t1", source="s1")) == AgentId(type="a1", key="s1")
|
|
|
|
with pytest.raises(CantHandleException):
|
|
_agent_id = sub.map_to_agent(TopicId(type="t0", source="s1"))
|