openstack/deb-python-xvfbwrapper
Code Issues Proposed changes

add more assertions

Browse Source
This commit is contained in:
Stephen J. Fuhry 2016-07-03 11:34:32 -04:00
parent 207a4070e0
commit b4e8aac662
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test_xvfb.py
View File

@ -89,13 +89,19 @@ class TestXvfb(unittest.TestCase):
self.addCleanup(xvfb._cleanup_lock_file)
self.addCleanup(xvfb2._cleanup_lock_file)
self.addCleanup(xvfb3._cleanup_lock_file)
side_effect = [11, 11, 22, 11, 22, 11, 22, 22, 22, 33]
with patch('xvfbwrapper.randint',
side_effect=[11, 11, 22, 11, 22, 11, 22, 22, 22, 33]):
side_effect=side_effect) as mockrandint:
self.assertEqual(xvfb._get_next_unused_display(), 11)
self.assertEqual(mockrandint.call_count, 1)
if sys.version_info >= (3, 2):
with self.assertWarns(ResourceWarning):
self.assertEqual(xvfb2._get_next_unused_display(), 22)
self.assertEqual(mockrandint.call_count, 3)
self.assertEqual(xvfb3._get_next_unused_display(), 33)
self.assertEqual(mockrandint.call_count, 10)
else:
self.assertEqual(xvfb2._get_next_unused_display(), 22)
self.assertEqual(mockrandint.call_count, 3)
self.assertEqual(xvfb3._get_next_unused_display(), 33)
self.assertEqual(mockrandint.call_count, 10)