d5ab1fd26f
- Simulation model: add tactic_ids JSON column (nullable=False, default=[])
- Migration 0004: ADD COLUMN tactic_ids (server_default='[]', no batch needed)
- mitre.py: add _TACTIC_IDS map, lookup_tactic(), get_tactic_name()
- simulation_workflow.py: done guard (409) before RBAC; SOC gate += tactic_ids;
_resolve_tactic_ids() validates against hardcoded map; auto-transition += tactic_ids;
transition done→review_required is Reopen (all 3 roles); _maybe_activate_engagement hook
- serializers.py: _enrich_tactics() → serialize_simulation adds tactics:[{id,name}]
- test_simulations_tactics.py: valid/invalid/dedup/SOC gate/auto-transition/no-bundle
- test_simulations_done_readonly.py: 409 all roles, Reopen all roles, invalid transitions, after-reopen ok
- test_engagement_lifecycle.py: planned→active on auto-transition, already active/closed unchanged, migration 0004 round-trip
- Updated test_simulations_patch.py + test_simulations_workflow.py for AC-18 behavior
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
"""JSON serializers for API responses."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from backend.app.models import Engagement, User
|
|
from backend.app.models.simulation import Simulation
|
|
|
|
|
|
def serialize_user(user: User) -> dict[str, Any]:
|
|
return {
|
|
"id": user.id,
|
|
"username": user.username,
|
|
"role": user.role.value,
|
|
"created_at": user.created_at.isoformat() if user.created_at else None,
|
|
}
|
|
|
|
|
|
def serialize_user_brief(user: User) -> dict[str, Any]:
|
|
return {"id": user.id, "username": user.username}
|
|
|
|
|
|
def _enrich_techniques(raw: list[dict[str, Any]]) -> list[dict[str, Any]]:
|
|
"""Attach tactics to each {id, name} snapshot from the MITRE service."""
|
|
from backend.app.services import mitre as mitre_svc
|
|
|
|
return [
|
|
{"id": t["id"], "name": t["name"], "tactics": mitre_svc.get_tactics(t["id"])}
|
|
for t in (raw or [])
|
|
]
|
|
|
|
|
|
def _enrich_tactics(tactic_ids: list[str]) -> list[dict[str, str]]:
|
|
"""Resolve TA-ids to {id, name} at runtime."""
|
|
from backend.app.services import mitre as mitre_svc
|
|
|
|
result = []
|
|
for tid in tactic_ids or []:
|
|
entry = mitre_svc.lookup_tactic(tid)
|
|
if entry is not None:
|
|
result.append(entry)
|
|
else:
|
|
result.append({"id": tid, "name": ""})
|
|
return result
|
|
|
|
|
|
def serialize_simulation(simulation: Simulation) -> dict[str, Any]:
|
|
return {
|
|
"id": simulation.id,
|
|
"engagement_id": simulation.engagement_id,
|
|
"name": simulation.name,
|
|
"techniques": _enrich_techniques(simulation.techniques or []),
|
|
"tactics": _enrich_tactics(simulation.tactic_ids or []),
|
|
"description": simulation.description,
|
|
"commands": simulation.commands,
|
|
"prerequisites": simulation.prerequisites,
|
|
"executed_at": simulation.executed_at.isoformat() if simulation.executed_at else None,
|
|
"execution_result": simulation.execution_result,
|
|
"log_source": simulation.log_source,
|
|
"logs": simulation.logs,
|
|
"soc_comment": simulation.soc_comment,
|
|
"incident_number": simulation.incident_number,
|
|
"status": simulation.status.value,
|
|
"created_at": simulation.created_at.isoformat() if simulation.created_at else None,
|
|
"updated_at": simulation.updated_at.isoformat() if simulation.updated_at else None,
|
|
"created_by": serialize_user_brief(simulation.created_by) # type: ignore[arg-type]
|
|
if simulation.created_by
|
|
else None,
|
|
}
|
|
|
|
|
|
def serialize_engagement(engagement: Engagement) -> dict[str, Any]:
|
|
return {
|
|
"id": engagement.id,
|
|
"name": engagement.name,
|
|
"description": engagement.description,
|
|
"start_date": engagement.start_date.isoformat() if engagement.start_date else None,
|
|
"end_date": engagement.end_date.isoformat() if engagement.end_date else None,
|
|
"status": engagement.status.value,
|
|
"created_at": engagement.created_at.isoformat() if engagement.created_at else None,
|
|
"created_by": serialize_user_brief(engagement.created_by) # type: ignore[arg-type]
|
|
if engagement.created_by
|
|
else None,
|
|
}
|