Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,59 @@
|
||||
"""Regression tests for the Feishu adapter's owned SDK executor.
|
||||
|
||||
Blocking Feishu SDK calls used to run on asyncio's shared default executor.
|
||||
When that executor was torn down (agent thread exit / loop cleanup), every
|
||||
subsequent send failed permanently with "Executor shutdown has been called"
|
||||
and the gateway became a zombie. The adapter now owns its own
|
||||
ThreadPoolExecutor and recreates it on demand if it has been shut down.
|
||||
|
||||
Covers: #10849
|
||||
"""
|
||||
import concurrent.futures
|
||||
|
||||
import pytest
|
||||
|
||||
from plugins.platforms.feishu.adapter import FeishuAdapter
|
||||
|
||||
|
||||
def _bare_adapter() -> FeishuAdapter:
|
||||
"""A FeishuAdapter with only the executor fields wired (no __init__)."""
|
||||
adapter = object.__new__(FeishuAdapter)
|
||||
import threading
|
||||
|
||||
adapter._sdk_executor_lock = threading.Lock()
|
||||
adapter._sdk_executor = None
|
||||
adapter._sdk_executor_closing = False
|
||||
return adapter
|
||||
|
||||
|
||||
def test_get_executor_recreates_after_shutdown():
|
||||
"""A shut-down pool must be transparently replaced — the #10849 recovery."""
|
||||
adapter = _bare_adapter()
|
||||
first = adapter._get_sdk_executor()
|
||||
first.shutdown(wait=True)
|
||||
assert getattr(first, "_shutdown", False) is True
|
||||
|
||||
second = adapter._get_sdk_executor()
|
||||
assert second is not first
|
||||
assert getattr(second, "_shutdown", False) is False
|
||||
adapter._shutdown_sdk_executor()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_run_blocking_executes_on_owned_pool():
|
||||
adapter = _bare_adapter()
|
||||
captured = {}
|
||||
|
||||
def _work(value):
|
||||
import threading
|
||||
|
||||
captured["thread"] = threading.current_thread().name
|
||||
return value * 2
|
||||
|
||||
result = await adapter._run_blocking(_work, 21)
|
||||
assert result == 42
|
||||
# Ran on the adapter-owned pool, not the default executor.
|
||||
assert captured["thread"].startswith("hermes-feishu-sdk")
|
||||
adapter._shutdown_sdk_executor()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user