import time
from gridverse.energi_bridge import EnergiBridge


def test_to_ir_and_from_ir_roundtrip():
    contract_type = "LocalProblem"
    payload = {"id": "lp-1", "name": "test", "variables": {"p": 1}}

    ir = EnergiBridge.to_ir(contract_type, payload)

    assert isinstance(ir, dict)
    assert ir.get("ir_type") == contract_type
    assert ir.get("ir_payload") == payload
    assert "metadata" in ir

    back_type, back_payload = EnergiBridge.from_ir(ir)
    assert back_type == contract_type
    assert back_payload == payload


def test_map_gridverse_to_catopt_basic():
    gridverse_contract = {
        "type": "LocalProblem",
        "payload": {"id": "lp-2", "name": "second"},
    }
    ir = EnergiBridge.map_gridverse_to_catopt(gridverse_contract)
    assert isinstance(ir, dict)
    assert ir.get("ir_type") == "LocalProblem"
    assert ir.get("ir_payload") == gridverse_contract["payload"]