from gridguard_secure_attested_cross_domain_e.bridge import EnergiBridge


def test_energi_bridge_roundtrip():
    local = {"asset_id": "substation-1", "objective": {"dispatch": 5}}
    canonical = EnergiBridge.to_canonical(local)
    restored = EnergiBridge.from_canonical(canonical)
    assert isinstance(restored, dict)
    assert "LocalProblems" in restored
    assert restored["LocalProblems"][0] == local