mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 08:38:30 +00:00
43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
import pytest
|
|
from rich.text import Text
|
|
from rich.tree import Tree
|
|
|
|
from crewai.utilities.events.utils.console_formatter import ConsoleFormatter
|
|
|
|
|
|
def test_console_formatter_gbk_encoding():
|
|
"""Test that console formatter output can be encoded with GBK."""
|
|
formatter = ConsoleFormatter(verbose=True)
|
|
|
|
tree = Tree("Test Tree")
|
|
|
|
formatter.update_tree_label(tree, "[OK] Test:", "Test Name", "green")
|
|
label_text = str(tree.label)
|
|
|
|
node = formatter.add_tree_node(tree, "[OK] Test Node", "green")
|
|
node_text = str(node.label)
|
|
|
|
try:
|
|
label_text.encode("gbk")
|
|
node_text.encode("gbk")
|
|
assert True, "Text can be encoded with GBK"
|
|
except UnicodeEncodeError as e:
|
|
assert False, f"Failed to encode with GBK: {e}"
|
|
|
|
crew_tree = formatter.create_crew_tree("Test Crew", "crew-123")
|
|
task_branch = formatter.create_task_branch(crew_tree, "1")
|
|
agent_branch = formatter.create_agent_branch(task_branch, "Test Agent", crew_tree)
|
|
|
|
formatter.update_task_status(crew_tree, "1", "Test Agent", "completed")
|
|
formatter.update_agent_status(agent_branch, "Test Agent", crew_tree, "completed")
|
|
|
|
flow_tree = formatter.create_flow_tree("Test Flow", "flow-123")
|
|
formatter.update_flow_status(flow_tree, "Test Flow", "flow-123", "completed")
|
|
|
|
for tree_obj in [crew_tree, task_branch, agent_branch, flow_tree]:
|
|
tree_text = str(tree_obj)
|
|
try:
|
|
tree_text.encode("gbk")
|
|
except UnicodeEncodeError as e:
|
|
assert False, f"Failed to encode tree with GBK: {e}"
|