如何检测批量测试中Python方法调用的input()阻塞并终止该调用以继续程序执行

我之前做类似的动态测试套件执行框架时，也碰到过这个头疼的问题——总有测试代码不小心留了input()导致整个流程卡住。分享两个实用的解决方案，分两步解决你的问题：

一、检测是否因input()导致程序挂起 #

这里有两种思路，一种是提前预防式检测（从根源避免阻塞），另一种是事后监控式检测（针对已经运行的代码判断状态）：

1. 提前预防：重定向标准输入让input()直接报错 #

input()本质是读取标准输入（sys.stdin），如果我们把标准输入重定向到一个空的输入流，那么input()调用时会立刻触发EOFError，既不会阻塞，还能直接检测到代码里有input()存在。

示例代码：

import sys
from io import BytesIO

def execute_test_code(code_str):
    original_stdin = sys.stdin  # 保存原始标准输入
    has_input_call = False
    try:
        # 把标准输入替换为空的BytesIO，读取时直接返回EOF
        sys.stdin = BytesIO()
        # 动态执行测试代码
        exec(code_str)
    except EOFError:
        has_input_call = True
        print("检测到测试代码中包含input()调用，会导致程序阻塞")
    finally:
        sys.stdin = original_stdin  # 恢复原始输入流
    return has_input_call

这种方法的好处是不需要额外的进程/线程开销，而且能在代码执行初期就发现问题，避免流程卡住。

2. 事后监控：用多进程+超时判断阻塞状态 #

如果必须允许代码正常读取输入（部分合法场景），但要排查错误代码的话，可以把测试代码放到子进程中运行，主进程设置超时时间，超过时间子进程还没结束，就判定为被input()阻塞了。

示例代码：

import multiprocessing

def run_test_func(test_func):
    # 这里执行动态定义的测试方法
    test_func()

def check_if_blocked(test_func, timeout=3):
    test_process = multiprocessing.Process(target=run_test_func, args=(test_func,))
    test_process.start()
    # 等待timeout秒，如果进程还活着，说明大概率被阻塞了
    test_process.join(timeout)
    
    if test_process.is_alive():
        print("测试方法疑似因input()调用导致挂起")
        return True
    return False

二、终止阻塞的方法调用并恢复程序运行 #

因为Python的线程受GIL限制，无法强制终止一个阻塞的线程（强行终止可能导致资源泄漏），所以最安全的方式是用多进程隔离测试代码，一旦检测到阻塞，直接终止子进程即可：

结合上面的监控代码，完善终止逻辑：

def run_test_with_timeout(test_func, timeout=3):
    test_process = multiprocessing.Process(target=run_test_func, args=(test_func,))
    test_process.start()
    test_process.join(timeout)
    
    if test_process.is_alive():
        print("终止阻塞的测试用例...")
        test_process.terminate()  # 强制终止子进程
        test_process.join()  # 等待进程彻底退出
        # 这里可以添加标记逻辑，比如记录该测试用例为"阻塞失败"
        return False
    # 测试用例正常执行完成
    return True

补充说明 #

• 为什么不用线程？因为线程共享进程资源，强制终止线程可能导致锁未释放、文件句柄泄漏等问题，而进程是独立的，terminate()可以安全地结束它。
• 如果你的测试代码需要读取合法输入，可以在正常场景下不重定向stdin，只在批量测试时开启重定向或者超时监控。

内容的提问来源于stack exchange，提问作者Bai Yang