From 347a7f3f2c0296c9a0ddf70d4b9c9f93ec489105 Mon Sep 17 00:00:00 2001
From: Corey Goldberg <cgoldberg@gmail.com>
Date: Thu, 4 Apr 2013 09:53:50 -0400
Subject: [PATCH] new unit tests

---
 xvfbwrapper/test_xvfb.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 xvfbwrapper/test_xvfb.py

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)