JiuContinent
db751f03b4
refactor(Dockerfile): 使用requirements.txt替代硬编码依赖 将Dockerfile中的硬编码pip包列表替换为通过requirements.txt文件安装， 提高依赖管理的灵活性和可维护性。 feat(scheduling): 移除内置APScheduler，采用独立调度系统 移除app/core/background/scheduler.py中原来的APScheduler实现， 改为使用新的应用级调度系统app.application.scheduling。 refactor(task_queue): 调整任务队列模块结构和导入路径 将任务队列相关常量从app.core.background.task_queue.constants迁移至 app.tasks.constants，并更新所有相关导入路径和引用。 refactor(events): 将事件服务重构为独立的应用层组件 将事件分发逻辑从核心层迁移到应用层，使用app.application.events.EventDispatchService 替代原有的app.modules.events.services.EventService。 feat(ai_memory): 增强AI记忆自动写入的安全策略 新增ai_memory_blocked_content_terms配置项用于阻止敏感内容， 添加TTL过期机制控制自动写入条目的生命周期。 fix(security): 强化生产环境安全验证机制 增加model_validator确保生产环境中数据库连接、API密钥、CORS设置等 关键安全配置符合要求。 feat(risks): 优化风险事件操作动作的外键约束 为RiskEventAction模型的风险事件ID字段添加外键约束， 防止孤立记录并增强数据完整性。 refactor(audit): 优化审计服务方法命名和事务处理 将AuditService的log方法重命名为record以反映其阶段行为， 并调整事务提交时机以提高性能。 feat(events): 增强领域事件并发处理和响应模型 添加事件锁定机制防止重复处理，更新API响应模型以提供 更准确的数据类型定义。 ```
76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
from alembic import command
|
|
from alembic.config import Config
|
|
from sqlalchemy import create_engine, func, select
|
|
from sqlalchemy.orm import Session, sessionmaker
|
|
|
|
from app.core.config import get_settings
|
|
from app.core.config import Settings
|
|
from app.core.database import Base
|
|
from app.core.security import OperationsDisabledError
|
|
from app.modules.audit.models import AuditLog
|
|
from app.modules.audit.schemas import AuditLogCreate
|
|
from app.modules.audit.service import AuditService
|
|
from app.modules.events.models import DomainEvent
|
|
from app.modules.events.services import EventService
|
|
from app.modules.risk.services import RiskService
|
|
|
|
|
|
def test_audit_and_outbox_rollback_with_business_transaction() -> None:
|
|
engine = create_engine("sqlite://")
|
|
Base.metadata.create_all(engine)
|
|
session_factory = sessionmaker(bind=engine, expire_on_commit=False)
|
|
|
|
with session_factory() as db:
|
|
AuditService(db).record(AuditLogCreate(action="transaction.rollback"))
|
|
EventService(db).enqueue(
|
|
event_type="test.rollback",
|
|
source="pytest",
|
|
aggregate_type="test",
|
|
aggregate_id="rollback",
|
|
idempotency_key="test-transaction-rollback",
|
|
)
|
|
db.rollback()
|
|
|
|
with session_factory() as db:
|
|
assert db.scalar(select(func.count()).select_from(AuditLog)) == 0
|
|
assert db.scalar(select(func.count()).select_from(DomainEvent)) == 0
|
|
|
|
|
|
def test_business_service_enforces_read_only_policy(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("READ_ONLY_MODE", "true")
|
|
get_settings.cache_clear()
|
|
try:
|
|
with pytest.raises(OperationsDisabledError):
|
|
RiskService(Session()).generate_events(actor="pytest")
|
|
finally:
|
|
get_settings.cache_clear()
|
|
|
|
|
|
def test_production_settings_fail_closed() -> None:
|
|
with pytest.raises(ValueError, match="PostgreSQL"):
|
|
Settings(
|
|
app_env="production",
|
|
database_url="sqlite:///unsafe.db",
|
|
api_key="api",
|
|
audit_api_key="audit",
|
|
cors_origins=["https://internal.example.com"],
|
|
)
|
|
|
|
|
|
def test_migrations_match_sqlalchemy_metadata(
|
|
tmp_path: Path,
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
database_path = tmp_path / "migration-check.db"
|
|
monkeypatch.setenv("DATABASE_URL", f"sqlite:///{database_path}")
|
|
get_settings.cache_clear()
|
|
config = Config("alembic.ini")
|
|
try:
|
|
command.upgrade(config, "head")
|
|
command.check(config)
|
|
finally:
|
|
get_settings.cache_clear()
|