Python服务信号处理器单元测试：无法触发信号与验证清理逻辑

我来帮你搞定这个测试难题！带无限循环和信号触发清理的Python服务，确实没法直接在unittest的主进程里发信号测试（因为测试进程自己会被信号打断），不过有几个实用的方法能完美解决你的需求，我给你拆解一下：

方法1：直接模拟SystemExit异常测试清理逻辑 #

既然信号最终是通过sys.exit(0)抛出SystemExit异常来触发清理，那我们完全可以跳过信号环节，直接模拟这个异常。这种方法适合单元测试，不用处理跨进程的复杂逻辑。

首先假设你的service.py结构是这样的（把清理逻辑抽成单独函数会更易测试）：

# service.py
import signal
import sys
import time

cleanup_done = False

def cleanup():
    global cleanup_done
    cleanup_done = True
    print("清理操作已执行")

def signal_handler(signum, frame):
    sys.exit(0)

def main():
    # 绑定信号处理函数
    signal.signal(signal.SIGINT, signal_handler)
    signal.signal(signal.SIGTERM, signal_handler)
    
    try:
        # 无限循环业务逻辑
        while True:
            print("服务运行中...")
            time.sleep(1)
    except SystemExit:
        # 捕获退出异常，执行清理
        cleanup()
        raise  # 保持原有的退出行为

对应的测试用例可以这样写：

# test_service.py
import unittest
from unittest.mock import patch
from service import main, cleanup_done
import sys

class TestServiceCleanup(unittest.TestCase):
    def setUp(self):
        # 每个测试用例前重置清理状态，避免用例间干扰
        global cleanup_done
        cleanup_done = False

    def test_cleanup_triggered_on_system_exit(self):
        # 替换time.sleep为立刻抛出SystemExit的函数，让循环提前退出
        def mock_sleep(*args):
            raise SystemExit(0)
        
        with patch('time.sleep', side_effect=mock_sleep):
            try:
                main()
            except SystemExit:
                # 捕获异常，不影响测试进程
                pass
        
        # 验证清理操作是否执行
        self.assertTrue(cleanup_done)

方法2：用子进程+真实信号测试完整流程 #

如果你想测试真实的信号触发流程（比如信号是否正确绑定、收到信号后是否正常退出并清理），可以用subprocess启动服务子进程，然后给它发送信号，最后验证结果。

测试用例示例：

# test_service.py
import unittest
import subprocess
import signal
import time

class TestServiceSignalHandling(unittest.TestCase):
    def test_signal_triggers_cleanup_and_exit(self):
        # 启动服务子进程
        proc = subprocess.Popen(
            ['python', 'service.py'],
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True
        )
        
        # 等待服务初始化完成
        time.sleep(0.5)
        
        # 发送SIGINT信号（模拟Ctrl+C）
        proc.send_signal(signal.SIGINT)
        
        # 等待进程退出，超时时间避免测试卡住
        return_code = proc.wait(timeout=5)
        
        # 验证进程是否正常退出（sys.exit(0)返回码为0）
        self.assertEqual(return_code, 0)
        
        # 验证清理操作的痕迹（比如打印日志）
        stdout, stderr = proc.communicate()
        self.assertIn("清理操作已执行", stdout + stderr)

方法3：Mock信号处理和退出逻辑做单元测试 #

如果不想改动原代码结构，也可以用unittest.mock来单独测试信号处理函数和清理逻辑的关联：

# test_service.py
import unittest
from unittest.mock import patch, call
import signal
from service import signal_handler, cleanup_done, cleanup

class TestSignalAndCleanup(unittest.TestCase):
    def setUp(self):
        global cleanup_done
        cleanup_done = False

    def test_signal_handler_calls_sys_exit(self):
        # 验证信号处理函数是否正确调用sys.exit(0)
        with patch('sys.exit') as mock_exit:
            signal_handler(signal.SIGTERM, None)
            mock_exit.assert_called_once_with(0)

    def test_system_exit_triggers_cleanup(self):
        # 验证捕获SystemExit时会执行清理
        try:
            raise SystemExit(0)
        except SystemExit:
            cleanup()
        
        self.assertTrue(cleanup_done)

几个关键注意事项 #

• 如果你的清理逻辑依赖全局变量或外部资源，一定要在每个测试用例前重置状态，避免用例之间互相干扰。
• 使用子进程测试时，记得设置超时时间，防止服务挂起导致测试卡住。
• 尽可能把清理逻辑抽成单独函数，这样不管是单元测试还是集成测试，都会更灵活。

内容的提问来源于stack exchange，提问作者Michal Charemza