import pytest

from cosmosmesh_privacy_preserving_federated_.reference_adapters import (
    RoverPlannerAdapter,
    HabitatLifeSupportAdapter,
)


def test_rover_adapter_basic():
    rover = RoverPlannerAdapter()
    assert isinstance(rover.readState(), dict)
    data = rover.exposeLocalProblemData()
    assert isinstance(data, dict)
    resp = rover.applyCommand({"cmd": "test"})
    assert resp.get("status") == "accepted"


def test_habitat_adapter_basic():
    hab = HabitatLifeSupportAdapter()
    assert isinstance(hab.readState(), dict)
    data = hab.exposeLocalProblemData()
    assert isinstance(data, dict)
    resp = hab.applyCommand({"cmd": "test"})
    assert resp.get("status") == "accepted"