使用pyvirtualdisplay在Selenium测试后自动关闭xvfb

Question

我在火狐中通过pyvirtualdisplay运行Selenium测试，之前遇到了一个意外错误(connection refused)，而且响应时间非常长，在75到80秒之间。经过一些实验后，我发现我之前的功能测试留下了三个Xvfb实例运行，在手动杀死这些实例后，我的下一个测试在5.7秒内达到了预期的失败。所以，问题解决了，但是ps透露，这个测试也留下了一个Xvfb进程挂起。现在，根据文档，stop()方法似乎是最好的候选者，但是将它包含在我的代码中似乎没有任何效果。下面是有问题的测试：

from selenium import webdriver
from pyvirtualdisplay import Display
import unittest

display = Display(visible=0, size=(800,600))
display.start()

class NewVisitorTest(unittest.TestCase):
    def setUp(self):
        self.browser = webdriver.Firefox()

    def tearDown(self):
        self.browser.quit()

    def test_start_retrievable_list(self):
        self.browser.get('http://localhost:8000')    
        self.assertIn('To-Do', self.browser.title)
        self.fail('Finish the test!')

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

display.stop()

如何修改脚本以关闭Xvfb实例，而不必获取PID并手动关闭它？

Answer 1

只需将显示的开始和停止移动到您的测试用例函数：

from selenium import webdriver
from pyvirtualdisplay import Display
import unittest


class NewVisitorTest(unittest.TestCase):
    def setUp(self):
        self.display = Display(visible=0, size=(800,600))
        self.display.start()
        self.browser = webdriver.Firefox()

    def tearDown(self):
        self.browser.quit()
        self.display.stop()

    def test_start_retrievable_list(self):
        self.browser.get('http://localhost:8000')    
        self.assertIn('To-Do', self.browser.title)
        self.fail('Finish the test!')

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