如何利用pyvirtualdisplay在Selenium测试后自动关闭Xvfb进程？

我来帮你解决这个Xvfb进程残留的问题——这确实是pyvirtualdisplay + Selenium组合里挺常见的坑，尤其是当测试意外中断的时候。下面几种方法都能帮你实现测试完成后自动清理Xvfb实例：

方法1：使用Python上下文管理器（推荐） #

最省心的方式就是用Python的with语句（上下文管理器），它会自动处理资源的创建和销毁，不管测试是正常跑完还是中途抛出异常，都能保证Xvfb被关闭。

from pyvirtualdisplay import Display
from selenium import webdriver

# 用with语句包裹Display实例，自动管理生命周期
with Display(visible=0, size=(1920, 1080)) as display:
    driver = webdriver.Firefox()
    try:
        # 这里写你的测试逻辑
        driver.get("https://example.com")
        print(f"页面标题：{driver.title}")
    finally:
        # 务必先关闭Selenium Driver，避免资源泄漏
        driver.quit()

当with块执行结束时，会自动调用display.stop()，彻底关闭Xvfb实例，不会留下残留进程。

方法2：显式调用stop()（兼容复杂测试结构） #

如果你的测试结构没法用with语句（比如Display实例需要在多个函数间传递），可以用try-finally块确保stop()一定会被执行：

from pyvirtualdisplay import Display
from selenium import webdriver

# 启动Xvfb
display = Display(visible=0, size=(1920, 1080))
display.start()

driver = webdriver.Firefox()
try:
    # 测试逻辑
    driver.get("https://example.com")
    # 模拟测试操作
    assert driver.title == "Example Domain"
finally:
    # 先关闭Driver，再关闭Xvfb
    driver.quit()
    display.stop()

finally块里的代码不管测试是否抛出异常都会执行，能最大程度避免进程残留。

方法3：用测试框架夹具自动化清理（比如Pytest） #

如果用Pytest做测试框架，可以写一个自定义夹具，让框架帮你自动管理Xvfb的启动和关闭：

# 在conftest.py中定义夹具
import pytest
from pyvirtualdisplay import Display
from selenium import webdriver

@pytest.fixture(scope="function")
def headless_display():
    # 测试前启动Xvfb
    display = Display(visible=0, size=(1920, 1080))
    display.start()
    yield display
    # 测试结束后自动关闭
    display.stop()

@pytest.fixture(scope="function")
def firefox_driver(headless_display):
    driver = webdriver.Firefox()
    yield driver
    driver.quit()

# 测试用例文件
def test_page_title(firefox_driver):
    firefox_driver.get("https://example.com")
    assert firefox_driver.title == "Example Domain"

这样每个测试用例运行前会自动启动Xvfb，测试结束后自动关闭，完全不用手动处理。

方法4：处理强制中断场景（比如Ctrl+C） #

如果测试被手动终止（比如按Ctrl+C），上面的方法可能没机会执行清理，这时候可以注册信号处理器，捕获中断信号后再清理：

import signal
from pyvirtualdisplay import Display
from selenium import webdriver

display = Display(visible=0, size=(1920, 1080))
display.start()
driver = webdriver.Firefox()

def handle_interrupt(signum, frame):
    print("收到中断信号，正在清理资源...")
    driver.quit()
    display.stop()
    exit(0)

# 注册Ctrl+C（SIGINT）和进程终止（SIGTERM）信号处理器
signal.signal(signal.SIGINT, handle_interrupt)
signal.signal(signal.SIGTERM, handle_interrupt)

try:
    # 测试逻辑
    driver.get("https://example.com")
    # 模拟长时间测试
    import time
    time.sleep(30)
finally:
    driver.quit()
    display.stop()

额外注意事项 #

• 一定要先关闭Selenium Driver，再关闭Xvfb：Driver依赖Xvfb的显示环境，反过来关闭可能导致资源泄漏。
• 升级pyvirtualdisplay到最新版本：旧版本可能存在一些资源泄漏的bug，执行pip install --upgrade pyvirtualdisplay即可。
• 如果仍有残留，可以用ps aux | grep Xvfb排查进程，再用kill -9 <进程ID>手动终止，但自动清理才是长久之计。

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