new unit tests

This commit is contained in:
Corey Goldberg 2013-04-04 09:53:50 -04:00
parent 3e22285f69
commit 347a7f3f2c
1 changed files with 29 additions and 0 deletions

29
xvfbwrapper/test_xvfb.py Normal file
View File

@ -0,0 +1,29 @@
#!/usr/bin/env python
from xvfbwrapper import Xvfb
import os
import unittest
class TestXvfb(unittest.TestCase):
def test_start(self):
xvfb = Xvfb()
self.addCleanup(xvfb.stop)
xvfb.start()
self.assertEqual(':%d' % xvfb.vdisplay_num, os.environ['DISPLAY'])
self.assertIsNot(None, xvfb.proc)
def test_stop(self):
orig = os.environ['DISPLAY']
xvfb = Xvfb()
xvfb.start()
self.assertNotEqual(orig, os.environ['DISPLAY'])
xvfb.stop()
self.assertEquals(orig, os.environ['DISPLAY'])
if __name__ == '__main__':
unittest.main(verbosity=2)