| import json | |
| import pytest | |
| from solomon_mlx.budget import Budget | |
| def test_budget_reserves_before_dispatch_and_exhausts(tmp_path): | |
| path = tmp_path / "budget.json" | |
| budget = Budget(path, 20) | |
| with budget.reserve(15, "one"): | |
| assert json.loads(path.read_text())["entries"][0]["status"] == "reserved" | |
| with pytest.raises(RuntimeError), budget.reserve(1, "concurrent"): | |
| pass | |
| with pytest.raises(RuntimeError), budget.reserve(6, "too expensive"): | |
| pass | |
| def test_uncertain_dispatch_blocks_followups(tmp_path): | |
| budget = Budget(tmp_path / "budget.json") | |
| with pytest.raises(TimeoutError), budget.reserve(20, "uncertain"): | |
| raise TimeoutError() | |
| with pytest.raises(RuntimeError), budget.reserve(20, "retry"): | |
| pass | |