from application.forecast import ForecastService
from infrastructure.ai.assistant import AssistantService


class TestAssistantService:
    def test_analyze_returns_prompt(self, sample_model):
        forecast_service = ForecastService(sample_model)
        assistant = AssistantService(sample_model, forecast_service=forecast_service)
        result = assistant.analyze(months=6)
        assert "prompt" in result
        assert "summary" in result
        assert result["ai_response"] is None

    def test_advice_returns_prompt(self, sample_model):
        forecast_service = ForecastService(sample_model)
        assistant = AssistantService(sample_model, forecast_service=forecast_service)
        result = assistant.advice("Как мне сэкономить?", months=6)
        assert "prompt" in result
        assert "ai_response" in result