diff options
Diffstat (limited to 'devbot/xvfb.py')
-rw-r--r-- | devbot/xvfb.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/devbot/xvfb.py b/devbot/xvfb.py new file mode 100644 index 0000000..6549b9d --- /dev/null +++ b/devbot/xvfb.py @@ -0,0 +1,23 @@ +import os +import subprocess + +from devbot import utils + +xvfb_display = ":100" + +def start(): + 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"] = xvfb_display + else: + del os.environ["DISPLAY"] + + xvfb_proc.terminate() |