Add NopChannel for testing purposes

Author: RussellLuo

coagent/agents/mcp_server.py

@@ -4,14 +4,14 @@
 
 from coagent.core import BaseAgent, Context, handler, logger, Message
 from coagent.core.exceptions import InternalError
-from mcp import ClientSession, Tool as MCPTool  # ruff: noqa: F401
+from mcp import ClientSession, Tool as MCPTool  # noqa: F401
 from mcp.client.sse import sse_client
 from mcp.client.stdio import stdio_client, StdioServerParameters
 from mcp.types import (
     CallToolResult as MCPCallToolResult,
     ListToolsResult as MCPListToolsResult,
-    ImageContent as MCPImageContent,  # ruff: noqa: F401
-    TextContent as MCPTextContent,  # ruff: noqa: F401
+    ImageContent as MCPImageContent,  # noqa: F401
+    TextContent as MCPTextContent,  # noqa: F401
 )
 from pydantic import BaseModel
 

coagent/core/runtime.py

@@ -1,6 +1,6 @@
 import abc
 import asyncio
-from typing import AsyncIterator
+from typing import AsyncIterator, Awaitable, Callable
 
 import pydantic
 
@@ -14,6 +14,7 @@
     Runtime,
     Address,
     RawMessage,
+    Subscription,
 )
 
 
@@ -179,3 +180,39 @@ async def __anext__(self) -> RawMessage:
 
     def __aiter__(self):
         return self
+
+
+class NopChannel(Channel):
+    """A no-op channel mainly for testing purposes."""
+
+    async def connect(self) -> None:
+        pass
+
+    async def close(self) -> None:
+        pass
+
+    async def publish(
+        self,
+        addr: Address,
+        msg: RawMessage,
+        stream: bool = False,
+        request: bool = False,
+        reply: str = "",
+        timeout: float = 0.5,
+        probe: bool = True,
+    ) -> AsyncIterator[RawMessage] | RawMessage | None:
+        pass
+
+    async def subscribe(
+        self,
+        addr: Address,
+        handler: Callable[[RawMessage], Awaitable[None]],
+        queue: str = "",
+    ) -> Subscription:
+        pass
+
+    async def new_reply_topic(self) -> str:
+        return ""
+
+    async def cancel(self, addr: Address) -> None:
+        pass

tests/core/conftest.py

@@ -1,46 +1,15 @@
 import asyncio
 import functools
 import inspect
-from typing import Callable, Awaitable
 
 import pytest
 
-from coagent.core.types import Address, Agent, RawMessage, Channel, Subscription
+from coagent.core.types import Agent
 from coagent.core.util import idle_loop
+from coagent.core.runtime import NopChannel
 from coagent.runtimes.local_runtime import LocalChannel
 
 
-class NopChannel(Channel):
-    async def connect(self) -> None:
-        pass
-
-    async def close(self) -> None:
-        pass
-
-    async def publish(
-        self,
-        addr: Address,
-        msg: RawMessage,
-        stream: bool = False,
-        request: bool = False,
-        reply: str = "",
-        timeout: float = 0.5,
-        probe: bool = True,
-    ) -> RawMessage | None:
-        pass
-
-    async def subscribe(
-        self,
-        addr: Address,
-        handler: Callable[[RawMessage], Awaitable[None]],
-        queue: str = "",
-    ) -> Subscription:
-        pass
-
-    async def new_reply_topic(self) -> str:
-        pass
-
-
 @pytest.fixture
 def nop_channel() -> NopChannel:
     return NopChannel()