import pytest
from unittest.mock import patch, AsyncMock
from datetime import datetime
from app.models.session import SessionType, TaskStatus, CreateSessionRequest
from app.services.session_manager import SessionManager


class TestSessionManager:
    @pytest.fixture(autouse=True)
    def setup(self, mock_settings, mock_opencode_client):
        self.manager = SessionManager()
        self.manager.tasks = {}
        self.mock_client = mock_opencode_client

    @pytest.mark.asyncio
    async def test_create_task(self):
        with patch("app.services.session_manager.opencode_client", self.mock_client):
            request = CreateSessionRequest(
                type=SessionType.EPHEMERAL, prompt="test prompt"
            )

            task = await self.manager.create_task(request)

            assert task.id is not None
            assert task.type == SessionType.EPHEMERAL
            assert task.prompt == "test prompt"
            assert task.status == TaskStatus.PENDING
            assert task.created_at is not None

    @pytest.mark.asyncio
    async def test_execute_task_sync(self):
        with patch("app.services.session_manager.opencode_client", self.mock_client):
            request = CreateSessionRequest(prompt="sync task")
            task = await self.manager.create_task(request)

            result = await self.manager.execute_task(task.id)

            assert result is not None
            updated_task = await self.manager.get_task(task.id)
            assert updated_task.status == TaskStatus.COMPLETED

    @pytest.mark.asyncio
    async def test_execute_task_async(self):
        with patch("app.services.session_manager.opencode_client", self.mock_client):
            request = CreateSessionRequest(prompt="async task")
            task = await self.manager.create_task(request)

            await self.manager.execute_task_async(task.id)

            updated_task = await self.manager.get_task(task.id)
            assert updated_task.status == TaskStatus.RUNNING
            assert updated_task.session_id is not None

    @pytest.mark.asyncio
    async def test_abort_task(self):
        with patch("app.services.session_manager.opencode_client", self.mock_client):
            request = CreateSessionRequest(prompt="abortable task")
            task = await self.manager.create_task(request)
            await self.manager.execute_task_async(task.id)

            result = await self.manager.abort_task(task.id)

            assert result is True
            updated_task = await self.manager.get_task(task.id)
            assert updated_task.status == TaskStatus.ABORTED

    @pytest.mark.asyncio
    async def test_get_task_not_found(self):
        with patch("app.services.session_manager.opencode_client", self.mock_client):
            task = await self.manager.get_task("nonexistent")
            assert task is None

    @pytest.mark.asyncio
    async def test_list_tasks(self):
        with patch("app.services.session_manager.opencode_client", self.mock_client):
            request1 = CreateSessionRequest(prompt="task 1")
            request2 = CreateSessionRequest(prompt="task 2")

            await self.manager.create_task(request1)
            await self.manager.create_task(request2)

            tasks = await self.manager.list_tasks()
            assert len(tasks) == 2