diff --git a/xvfbwrapper/test_xvfb.py b/xvfbwrapper/test_xvfb.py new file mode 100644 index 0000000..eb9c746 --- /dev/null +++ b/xvfbwrapper/test_xvfb.py @@ -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)