new unit tests
parent
3e22285f69
commit
347a7f3f2c
|
@ -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)
|
Loading…
Reference in New Issue