from citygrid.adapters.der_controller import DerControllerAdapter
from citygrid.adapters.water_pump_controller import WaterPumpControllerAdapter
from citygrid.bridge.energi_bridge import EnergiBridge


def test_adapters_to_canonical_and_ack():
    der = DerControllerAdapter()
    water = WaterPumpControllerAdapter()
    lp_der = der.build_local_problem()
    lp_water = water.build_local_problem()
    canon_der = EnergiBridge.to_canonical({"id": lp_der.id, "domain": lp_der.domain, "assets": lp_der.assets, "objective": lp_der.objective, "constraints": lp_der.constraints})
    canon_water = EnergiBridge.to_canonical({"id": lp_water.id, "domain": lp_water.domain, "assets": lp_water.assets, "objective": lp_water.objective, "constraints": lp_water.constraints})
    assert canon_der.domain == "electricity"
    assert canon_water.domain == "water"