from datetime import datetime, timezone, timedelta
from urllib.parse import urlencode

import pytest
from uuid import uuid4
from httpx import AsyncClient

from jb.config import settings


@pytest.mark.anyio
async def test_get_statuses(httpxclient: AsyncClient, no_limit, amt_worker_id):
    # Expects settings.fsb_host to point to a functional thl-fsb
    client = httpxclient
    now = datetime.now(tz=timezone.utc)

    params = {"worker_id": amt_worker_id}
    res = await client.get(f"/statuses/", params=params)
    assert res.status_code == 200
    assert res.json() == []

    params = {"worker_id": amt_worker_id, "started_after": now.isoformat()}
    res = await client.get(f"/statuses/", params=params)
    assert res.status_code == 422
    assert "Input should be a valid integer" in res.text


@pytest.fixture
def fsb_get_statuses_example_response(amt_worker_id, tsid1, tsid2):
    return {
        "tasks_status": [
            {
                "tsid": tsid1,
                "product_id": settings.product_id,
                "bpuid": amt_worker_id,
                "started": "2025-06-12T03:27:24.902280Z",
                "finished": "2025-06-12T03:29:37.626481Z",
                "status": 2,
                "payout": 0,
                "user_payout": None,
                "payout_format": None,
                "user_payout_string": None,
                "kwargs": {},
                "status_code_1": "SESSION_START_QUALITY_FAIL",
                "status_code_2": "ENTRY_URL_MODIFICATION",
            },
            {
                "tsid": tsid2,
                "product_id": settings.product_id,
                "bpuid": amt_worker_id,
                "started": "2025-06-12T03:30:18.176826Z",
                "finished": "2025-06-12T03:36:58.789059Z",
                "status": 2,
                "payout": 0,
                "user_payout": None,
                "payout_format": None,
                "user_payout_string": None,
                "kwargs": {},
                "status_code_1": "BUYER_QUALITY_FAIL",
                "status_code_2": None,
            },
        ]
    }


@pytest.mark.anyio
async def test_get_statuses_mock(
    httpxclient: AsyncClient,
    no_limit,
    amt_worker_id,
    mock_requests,
    fsb_get_statuses_example_response,
    tsid1,
    tsid2,
):
    client = httpxclient
    now = datetime.now(tz=timezone.utc)
    started_after = now - timedelta(minutes=5)

    # The fsb call we are mocking ------v
    params = {
        "bpuid": amt_worker_id,
        "started_after": round(started_after.timestamp()),
        "started_before": round(now.timestamp()),
    }
    url = f"{settings.fsb_host}{settings.product_id}/status/" + "?" + urlencode(params)
    mock_requests.get(url, json=fsb_get_statuses_example_response, status_code=200)
    # ---- end mock

    params = {
        "worker_id": amt_worker_id,
        "started_after": round(started_after.timestamp()),
        "started_before": round(now.timestamp()),
    }
    result = await client.get(f"/statuses/", params=params)
    assert result.status_code == 200
    res = result.json()
    assert len(res) == 2
    assert res == [
        {"status": 2, "tsid": tsid1},
        {"status": 2, "tsid": tsid2},
    ]