183 lines
6.4 KiB
Python
183 lines
6.4 KiB
Python
"""
|
|
Test index generator - creates browsable view of available tests.
|
|
"""
|
|
|
|
from pathlib import Path
|
|
from typing import Dict, List
|
|
import ast
|
|
|
|
|
|
def parse_test_file(file_path: Path) -> Dict:
|
|
"""Parse a test file and extract test methods with docstrings."""
|
|
try:
|
|
with open(file_path, 'r') as f:
|
|
tree = ast.parse(f.read())
|
|
|
|
module_doc = ast.get_docstring(tree)
|
|
classes = []
|
|
|
|
for node in ast.walk(tree):
|
|
if isinstance(node, ast.ClassDef):
|
|
class_doc = ast.get_docstring(node)
|
|
methods = []
|
|
|
|
for item in node.body:
|
|
if isinstance(item, ast.FunctionDef) and item.name.startswith('test_'):
|
|
method_doc = ast.get_docstring(item)
|
|
methods.append({
|
|
'name': item.name,
|
|
'doc': method_doc or "No description"
|
|
})
|
|
|
|
if methods: # Only include classes with test methods
|
|
classes.append({
|
|
'name': node.name,
|
|
'doc': class_doc or "No description",
|
|
'methods': methods
|
|
})
|
|
|
|
return {
|
|
'file': file_path.name,
|
|
'module_doc': module_doc or "No module description",
|
|
'classes': classes
|
|
}
|
|
except Exception as e:
|
|
return {
|
|
'file': file_path.name,
|
|
'error': str(e)
|
|
}
|
|
|
|
|
|
def build_test_index(tests_dir: Path) -> Dict:
|
|
"""
|
|
Build a hierarchical index of all tests.
|
|
|
|
Returns structure:
|
|
{
|
|
'mascotas': {
|
|
'test_pet_owners.py': {...},
|
|
'test_pets.py': {...}
|
|
},
|
|
'productos': {...},
|
|
...
|
|
}
|
|
"""
|
|
index = {}
|
|
|
|
# Find all domain directories (mascotas, productos, etc.)
|
|
for domain_dir in tests_dir.iterdir():
|
|
if not domain_dir.is_dir():
|
|
continue
|
|
if domain_dir.name.startswith('_'):
|
|
continue
|
|
|
|
domain_tests = {}
|
|
|
|
# Find all test_*.py files in domain
|
|
for test_file in domain_dir.glob('test_*.py'):
|
|
test_info = parse_test_file(test_file)
|
|
domain_tests[test_file.name] = test_info
|
|
|
|
if domain_tests: # Only include domains with tests
|
|
index[domain_dir.name] = domain_tests
|
|
|
|
return index
|
|
|
|
|
|
def generate_markdown_index(index: Dict) -> str:
|
|
"""Generate markdown representation of test index."""
|
|
lines = ["# Contract Tests Index\n"]
|
|
|
|
for domain, files in sorted(index.items()):
|
|
lines.append(f"## {domain.capitalize()}\n")
|
|
|
|
for filename, file_info in sorted(files.items()):
|
|
if 'error' in file_info:
|
|
lines.append(f"### {filename} ⚠️ Parse Error")
|
|
lines.append(f"```\n{file_info['error']}\n```\n")
|
|
continue
|
|
|
|
lines.append(f"### {filename}")
|
|
lines.append(f"{file_info['module_doc']}\n")
|
|
|
|
for cls in file_info['classes']:
|
|
lines.append(f"#### {cls['name']}")
|
|
lines.append(f"*{cls['doc']}*\n")
|
|
|
|
for method in cls['methods']:
|
|
# Extract first line of docstring
|
|
first_line = method['doc'].split('\n')[0].strip()
|
|
lines.append(f"- `{method['name']}` - {first_line}")
|
|
|
|
lines.append("")
|
|
|
|
lines.append("")
|
|
|
|
return "\n".join(lines)
|
|
|
|
|
|
def generate_html_index(index: Dict) -> str:
|
|
"""Generate HTML representation of test index."""
|
|
html = ['<!DOCTYPE html><html><head>']
|
|
html.append('<meta charset="utf-8">')
|
|
html.append('<title>Contract Tests Index</title>')
|
|
html.append('<style>')
|
|
html.append('''
|
|
body { font-family: system-ui, -apple-system, sans-serif; max-width: 1200px; margin: 0 auto; padding: 20px; }
|
|
h1 { color: #2c3e50; border-bottom: 3px solid #3498db; padding-bottom: 10px; }
|
|
h2 { color: #34495e; margin-top: 40px; border-bottom: 2px solid #95a5a6; padding-bottom: 8px; }
|
|
h3 { color: #7f8c8d; margin-top: 30px; }
|
|
h4 { color: #95a5a6; margin-top: 20px; margin-bottom: 10px; }
|
|
.module-doc { font-style: italic; color: #7f8c8d; margin-bottom: 15px; }
|
|
.class-doc { font-style: italic; color: #95a5a6; margin-bottom: 10px; }
|
|
.test-method { margin-left: 20px; padding: 8px; background: #ecf0f1; margin-bottom: 5px; border-radius: 4px; }
|
|
.test-name { font-family: monospace; color: #2980b9; font-weight: bold; }
|
|
.test-doc { color: #34495e; margin-left: 10px; }
|
|
.error { background: #e74c3c; color: white; padding: 10px; border-radius: 4px; }
|
|
.domain-badge { display: inline-block; background: #3498db; color: white; padding: 3px 10px; border-radius: 12px; font-size: 12px; margin-left: 10px; }
|
|
''')
|
|
html.append('</style></head><body>')
|
|
|
|
html.append('<h1>Contract Tests Index</h1>')
|
|
html.append(f'<p>Total domains: {len(index)}</p>')
|
|
|
|
for domain, files in sorted(index.items()):
|
|
test_count = sum(len(f.get('classes', [])) for f in files.values())
|
|
html.append(f'<h2>{domain.capitalize()} <span class="domain-badge">{test_count} test classes</span></h2>')
|
|
|
|
for filename, file_info in sorted(files.items()):
|
|
if 'error' in file_info:
|
|
html.append(f'<h3>{filename} ⚠️</h3>')
|
|
html.append(f'<div class="error">Parse Error: {file_info["error"]}</div>')
|
|
continue
|
|
|
|
html.append(f'<h3>{filename}</h3>')
|
|
html.append(f'<div class="module-doc">{file_info["module_doc"]}</div>')
|
|
|
|
for cls in file_info['classes']:
|
|
html.append(f'<h4>{cls["name"]}</h4>')
|
|
html.append(f'<div class="class-doc">{cls["doc"]}</div>')
|
|
|
|
for method in cls['methods']:
|
|
first_line = method['doc'].split('\n')[0].strip()
|
|
html.append(f'<div class="test-method">')
|
|
html.append(f'<span class="test-name">{method["name"]}</span>')
|
|
html.append(f'<span class="test-doc">{first_line}</span>')
|
|
html.append('</div>')
|
|
|
|
html.append('</body></html>')
|
|
return '\n'.join(html)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# CLI usage
|
|
import sys
|
|
|
|
tests_dir = Path(__file__).parent / 'tests'
|
|
index = build_test_index(tests_dir)
|
|
|
|
if '--html' in sys.argv:
|
|
print(generate_html_index(index))
|
|
else:
|
|
print(generate_markdown_index(index))
|