import time
from exoroute import LocalProblem, SharedVariables, PlanDelta, EnergiBridge


def test_energi_bridge_map_to_ir_basic():
    lp = LocalProblem(id="LP-001", domain="equities", assets=["AAPL", "MSFT"], objective="min_latency", constraints={})
    sv = SharedVariables(forecasts={"AAPL": 150.0}, priors={"AAPL": 149.5}, version=1)
    pd = PlanDelta(delta={"route": {"AAPL": "optim"}}, timestamp=time.time(), author="tester", contract_id=lp.id)

    bridge = EnergiBridge()
    ir = bridge.map_to_ir(lp, sv, pd)

    # Basic structural checks
    assert "object" in ir
    assert ir["object"]["type"] == "LocalProblem"
    assert ir["morphisms"]["shared_variables"]["forecasts"]["AAPL"] == 150.0
    assert ir["plan_delta"]["author"] == "tester"