mirror of
https://github.com/python/cpython.git
synced 2024-12-04 07:15:09 +08:00
a21e4ca3c6
........ r70734 | r.david.murray | 2009-03-30 15:04:00 -0400 (Mon, 30 Mar 2009) | 7 lines Add import_function method to test.test_support, and modify a number of tests that expect to be skipped if imports fail or functions don't exist to use import_function and import_module. The ultimate goal is to change regrtest to not skip automatically on ImportError. Checking in now to make sure the buldbots don't show any errors on platforms I can't direct test on. ........ r70775 | r.david.murray | 2009-03-30 19:05:48 -0400 (Mon, 30 Mar 2009) | 4 lines Change more tests to use import_module for the modules that should cause tests to be skipped. Also rename import_function to the more descriptive get_attribute and add a docstring. ........ r70856 | r.david.murray | 2009-03-31 14:32:17 -0400 (Tue, 31 Mar 2009) | 7 lines A few more test skips via import_module, and change import_module to return the error message produced by importlib, so that if an import in the package whose import is being wrapped is what failed the skip message will contain the name of that module instead of the name of the wrapped module. Also fixed formatting of some previous comments. ........ r70874 | r.david.murray | 2009-03-31 15:33:15 -0400 (Tue, 31 Mar 2009) | 5 lines Improve test_support.import_module docstring, remove deprecated flag from get_attribute since it isn't likely to do anything useful. ........ r70876 | r.david.murray | 2009-03-31 15:49:15 -0400 (Tue, 31 Mar 2009) | 4 lines Remove the regrtest check that turns any ImportError into a skipped test. Hopefully all modules whose imports legitimately result in a skipped test have been properly wrapped by the previous commits. ........ r70877 | r.david.murray | 2009-03-31 15:57:24 -0400 (Tue, 31 Mar 2009) | 2 lines Add NEWS entry for regrtest change. ........
111 lines
4.3 KiB
Python
111 lines
4.3 KiB
Python
import unittest
|
|
from test import support
|
|
import time
|
|
|
|
resource = support.import_module('resource')
|
|
|
|
# This test is checking a few specific problem spots with the resource module.
|
|
|
|
class ResourceTest(unittest.TestCase):
|
|
|
|
def test_args(self):
|
|
self.assertRaises(TypeError, resource.getrlimit)
|
|
self.assertRaises(TypeError, resource.getrlimit, 42, 42)
|
|
self.assertRaises(TypeError, resource.setrlimit)
|
|
self.assertRaises(TypeError, resource.setrlimit, 42, 42, 42)
|
|
|
|
def test_fsize_ismax(self):
|
|
try:
|
|
(cur, max) = resource.getrlimit(resource.RLIMIT_FSIZE)
|
|
except AttributeError:
|
|
pass
|
|
else:
|
|
# RLIMIT_FSIZE should be RLIM_INFINITY, which will be a really big
|
|
# number on a platform with large file support. On these platforms,
|
|
# we need to test that the get/setrlimit functions properly convert
|
|
# the number to a C long long and that the conversion doesn't raise
|
|
# an error.
|
|
self.assertEqual(resource.RLIM_INFINITY, max)
|
|
resource.setrlimit(resource.RLIMIT_FSIZE, (cur, max))
|
|
|
|
def test_fsize_enforced(self):
|
|
try:
|
|
(cur, max) = resource.getrlimit(resource.RLIMIT_FSIZE)
|
|
except AttributeError:
|
|
pass
|
|
else:
|
|
# Check to see what happens when the RLIMIT_FSIZE is small. Some
|
|
# versions of Python were terminated by an uncaught SIGXFSZ, but
|
|
# pythonrun.c has been fixed to ignore that exception. If so, the
|
|
# write() should return EFBIG when the limit is exceeded.
|
|
|
|
# At least one platform has an unlimited RLIMIT_FSIZE and attempts
|
|
# to change it raise ValueError instead.
|
|
try:
|
|
try:
|
|
resource.setrlimit(resource.RLIMIT_FSIZE, (1024, max))
|
|
limit_set = True
|
|
except ValueError:
|
|
limit_set = False
|
|
f = open(support.TESTFN, "wb")
|
|
try:
|
|
f.write(b"X" * 1024)
|
|
try:
|
|
f.write(b"Y")
|
|
f.flush()
|
|
# On some systems (e.g., Ubuntu on hppa) the flush()
|
|
# doesn't always cause the exception, but the close()
|
|
# does eventually. Try flushing several times in
|
|
# an attempt to ensure the file is really synced and
|
|
# the exception raised.
|
|
for i in range(5):
|
|
time.sleep(.1)
|
|
f.flush()
|
|
except IOError:
|
|
if not limit_set:
|
|
raise
|
|
if limit_set:
|
|
# Close will attempt to flush the byte we wrote
|
|
# Restore limit first to avoid getting a spurious error
|
|
resource.setrlimit(resource.RLIMIT_FSIZE, (cur, max))
|
|
finally:
|
|
f.close()
|
|
finally:
|
|
if limit_set:
|
|
resource.setrlimit(resource.RLIMIT_FSIZE, (cur, max))
|
|
support.unlink(support.TESTFN)
|
|
|
|
def test_fsize_toobig(self):
|
|
# Be sure that setrlimit is checking for really large values
|
|
too_big = 10**50
|
|
try:
|
|
(cur, max) = resource.getrlimit(resource.RLIMIT_FSIZE)
|
|
except AttributeError:
|
|
pass
|
|
else:
|
|
try:
|
|
resource.setrlimit(resource.RLIMIT_FSIZE, (too_big, max))
|
|
except (OverflowError, ValueError):
|
|
pass
|
|
try:
|
|
resource.setrlimit(resource.RLIMIT_FSIZE, (max, too_big))
|
|
except (OverflowError, ValueError):
|
|
pass
|
|
|
|
def test_getrusage(self):
|
|
self.assertRaises(TypeError, resource.getrusage)
|
|
self.assertRaises(TypeError, resource.getrusage, 42, 42)
|
|
usageself = resource.getrusage(resource.RUSAGE_SELF)
|
|
usagechildren = resource.getrusage(resource.RUSAGE_CHILDREN)
|
|
# May not be available on all systems.
|
|
try:
|
|
usageboth = resource.getrusage(resource.RUSAGE_BOTH)
|
|
except (ValueError, AttributeError):
|
|
pass
|
|
|
|
def test_main(verbose=None):
|
|
support.run_unittest(ResourceTest)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|