mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-16 09:40:30 +08:00
6070da9ae2
* fix: resume ChatUI streams after refresh * fix: hide chat run service errors * fix: harden resumed chat streams
31 lines
897 B
Python
31 lines
897 B
Python
import asyncio
|
|
|
|
import pytest
|
|
|
|
from astrbot.core.platform.sources.webchat.webchat_queue_mgr import WebChatQueueMgr
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_removed_back_queue_unblocks_pending_writer():
|
|
queue_manager = WebChatQueueMgr(back_queue_maxsize=1)
|
|
request_id = "request-1"
|
|
queue = queue_manager.get_or_create_back_queue(request_id, "conversation-1")
|
|
await queue.put({"type": "plain", "data": "first"})
|
|
|
|
blocked_writer = asyncio.create_task(
|
|
queue_manager.put_back_queue(
|
|
request_id,
|
|
{"type": "plain", "data": "second"},
|
|
)
|
|
)
|
|
await asyncio.sleep(0)
|
|
assert not blocked_writer.done()
|
|
|
|
queue_manager.remove_back_queue(request_id)
|
|
|
|
assert await asyncio.wait_for(blocked_writer, timeout=1) is False
|
|
assert not await queue_manager.put_back_queue(
|
|
request_id,
|
|
{"type": "plain", "data": "late"},
|
|
)
|