apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-05 14:55:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: According to the routing feature plan in Chapter 2.2.3.2 of RFC 113, the routing functionality is to be consolidated into the Environment class.

Browse source
This commit is contained in:
莘权 马 2023-11-04 14:07:33 +08:00
parent 8137e1af50
commit 2688fe680a
3 changed files with 33 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/metagpt/test_role.py
View file

@ -6,7 +6,11 @@
@File : test_role.py
@Modified By: mashenquan, 2023-11-1. In line with Chapter 2.2.1 and 2.2.2 of RFC 116, introduce unit tests for
the utilization of the new message distribution feature in message handling.
@Modified By: mashenquan, 2023-11-4. According to the routing feature plan in Chapter 2.2.3.2 of RFC 113, the routing
functionality is to be consolidated into the `Environment` class.
"""
import uuid
import pytest
from pydantic import BaseModel
@ -64,6 +68,7 @@ async def test_react():
assert role.is_idle
env = Environment()
env.add_role(role)
assert env.get_subscribed_tags(role) == {seed.subscription}
env.publish_message(Message(content="test", tx_to=seed.subscription))
assert not role.is_idle
while not env.is_idle:
@ -74,6 +79,9 @@ async def test_react():
while not env.is_idle:
await env.run()
assert role.is_idle
tag = uuid.uuid4().hex
role.subscribe({tag})
assert env.get_subscribed_tags(role) == {seed.subscription, tag}
if __name__ == "__main__":