from __future__ import annotations

from jarvis_gateway.adapters.base import BaseAdapter
from jarvis_gateway.adapters.health_risk_cockpit import HealthRiskCockpitAdapter
from jarvis_gateway.contracts import ModuleId, ModuleSnapshot


class MockHealthAdapter(BaseAdapter):
    module_id = ModuleId.HEALTH
    title = "Health Risk Cockpit"

    def get_snapshot(self) -> ModuleSnapshot:
        return HealthRiskCockpitAdapter(dashboard_available=False, stale_data=True).get_snapshot()
