| from core.base_agent import BaseAgent
|
| from core.database import db
|
| from typing import Dict, Any
|
|
|
| class TesterAgent(BaseAgent):
|
| def __init__(self):
|
| super().__init__("Tester")
|
| self.create_chain("""
|
| You are a Software Tester. Your task is to create comprehensive test cases and execute them on the given code.
|
|
|
| Code to Test:
|
| {input}
|
|
|
| Create test cases that:
|
| 1. Cover all user stories and acceptance criteria
|
| 2. Include unit tests, integration tests, and system tests
|
| 3. Test edge cases and error conditions
|
| 4. Follow testing best practices
|
| 5. Include test data and expected results
|
|
|
| Please provide detailed test cases with clear steps and expected outcomes.
|
| """)
|
|
|
| async def create_test_cases(self, code: str) -> Dict[str, Any]:
|
| """Create test cases for the given code"""
|
| result = await self.process({"input": code})
|
|
|
|
|
| db.store_artifact(
|
| "test_cases",
|
| result,
|
| {
|
| "type": "test_case",
|
| "source": "tester",
|
| "status": "created"
|
| }
|
| )
|
|
|
| return {
|
| "status": "success",
|
| "test_cases": result,
|
| "message": "Test cases created successfully"
|
| }
|
|
|
| async def execute_tests(self, code: str, test_cases: str) -> Dict[str, Any]:
|
| """Execute test cases on the given code"""
|
| result = await self.process({
|
| "input": f"Code:\n{code}\n\nTest Cases:\n{test_cases}"
|
| })
|
|
|
|
|
| db.store_artifact(
|
| "test_results",
|
| result,
|
| {
|
| "type": "test_result",
|
| "source": "tester",
|
| "status": "executed"
|
| }
|
| )
|
|
|
| return {
|
| "status": "success",
|
| "test_results": result,
|
| "message": "Tests executed successfully"
|
| } |
