import pytest
from energi_bridge.schemas import LocalProblem, SharedVariables, PlanDelta, DualVariables, AuditLog, PrivacyBudget
from datetime import datetime


def test_local_problem_defaults():
    lp = LocalProblem(id="lp1", domain="solar", assets=["panel1"])
    assert lp.id == "lp1"
    assert lp.domain == "solar"
    assert lp.assets == ["panel1"]
    assert isinstance(lp.constraints, dict)


def test_shared_variables_defaults():
    sv = SharedVariables(version=1)
    assert sv.version == 1
    assert sv.aggregation_method == "mean"


def test_plan_delta_creation():
    pd = PlanDelta(delta={"dispatch": "up"}, timestamp=datetime.utcnow(), author="tester", contract_id="c1")
    assert pd.delta["dispatch"] == "up"
    assert pd.author == "tester"


def test_dual_variables_defaults():
    dv = DualVariables()
    assert dv.lagrange_multipliers == {}