Make resource tests more robust.

This commit is contained in:
Christian Heimes 2013-10-22 11:45:30 +02:00
parent b7bd5df809
commit d885aa4d66

View File

@ -1,4 +1,5 @@
import sys import sys
import os
import unittest import unittest
from test import support from test import support
import time import time
@ -142,13 +143,15 @@ class ResourceTest(unittest.TestCase):
@unittest.skipUnless(hasattr(resource, 'prlimit'), 'no prlimit') @unittest.skipUnless(hasattr(resource, 'prlimit'), 'no prlimit')
def test_prlimit(self): def test_prlimit(self):
self.assertRaises(TypeError, resource.prlimit) self.assertRaises(TypeError, resource.prlimit)
if os.geteuid() != 0:
self.assertRaises(PermissionError, resource.prlimit, self.assertRaises(PermissionError, resource.prlimit,
1, resource.RLIMIT_AS) 1, resource.RLIMIT_AS)
self.assertRaises(ProcessLookupError, resource.prlimit, self.assertRaises(ProcessLookupError, resource.prlimit,
-1, resource.RLIMIT_AS) -1, resource.RLIMIT_AS)
self.assertEqual(resource.prlimit(0, resource.RLIMIT_AS), (-1, -1)) limit = resource.getrlimit(resource.RLIMIT_AS)
self.assertEqual(resource.prlimit(0, resource.RLIMIT_AS, (-1, -1)), self.assertEqual(resource.prlimit(0, resource.RLIMIT_AS), limit)
(-1, -1)) self.assertEqual(resource.prlimit(0, resource.RLIMIT_AS, limit),
limit)
def test_main(verbose=None): def test_main(verbose=None):