import time
from energiamesh.core import LocalProblem, SharedVariables, PlanDelta, DualVariables, AuditLog


def test_core_dataclasses_simple():
    lp = LocalProblem(site_id="site-1", problem_id="p1", description="test", data={"a": 1})
    sv = SharedVariables(signals={"forecast": 10}, version=1)
    dv = DualVariables(multipliers={"lambda": 0.5}, version=1)
    pd = PlanDelta(delta={"dx": 1}, metadata={"source": "test"}, timestamp=time.time())
    log = AuditLog()
    log.log("created", {"obj": lp.problem_id})

    assert lp.site_id == "site-1"
    assert sv.version == 1
    assert dv.multipliers["lambda"] == 0.5
    assert "dx" in pd.delta
    assert len(log.entries) == 1