blob: b650bf1c3b598c6e9a83b3cd9b2dfc852ad84b8e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
import os
import subprocess
from devbot import utils
_display_provider = None
def set_display_provider(provider):
global _display_provider
_display_provider = provider
def start():
xvfb_display = _display_provider.find_free_display()
xvfb_proc = subprocess.Popen(args=["Xvfb", xvfb_display],
stdout=utils.devnull,
stderr=subprocess.STDOUT)
orig_display = os.environ.get("DISPLAY", None)
os.environ["DISPLAY"] = xvfb_display
return (xvfb_proc, orig_display)
def stop(xvfb_proc, orig_display):
if orig_display:
os.environ["DISPLAY"] = orig_display
else:
del os.environ["DISPLAY"]
xvfb_proc.terminate()
|