import uuid
from datetime import datetime
from unittest.mock import AsyncMock, MagicMock

import pytest

from proxy_pool.plugins.protocols import CheckContext, CheckResult
from proxy_pool.worker.tasks_validate import _run_single_check


class TestRunSingleCheck:
    async def test_successful_check(self):
        checker = MagicMock()
        checker.name = "test"
        checker.timeout = 5.0
        checker.check = AsyncMock(
            return_value=CheckResult(passed=True, detail="ok")
        )

        proxy = MagicMock()
        proxy.ip = "1.2.3.4"
        proxy.port = 8080
        proxy.protocol = MagicMock(value="http")

        context = CheckContext(
            started_at=datetime.now(),
            http_client=None,
        )

        result = await _run_single_check(checker, proxy, context)

        assert result.passed is True
        assert result.detail == "ok"

    async def test_timeout_returns_failed(self):
        async def slow_check(*args, **kwargs):
            import asyncio
            await asyncio.sleep(10)

        checker = MagicMock()
        checker.name = "slow"
        checker.timeout = 0.1
        checker.check = slow_check

        proxy = MagicMock()
        proxy.ip = "1.2.3.4"
        proxy.port = 8080
        proxy.protocol = MagicMock(value="http")

        context = CheckContext(
            started_at=datetime.now(),
            http_client=None,
        )

        result = await _run_single_check(checker, proxy, context)

        assert result.passed is False
        assert "timed out" in result.detail

    async def test_exception_returns_failed(self):
        checker = MagicMock()
        checker.name = "broken"
        checker.timeout = 5.0
        checker.check = AsyncMock(side_effect=ConnectionError("boom"))

        proxy = MagicMock()
        proxy.ip = "1.2.3.4"
        proxy.port = 8080
        proxy.protocol = MagicMock(value="http")

        context = CheckContext(
            started_at=datetime.now(),
            http_client=None,
        )

        result = await _run_single_check(checker, proxy, context)

        assert result.passed is False
        assert "ConnectionError" in result.detail
        assert "boom" in result.detail