update roles ut

This commit is contained in:
张建生 2025-01-24 18:32:18 +08:00
parent e8ab02b203
commit 991e49e2d1

View file

@ -7,20 +7,16 @@
"""
from pathlib import Path
from typing import List
import json
import pytest
from pydantic import BaseModel, Field
from unittest.mock import AsyncMock, MagicMock
from metagpt.actions import DebugError, RunCode, WriteTest, UserRequirement
from metagpt.actions.summarize_code import SummarizeCode
from metagpt.actions.prepare_documents import PrepareDocuments
from metagpt.environment import Environment
from metagpt.roles import QaEngineer
from metagpt.schema import Message, AIMessage, Document, RunCodeContext, TestingContext
from metagpt.utils.common import any_to_str, aread, awrite
from metagpt.utils.project_repo import ProjectRepo
from metagpt.schema import Message, AIMessage, Document, RunCodeContext
from metagpt.utils.common import any_to_str, awrite
class MockProjectRepo:
@ -220,4 +216,4 @@ async def test_qa_full_workflow(tmp_path):
env.msgs.clear()
role.test_round_allowed = 1
rsp = await role.run(with_message=msg)
assert "Exceeding" in rsp.content
assert "Exceeding" in rsp.content