"""CLI calculation mode workflow."""
from __future__ import annotations
import pytest
from pytest_aitest import Agent, Provider
from conftest import assert_cli_exit_codes, assert_regex, unique_path
pytestmark = [pytest.mark.aitest, pytest.mark.cli]
@pytest.mark.asyncio
async def test_cli_calculation_mode_batch_flow(aitest_run, excel_cli_server, excel_cli_skill):
agent = Agent(
name="cli-calc-mode",
provider=Provider(model="azure/gpt-5-mini", rpm=10, tpm=10000),
cli_servers=[excel_cli_server],
skill=excel_cli_skill,
max_turns=20,
)
prompt = f"""
Create a new Excel file at {unique_path('calc-mode-cli')}
Set calculation mode to manual.
On Sheet1, write this data in A1:C4:
Category, Budget, Actual
Rent, 1000, 1000
Food, 500, 450
Transport, 200, 180
Add a formula in D2:D4 for Variance = C2-B2, C3-B3, C4-B4.
After all writes, explicitly recalculate the workbook.
Switch calculation mode back to automatic.
Report the current calculation mode and the variance values.
"""
result = await aitest_run(agent, prompt)
assert result.success
assert_cli_exit_codes(result)
assert_regex(result.final_response, r"(?i)(manual|automatic|calculation)")
