Make Xvfb a contextmanager

This commit is contained in:
Moritz Schlarb 2013-09-26 14:24:46 +02:00
parent f5b14ccdb2
commit dc8d6e8e8a
1 changed files with 7 additions and 0 deletions

View File

@ -37,6 +37,13 @@ class Xvfb:
else: else:
self.old_display_num = 0 self.old_display_num = 0
def __enter__(self):
self.start()
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self.stop()
def start(self): def start(self):
self.vdisplay_num = self.search_for_free_display() self.vdisplay_num = self.search_for_free_display()
self.xvfb_cmd = ['Xvfb', ':%d' % self.vdisplay_num] + self.xvfb_cmd self.xvfb_cmd = ['Xvfb', ':%d' % self.vdisplay_num] + self.xvfb_cmd