from energysphere.models import LocalProblem
from energysphere.energi_bridge import EnergiBridgeMapper


def test_map_local_problem_to_object():
    lp = LocalProblem(site_id="site-42", objective="min-cost", parameters={"load": 2.5})
    mapper = EnergiBridgeMapper(namespace="test-ns")
    obj = mapper.map_local_problem(lp)
    assert obj["type"] == "Object"
    assert obj["site_id"] == "site-42"
    assert obj["namespace"] == "test-ns"
    assert obj["version"] == 1


def test_map_object_to_local_problem_roundtrip():
    lp = LocalProblem(site_id="site-7", objective="min-cost", parameters={"p": 1})
    mapper = EnergiBridgeMapper()
    obj = mapper.map_local_problem(lp)
    lp2 = mapper.map_object_to_local_problem(obj)
    assert lp2.site_id == lp.site_id
    assert lp2.objective == lp.objective
    assert lp2.parameters == lp.parameters