Mock ECS Task Run Stop Status Dict in Python

from datetime import datetime, timezone


def mock_ecs_task_status(task_id: str, state: str = "RUNNING") -> dict:
    """Return a mock ECS task status dictionary."""
    return {
        "taskArn": f"arn:aws:ecs:us-east-1:123456789012:task/cluster/{task_id}",
        "taskDefinition": "arn:aws:ecs:us-east-1:123456789012:task-definition/my-app:42",
        "lastStatus": state,
        "desiredStatus": "STOPPED" if state == "STOPPED" else "RUNNING",
        "clusterArn": "arn:aws:ecs:us-east-1:123456789012:cluster/my-cluster",
        "startedAt": "2025-01-15T10:00:00Z",
        "stoppedAt": datetime.now(timezone.utc).isoformat() if state == "STOPPED" else None,
        "stopCode": "EssentialContainerExited" if state == "STOPPED" else None,
        "stoppedReason": "Task stopped" if state == "STOPPED" else None,
    }


if __name__ == "__main__":
    running = mock_ecs_task_status("task-abc", "RUNNING")
    stopped = mock_ecs_task_status("task-def", "STOPPED")
    print("Running:", running)
    print("Stopped:", stopped)