xvfbwrapper 0.2.9

基本用法

from xvfbwrapper import Xvfb

vdisplay = Xvfb()
vdisplay.start()

# launch stuff inside
# virtual display here.

vdisplay.stop()

基本用法，指定显示几何形状

from xvfbwrapper import Xvfb

vdisplay = Xvfb(width=1280, height=740, colordepth=16)
vdisplay.start()

# launch stuff inside
# virtual display here.

vdisplay.stop()

作为上下文管理器的用法

from xvfbwrapper import Xvfb

with Xvfb() as xvfb:
    # launch stuff inside virtual display here.
    # It starts/stops around this code block.

测试示例：无头Selenium WebDriver测试

import unittest

from selenium import webdriver
from xvfbwrapper import Xvfb


class TestPages(unittest.TestCase):

    def setUp(self):
        self.xvfb = Xvfb(width=1280, height=720)
        self.addCleanup(self.xvfb.stop)
        self.xvfb.start()
        self.browser = webdriver.Firefox()
        self.addCleanup(self.browser.quit)

    def testUbuntuHomepage(self):
        self.browser.get('http://www.ubuntu.com')
        self.assertIn('Ubuntu', self.browser.title)

    def testGoogleHomepage(self):
        self.browser.get('http://www.google.com')
        self.assertIn('Google', self.browser.title)


if __name__ == '__main__':
    unittest.main()

上面的测试类使用 selenium 和 xvfbwrapper 在无头显示中运行每个测试案例，使用 Firefox。

• 启动虚拟显示

• 在虚拟显示中启动 Firefox（无头模式）

• 测试运行时浏览器不显示

• 在每个测试案例中断言条件

• 清理时浏览器退出

• 清理时停止虚拟显示

看吧，没有浏览器！

（您还可以在虚拟显示中进行截图以诊断测试失败）