Xvfb에서 Selenium을 실행하는 방법: '디스플레이를 열 수 없음' 오류 극복

Xvfb(X Virtual Framebuffer)는 다음과 같은 가상 디스플레이를 제공합니다. GUI가 없는 EC2 인스턴스와 같은 헤드리스 환경에서 그래픽 애플리케이션을 실행할 수 있습니다. 이는 그래픽 사용자 인터페이스 없이 서버에서 Selenium 테스트를 실행하는 데 필수적입니다.

그러나 Selenium 및 Xvfb를 설치한 후에도 테스트를 실행하려고 하면 "오류: 디스플레이를 열 수 없습니다: :0"이 발생할 수 있습니다. 이는 웹 페이지와 상호 작용하기 위해 Selenium이 그래픽 디스플레이에 의존하기 때문에 발생합니다.

해결책: PyVirtualDisplay 또는 Xvfbwrapper 활용

이 문제를 해결하려면 PyVirtualDisplay 또는 Xvfbwrapper를 사용할 수 있습니다. , Xvfb를 Python 모듈로 캡슐화합니다. 이러한 모듈은 가상 디스플레이를 시뮬레이션하여 Selenium이 헤드리스 환경에서 실행되도록 허용합니다.

옵션 1: PyVirtualDisplay

from pyvirtualdisplay import Display
from selenium import webdriver

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

browser = webdriver.Firefox()
browser.get('http://www.google.com')
print(browser.title)
browser.quit()

display.stop()

옵션 2: Xvfbwrapper

from xvfbwrapper import Xvfb

vdisplay = Xvfb()
vdisplay.start()

browser = webdriver.Firefox()
browser.get('http://www.google.com')
print(browser.title)
browser.quit()

vdisplay.stop()

컨텍스트 관리자를 활용하여 프로세스를 단순화할 수도 있습니다.

from xvfbwrapper import Xvfb

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

이러한 모듈을 활용하면 그래픽 사용자 인터페이스가 없는 헤드리스 서버에서 Selenium 테스트를 효과적으로 실행할 수 있어 어떤 환경에서도 테스트가 성공적으로 실행될 수 있습니다.