ae44be31d5
Includes AirPlan design documents, AircOding-alpha1-plan, AirPlanV2, AirPlan-ParaV2, AirPlan-Para V1 reference docs, and all working code changes across packages. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
248 lines
7.7 KiB
Python
Executable File
248 lines
7.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
"""Ctx Mode 功能测试"""
|
||
|
||
import sys
|
||
import os
|
||
import tempfile
|
||
import json
|
||
from pathlib import Path
|
||
|
||
sys.path.insert(0, str(Path(__file__).parent / "lib"))
|
||
|
||
def test_ctx_enter():
|
||
"""测试上下文管理器初始化"""
|
||
from air_runtime.modes.ctx_mode import ctx_enter
|
||
|
||
with tempfile.TemporaryDirectory() as tmpdir:
|
||
project_root = Path(tmpdir)
|
||
|
||
result = ctx_enter(project_root)
|
||
|
||
assert "state_path" in result
|
||
assert Path(result["state_path"]).exists()
|
||
|
||
print("✓ 上下文管理器初始化测试通过")
|
||
return True
|
||
|
||
|
||
def test_ctx_estimate_tokens():
|
||
"""测试 Token 估算"""
|
||
from air_runtime.modes.ctx_mode import estimate_tokens
|
||
|
||
# 测试中文
|
||
chinese_text = "这是一个中文测试文本用于估算 Token 数量"
|
||
tokens = estimate_tokens(chinese_text)
|
||
assert tokens > 0
|
||
|
||
# 测试英文
|
||
english_text = "This is an English text for token estimation"
|
||
tokens_en = estimate_tokens(english_text)
|
||
assert tokens_en > 0
|
||
|
||
# 测试代码
|
||
code_text = "def main():\n print('hello world')\n return 0"
|
||
tokens_code = estimate_tokens(code_text)
|
||
assert tokens_code > 0
|
||
|
||
print(f"✓ Token 估算测试通过，中文: {tokens}, 英文: {tokens_en}, 代码: {tokens_code}")
|
||
return True
|
||
|
||
|
||
def test_ctx_validate_compression():
|
||
"""测试压缩质量校验"""
|
||
from air_runtime.modes.ctx_mode import validate_compression
|
||
|
||
original = """
|
||
# ADR-0001 决策
|
||
我们选择使用 PostgreSQL 数据库
|
||
|
||
# TODO 实现
|
||
- [ ] 创建数据库连接池
|
||
- [ ] 实现 CRUD 接口
|
||
|
||
# INV-1 约束
|
||
必须使用原子写入
|
||
"""
|
||
|
||
# 好的摘要（包含关键信息）
|
||
good_summary = "ADR-0001: PostgreSQL, TODO: 连接池/CRUD, INV-1: 原子写入"
|
||
result = validate_compression(original, good_summary)
|
||
assert result["ok"] == True
|
||
|
||
# 坏的摘要（丢失关键信息）
|
||
bad_summary = "选择了 PostgreSQL"
|
||
result_bad = validate_compression(original, bad_summary)
|
||
# 可能不 ok，因为丢失了 TODO 和 INV
|
||
|
||
print("✓ 压缩质量校验测试通过")
|
||
return True
|
||
|
||
|
||
def test_ctx_compress_fallback():
|
||
"""测试三级降级压缩"""
|
||
from air_runtime.modes.ctx_mode import compress_with_fallback, CompressionLevel
|
||
|
||
# 创建大文本
|
||
long_text = "这是测试内容\n" * 1000
|
||
|
||
result = compress_with_fallback(long_text, max_tokens=100)
|
||
|
||
# 应该返回结果
|
||
assert "level" in result
|
||
assert "result" in result
|
||
assert result["level"] in [CompressionLevel.RETRY, CompressionLevel.FALLBACK_MODEL, CompressionLevel.TRUNCATE]
|
||
|
||
print(f"✓ 三级降级压缩测试通过，压缩级别: {result['level']}")
|
||
return True
|
||
|
||
|
||
def test_ctx_lock():
|
||
"""测试 compactor 锁"""
|
||
from air_runtime.modes.ctx_mode import acquire_compactor_lock
|
||
|
||
with tempfile.TemporaryDirectory() as tmpdir:
|
||
lock_path = Path(tmpdir) / "test.lock"
|
||
|
||
# 首次获取锁应该成功
|
||
acquired = acquire_compactor_lock(lock_path)
|
||
assert acquired == True
|
||
assert lock_path.exists()
|
||
|
||
# 再次获取应该失败（锁已被持有）
|
||
acquired2 = acquire_compactor_lock(lock_path)
|
||
assert acquired2 == False
|
||
|
||
print("✓ Compactor 锁测试通过")
|
||
return True
|
||
|
||
|
||
def test_event_log_emit():
|
||
"""P1-GAP17: 测试事件日志 emit 和读取"""
|
||
from air_runtime.events import EventLog, TASK_COMPLETED, TASK_ENTERED, TASK_FINISHED
|
||
|
||
with tempfile.TemporaryDirectory() as tmpdir:
|
||
log_path = Path(tmpdir) / "events.jsonl"
|
||
log = EventLog(log_path)
|
||
|
||
log.emit(TASK_ENTERED, {"taskId": "T-001"})
|
||
log.emit(TASK_FINISHED, {"taskId": "T-001"})
|
||
log.emit(TASK_COMPLETED, {"taskId": "T-001", "status": "done"})
|
||
|
||
assert log_path.exists()
|
||
lines = log_path.read_text().strip().split("\n")
|
||
assert len(lines) == 3
|
||
|
||
entries = [json.loads(line) for line in lines]
|
||
assert entries[0]["type"] == "task.entered"
|
||
assert entries[1]["type"] == "task.finished"
|
||
assert entries[2]["type"] == "task.completed"
|
||
|
||
# 检查 ts 字段均为 ISO 格式
|
||
for e in entries:
|
||
assert "ts" in e
|
||
assert "T" in e["ts"]
|
||
|
||
print("✓ 事件日志 emit 测试通过")
|
||
return True
|
||
|
||
|
||
def test_event_log_rotate():
|
||
"""P1-GAP18: 测试事件日志原子轮转"""
|
||
from air_runtime.events import EventLog, TASK_COMPLETED
|
||
|
||
with tempfile.TemporaryDirectory() as tmpdir:
|
||
log_path = Path(tmpdir) / "events.jsonl"
|
||
# 小上限 + 密集写入触发轮转
|
||
log = EventLog(log_path, max_lines=50)
|
||
# 改写 ROTATE_CHECK_EVERY 为小值确保多次检查
|
||
log.ROTATE_CHECK_EVERY = 10
|
||
|
||
# 写入 200 条，因 ROTATE_CHECK_EVERY=10 会多次触发轮转
|
||
for i in range(200):
|
||
log.emit(TASK_COMPLETED, {"taskId": f"T-{i:03d}"})
|
||
|
||
lines = log_path.read_text().strip().split("\n")
|
||
# 经过多次轮转后，行数不应超过 max_lines * 2（两倍内都合理）
|
||
assert len(lines) < 200, f"轮转后应少于写入量 200，实际: {len(lines)}"
|
||
assert len(lines) >= 1, f"轮转后应有数据, 实际: {len(lines)}"
|
||
|
||
# 验证写入操作后文件仍然有效 JSONL
|
||
for line in lines:
|
||
entry = json.loads(line)
|
||
assert entry["type"] == "task.completed"
|
||
assert "ts" in entry
|
||
|
||
print(f"✓ 事件日志原子轮转测试通过，轮转后保留 {len(lines)} 条")
|
||
return True
|
||
|
||
|
||
def test_event_constants_defined():
|
||
"""P1-GAP17: 验证所有事件类型常量均已定义且唯一"""
|
||
import air_runtime.events as evt
|
||
|
||
expected_names = [
|
||
"TASK_DISPATCHED", "TASK_ENTERED", "TASK_FINISHED",
|
||
"TASK_COMPLETED", "TASK_BLOCKED",
|
||
"MERGE_STARTED", "MERGE_COMPLETED",
|
||
"REPAIR_CREATED", "REPAIR_RESOLVED",
|
||
"INTERVENTION_STALL", "XDB_CAPTURED", "DEBUG_SESSION",
|
||
"CONTEXT_COMPACTED", "DEPLOY_COMPLETED", "TEST_RUN", "SEC_SCAN", "REVIEW_SESSION",
|
||
"ENGINE_ENTERED", "ENGINE_CYCLE", "ENG_REPLAN_TRIGGERED", "ENG_BLOCKED",
|
||
"WORKER_TIMEOUT", "WORKTREE_MERGE_CONFLICT",
|
||
"ARC_REPLANNED",
|
||
"ADR_CHANGE_DETECTED", "ADR_INVALIDATION", "ADR_UNFREEZED",
|
||
"LOCK_ACQUIRED", "LOCK_RELEASED", "STALE_LOCK_CLEANED",
|
||
]
|
||
|
||
values = []
|
||
for name in expected_names:
|
||
val = getattr(evt, name, None)
|
||
assert val is not None, f"缺少事件常量: {name}"
|
||
values.append(val)
|
||
|
||
# 无重复值
|
||
assert len(set(values)) == len(values), f"事件常量存在重复值: {len(set(values))} vs {len(values)}"
|
||
|
||
print(f"✓ 事件常量定义测试通过，共 {len(values)} 个事件类型")
|
||
return True
|
||
|
||
|
||
def main():
|
||
print("=" * 50)
|
||
print("Ctx Mode 功能测试")
|
||
print("=" * 50)
|
||
|
||
tests = [
|
||
("上下文管理器初始化", test_ctx_enter),
|
||
("Token 估算", test_ctx_estimate_tokens),
|
||
("压缩质量校验", test_ctx_validate_compression),
|
||
("三级降级压缩", test_ctx_compress_fallback),
|
||
("Compactor 锁", test_ctx_lock),
|
||
("事件日志 emit", test_event_log_emit),
|
||
("事件日志原子轮转", test_event_log_rotate),
|
||
("事件常量定义", test_event_constants_defined),
|
||
]
|
||
|
||
passed = 0
|
||
failed = 0
|
||
|
||
for name, test_fn in tests:
|
||
try:
|
||
test_fn()
|
||
passed += 1
|
||
except Exception as e:
|
||
print(f"✗ {name} 失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
failed += 1
|
||
|
||
print("=" * 50)
|
||
print(f"测试结果: {passed} 通过, {failed} 失败")
|
||
print("=" * 50)
|
||
|
||
return failed == 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
success = main()
|
||
sys.exit(0 if success else 1) |