autogen/tests/test_runtime.py

from typing import Any, Sequence
import pytest

from agnext.application_components import SingleThreadedAgentRuntime
from agnext.core import AgentRuntime
from agnext.core import BaseAgent
from agnext.core import CancellationToken

class NoopAgent(BaseAgent):
    def __init__(self, name: str, router: AgentRuntime) -> None:
        super().__init__(name, router)

    @property
    def subscriptions(self) -> Sequence[type]:
        return []

    async def on_message(self, message: Any, cancellation_token: CancellationToken) -> Any:
        raise NotImplementedError


@pytest.mark.asyncio
async def test_agent_names_must_be_unique() -> None:
    router = SingleThreadedAgentRuntime()

    _agent1 = NoopAgent("name1", router)

    with pytest.raises(ValueError):
        _agent1_again = NoopAgent("name1", router)

    _agent3 = NoopAgent("name3", router)