import asyncio
import json
import os
import sys
from unittest.mock import patch

import pytest
from fastapi.testclient import TestClient

sys.path.insert(
    0, os.path.abspath("../../..")
)  # Adds the parent directory to the system path

from litellm.constants import MAX_IN_MEMORY_QUEUE_FLUSH_COUNT
from litellm.proxy.db.db_transaction_queue.base_update_queue import BaseUpdateQueue


@pytest.mark.asyncio
async def test_queue_flush_limit():
    """
    Test to ensure we don't dequeue more than MAX_IN_MEMORY_QUEUE_FLUSH_COUNT items.
    """
    # Arrange
    queue = BaseUpdateQueue()
    # Override maxsize so the queue can hold all test items without blocking.
    # The default LITELLM_ASYNCIO_QUEUE_MAXSIZE (1000) equals MAX_IN_MEMORY_QUEUE_FLUSH_COUNT,
    # so adding more items than that would cause `await queue.put()` to block forever.
    items_to_add = MAX_IN_MEMORY_QUEUE_FLUSH_COUNT + 100
    queue.update_queue = asyncio.Queue(maxsize=items_to_add + 1)

    for i in range(items_to_add):
        await queue.add_update(f"test_update_{i}")

    # Act
    flushed_updates = await queue.flush_all_updates_from_in_memory_queue()

    # Assert
    assert (
        len(flushed_updates) == MAX_IN_MEMORY_QUEUE_FLUSH_COUNT
    ), f"Expected {MAX_IN_MEMORY_QUEUE_FLUSH_COUNT} items, but got {len(flushed_updates)}"

    # Verify remaining items are still in queue
    assert (
        queue.update_queue.qsize() == 100
    ), "Expected 100 items to remain in the queue"


def test_misconfigured_queue_thresholds_warns():
    """
    Test that a warning is logged when MAX_SIZE_IN_MEMORY_QUEUE >= LITELLM_ASYNCIO_QUEUE_MAXSIZE.

    This misconfiguration causes the spend aggregation check in SpendUpdateQueue.add_update()
    to never trigger because asyncio.Queue blocks before qsize() can reach the threshold.
    """
    import litellm.proxy.db.db_transaction_queue.base_update_queue as bq_module

    with patch.object(bq_module, "MAX_SIZE_IN_MEMORY_QUEUE", 2000), patch.object(
        bq_module, "LITELLM_ASYNCIO_QUEUE_MAXSIZE", 1000
    ), patch.object(bq_module.verbose_proxy_logger, "warning") as mock_warning:
        BaseUpdateQueue()
        mock_warning.assert_called_once()
        assert "Misconfigured queue thresholds" in mock_warning.call_args[0][0]