mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 08:38:30 +00:00
test: Improve crew verbose output test with event log filtering
- Filter out event listener logs in verbose output test - Ensure no output when verbose is set to False - Enhance test coverage for crew logging behavior
This commit is contained in:
@@ -833,6 +833,12 @@ def test_crew_verbose_output(capsys):
|
|||||||
|
|
||||||
crew.kickoff()
|
crew.kickoff()
|
||||||
captured = capsys.readouterr()
|
captured = capsys.readouterr()
|
||||||
|
|
||||||
|
# Filter out event listener logs (lines starting with '[')
|
||||||
|
filtered_output = "\n".join(
|
||||||
|
line for line in captured.out.split("\n") if not line.startswith("[")
|
||||||
|
)
|
||||||
|
|
||||||
expected_strings = [
|
expected_strings = [
|
||||||
"\x1b[1m\x1b[95m# Agent:\x1b[00m \x1b[1m\x1b[92mResearcher",
|
"\x1b[1m\x1b[95m# Agent:\x1b[00m \x1b[1m\x1b[92mResearcher",
|
||||||
"\x1b[00m\n\x1b[95m## Task:\x1b[00m \x1b[92mResearch AI advancements.",
|
"\x1b[00m\n\x1b[95m## Task:\x1b[00m \x1b[92mResearch AI advancements.",
|
||||||
@@ -845,27 +851,19 @@ def test_crew_verbose_output(capsys):
|
|||||||
]
|
]
|
||||||
|
|
||||||
for expected_string in expected_strings:
|
for expected_string in expected_strings:
|
||||||
assert expected_string in captured.out
|
assert expected_string in filtered_output
|
||||||
|
|
||||||
# Now test with verbose set to False
|
# Now test with verbose set to False
|
||||||
crew.verbose = False
|
crew.verbose = False
|
||||||
crew._logger = Logger(verbose=False)
|
crew._logger = Logger(verbose=False)
|
||||||
crew.kickoff()
|
crew.kickoff()
|
||||||
expected_listener_logs = [
|
|
||||||
"[🚀 CREW 'CREW' STARTED]",
|
|
||||||
"[📋 TASK STARTED: RESEARCH AI ADVANCEMENTS.]",
|
|
||||||
"[🤖 AGENT 'RESEARCHER' STARTED TASK]",
|
|
||||||
"[✅ AGENT 'RESEARCHER' COMPLETED TASK]",
|
|
||||||
"[✅ TASK COMPLETED: RESEARCH AI ADVANCEMENTS.]",
|
|
||||||
"[📋 TASK STARTED: WRITE ABOUT AI IN HEALTHCARE.]",
|
|
||||||
"[🤖 AGENT 'SENIOR WRITER' STARTED TASK]",
|
|
||||||
"[✅ AGENT 'SENIOR WRITER' COMPLETED TASK]",
|
|
||||||
"[✅ TASK COMPLETED: WRITE ABOUT AI IN HEALTHCARE.]",
|
|
||||||
"[✅ CREW 'CREW' COMPLETED]",
|
|
||||||
]
|
|
||||||
captured = capsys.readouterr()
|
captured = capsys.readouterr()
|
||||||
for log in expected_listener_logs:
|
filtered_output = "\n".join(
|
||||||
assert log in captured.out
|
line
|
||||||
|
for line in captured.out.split("\n")
|
||||||
|
if not line.startswith("[") and line.strip() and not line.startswith("\x1b")
|
||||||
|
)
|
||||||
|
assert filtered_output == "" # No output when verbose is False
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.vcr(filter_headers=["authorization"])
|
@pytest.mark.vcr(filter_headers=["authorization"])
|
||||||
|
|||||||
Reference in New Issue
Block a user