from energysphere.adapters.substation_meter import SubstationMeterAdapter
from energysphere.registry import GraphContractRegistry
from energysphere.models import LocalProblem


def test_substation_meter_adapter_generates_local_problem():
    registry = GraphContractRegistry()
    adapter = SubstationMeterAdapter(registry)
    lp = adapter.collect_local_problem()
    assert isinstance(lp, LocalProblem)
    # The adapter should have registered the problem into the registry
    contracts = registry.list_contracts()
    assert any(c.get("site_id") == lp.site_id for c in contracts)