fix: 插件测试时设置为 10 分钟超时

nonebot · Apr 10, 2024 · f3896dd · f3896dd
1 parent daf3d1e
commit f3896dd
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/src/utils/plugin_test.py b/src/utils/plugin_test.py
@@ -10,6 +10,7 @@
 """
 # ruff: noqa: T201, ASYNC101
 
+import asyncio
 import json
 import os
 import re
@@ -245,15 +246,21 @@ async def run_poetry_project(self) -> None:
                     )
                 )
 
-            proc = await create_subprocess_shell(
-                "poetry run python runner.py",
-                stdout=subprocess.PIPE,
-                stderr=subprocess.PIPE,
-                cwd=self.path,
-                env=self.get_env(),
-            )
-            stdout, stderr = await proc.communicate()
-            code = proc.returncode
+            try:
+                proc = await create_subprocess_shell(
+                    "poetry run python runner.py",
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.PIPE,
+                    cwd=self.path,
+                    env=self.get_env(),
+                )
+                stdout, stderr = await asyncio.wait_for(proc.communicate(), timeout=600)
+                code = proc.returncode
+            except asyncio.TimeoutError:
+                proc.terminate()
+                stdout = b""
+                stderr = "测试超时".encode()
+                code = 1
 
             self._run = not code