tests/test_telegram_benchmark_live.py

import asyncio
import unittest
from unittest.mock import AsyncMock, patch

from api.telegram_webhook import _cmd_bench


class TelegramExtendedBenchmarkTests(unittest.IsolatedAsyncioTestCase):
    async def test_bench_starts_background_extended_run(self):
        runner = AsyncMock()
        with patch("api.benchmark_handler.run_benchmark_task", runner):
            await _cmd_bench(123)
            await asyncio.sleep(0)
        runner.assert_awaited_once()
        args, kwargs = runner.await_args
        self.assertEqual(args[0], 123)
        self.assertEqual(kwargs["mode"], "full")


if __name__ == "__main__":
    unittest.main()