from energysphere.registry import GraphContractRegistry
from energysphere.models import LocalProblem


def test_registry_register_and_list():
    reg = GraphContractRegistry()
    lp = LocalProblem(site_id="site-1", objective="min-cost", parameters={"load": 1.0})
    reg.register_local_problem("contract-site-1", lp)
    contracts = reg.list_contracts()
    assert len(contracts) == 1
    assert contracts[0]["site_id"] == "site-1"