Update tests

Author: RussellLuo

tests/core/__init__.py

@@ -0,0 +1,48 @@
+from typing import Callable, Awaitable, AsyncIterator
+
+import pytest
+
+from coagent.core.types import Address, RawMessage, Channel, Subscription
+
+
+class TestChannel(Channel):
+    async def connect(self) -> None:
+        pass
+
+    async def close(self) -> None:
+        pass
+
+    async def publish(
+        self,
+        addr: Address,
+        msg: RawMessage,
+        request: bool = False,
+        reply: str = "",
+        timeout: float = 0.5,
+        probe: bool = True,
+    ) -> RawMessage | None:
+        pass
+
+    async def publish_multi(
+        self,
+        addr: Address,
+        msg: RawMessage,
+        probe: bool = True,
+    ) -> AsyncIterator[RawMessage]:
+        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 test_channel() -> TestChannel:
+    return TestChannel()

tests/core/conftest.py

@@ -1,50 +1,12 @@
 import asyncio
+
 import pytest
-from typing import Callable, Awaitable, AsyncIterator
 
-from coagent.core.types import Address, RawMessage, Channel, Subscription
+from coagent.core.types import Address
 from coagent.core.agent import BaseAgent, Context, handler
 from coagent.core.messages import Cancel, Message
 
 
-class TestChannel(Channel):
-    async def connect(self) -> None:
-        pass
-
-    async def close(self) -> None:
-        pass
-
-    async def publish(
-        self,
-        addr: Address,
-        msg: RawMessage,
-        request: bool = False,
-        reply: str = "",
-        timeout: float = 0.5,
-        probe: bool = True,
-    ) -> RawMessage | None:
-        pass
-
-    async def publish_multi(
-        self,
-        addr: Address,
-        msg: RawMessage,
-        probe: bool = True,
-    ) -> AsyncIterator[RawMessage]:
-        pass
-
-    async def subscribe(
-        self,
-        addr: Address,
-        handler: Callable[[RawMessage], Awaitable[None]],
-        queue: str = "",
-    ) -> Subscription:
-        pass
-
-    async def new_reply_topic(self) -> str:
-        pass
-
-
 class Run(Message):
     pass
 
@@ -64,9 +26,9 @@ async def handle(self, msg: Run, ctx: Context) -> None:
 
 class TestBlockingAgent:
     @pytest.mark.asyncio
-    async def test_receive(self):
+    async def test_receive(self, test_channel):
         agent = BlockingAgent()
-        agent.init(TestChannel(), Address(name="test", id="0"))
+        agent.init(test_channel, Address(name="test", id="0"))
         await agent.start()
 
         await agent.receive(Run().encode())