Question

我通过pyvirtualdisplay在Firefox中无头地运行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')

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

1 个答案: