mirror of
https://github.com/python/cpython.git
synced 2025-07-08 03:45:36 +00:00
8f8ec92de8
requires them. Disable executable bits and shebang lines in test and benchmark files in order to prevent using a random system python, and in source files of modules which don't provide command line interface. Fixed shebang lines in the unittestgui and checkpip scripts.
315 lines
10 KiB
Python
315 lines
10 KiB
Python
import importlib
|
|
import shutil
|
|
import sys
|
|
import os
|
|
import unittest
|
|
import socket
|
|
import tempfile
|
|
import errno
|
|
from test import support
|
|
|
|
TESTFN = support.TESTFN
|
|
TESTDIRN = os.path.basename(tempfile.mkdtemp(dir='.'))
|
|
|
|
|
|
class TestSupport(unittest.TestCase):
|
|
def setUp(self):
|
|
support.unlink(TESTFN)
|
|
support.rmtree(TESTDIRN)
|
|
tearDown = setUp
|
|
|
|
def test_import_module(self):
|
|
support.import_module("ftplib")
|
|
self.assertRaises(unittest.SkipTest, support.import_module, "foo")
|
|
|
|
def test_import_fresh_module(self):
|
|
support.import_fresh_module("ftplib")
|
|
|
|
def test_get_attribute(self):
|
|
self.assertEqual(support.get_attribute(self, "test_get_attribute"),
|
|
self.test_get_attribute)
|
|
self.assertRaises(unittest.SkipTest, support.get_attribute, self, "foo")
|
|
|
|
@unittest.skip("failing buildbots")
|
|
def test_get_original_stdout(self):
|
|
self.assertEqual(support.get_original_stdout(), sys.stdout)
|
|
|
|
def test_unload(self):
|
|
import sched
|
|
self.assertIn("sched", sys.modules)
|
|
support.unload("sched")
|
|
self.assertNotIn("sched", sys.modules)
|
|
|
|
def test_unlink(self):
|
|
with open(TESTFN, "w") as f:
|
|
pass
|
|
support.unlink(TESTFN)
|
|
self.assertFalse(os.path.exists(TESTFN))
|
|
support.unlink(TESTFN)
|
|
|
|
def test_rmtree(self):
|
|
os.mkdir(TESTDIRN)
|
|
os.mkdir(os.path.join(TESTDIRN, TESTDIRN))
|
|
support.rmtree(TESTDIRN)
|
|
self.assertFalse(os.path.exists(TESTDIRN))
|
|
support.rmtree(TESTDIRN)
|
|
|
|
def test_forget(self):
|
|
mod_filename = TESTFN + '.py'
|
|
with open(mod_filename, 'w') as f:
|
|
print('foo = 1', file=f)
|
|
sys.path.insert(0, os.curdir)
|
|
importlib.invalidate_caches()
|
|
try:
|
|
mod = __import__(TESTFN)
|
|
self.assertIn(TESTFN, sys.modules)
|
|
|
|
support.forget(TESTFN)
|
|
self.assertNotIn(TESTFN, sys.modules)
|
|
finally:
|
|
del sys.path[0]
|
|
support.unlink(mod_filename)
|
|
|
|
def test_HOST(self):
|
|
s = socket.socket()
|
|
s.bind((support.HOST, 0))
|
|
s.close()
|
|
|
|
def test_find_unused_port(self):
|
|
port = support.find_unused_port()
|
|
s = socket.socket()
|
|
s.bind((support.HOST, port))
|
|
s.close()
|
|
|
|
def test_bind_port(self):
|
|
s = socket.socket()
|
|
support.bind_port(s)
|
|
s.listen(1)
|
|
s.close()
|
|
|
|
# Tests for temp_dir()
|
|
|
|
def test_temp_dir(self):
|
|
"""Test that temp_dir() creates and destroys its directory."""
|
|
parent_dir = tempfile.mkdtemp()
|
|
parent_dir = os.path.realpath(parent_dir)
|
|
|
|
try:
|
|
path = os.path.join(parent_dir, 'temp')
|
|
self.assertFalse(os.path.isdir(path))
|
|
with support.temp_dir(path) as temp_path:
|
|
self.assertEqual(temp_path, path)
|
|
self.assertTrue(os.path.isdir(path))
|
|
self.assertFalse(os.path.isdir(path))
|
|
finally:
|
|
shutil.rmtree(parent_dir)
|
|
|
|
def test_temp_dir__path_none(self):
|
|
"""Test passing no path."""
|
|
with support.temp_dir() as temp_path:
|
|
self.assertTrue(os.path.isdir(temp_path))
|
|
self.assertFalse(os.path.isdir(temp_path))
|
|
|
|
def test_temp_dir__existing_dir__quiet_default(self):
|
|
"""Test passing a directory that already exists."""
|
|
def call_temp_dir(path):
|
|
with support.temp_dir(path) as temp_path:
|
|
raise Exception("should not get here")
|
|
|
|
path = tempfile.mkdtemp()
|
|
path = os.path.realpath(path)
|
|
try:
|
|
self.assertTrue(os.path.isdir(path))
|
|
self.assertRaises(FileExistsError, call_temp_dir, path)
|
|
# Make sure temp_dir did not delete the original directory.
|
|
self.assertTrue(os.path.isdir(path))
|
|
finally:
|
|
shutil.rmtree(path)
|
|
|
|
def test_temp_dir__existing_dir__quiet_true(self):
|
|
"""Test passing a directory that already exists with quiet=True."""
|
|
path = tempfile.mkdtemp()
|
|
path = os.path.realpath(path)
|
|
|
|
try:
|
|
with support.check_warnings() as recorder:
|
|
with support.temp_dir(path, quiet=True) as temp_path:
|
|
self.assertEqual(path, temp_path)
|
|
warnings = [str(w.message) for w in recorder.warnings]
|
|
# Make sure temp_dir did not delete the original directory.
|
|
self.assertTrue(os.path.isdir(path))
|
|
finally:
|
|
shutil.rmtree(path)
|
|
|
|
expected = ['tests may fail, unable to create temp dir: ' + path]
|
|
self.assertEqual(warnings, expected)
|
|
|
|
# Tests for change_cwd()
|
|
|
|
def test_change_cwd(self):
|
|
original_cwd = os.getcwd()
|
|
|
|
with support.temp_dir() as temp_path:
|
|
with support.change_cwd(temp_path) as new_cwd:
|
|
self.assertEqual(new_cwd, temp_path)
|
|
self.assertEqual(os.getcwd(), new_cwd)
|
|
|
|
self.assertEqual(os.getcwd(), original_cwd)
|
|
|
|
def test_change_cwd__non_existent_dir(self):
|
|
"""Test passing a non-existent directory."""
|
|
original_cwd = os.getcwd()
|
|
|
|
def call_change_cwd(path):
|
|
with support.change_cwd(path) as new_cwd:
|
|
raise Exception("should not get here")
|
|
|
|
with support.temp_dir() as parent_dir:
|
|
non_existent_dir = os.path.join(parent_dir, 'does_not_exist')
|
|
self.assertRaises(FileNotFoundError, call_change_cwd,
|
|
non_existent_dir)
|
|
|
|
self.assertEqual(os.getcwd(), original_cwd)
|
|
|
|
def test_change_cwd__non_existent_dir__quiet_true(self):
|
|
"""Test passing a non-existent directory with quiet=True."""
|
|
original_cwd = os.getcwd()
|
|
|
|
with support.temp_dir() as parent_dir:
|
|
bad_dir = os.path.join(parent_dir, 'does_not_exist')
|
|
with support.check_warnings() as recorder:
|
|
with support.change_cwd(bad_dir, quiet=True) as new_cwd:
|
|
self.assertEqual(new_cwd, original_cwd)
|
|
self.assertEqual(os.getcwd(), new_cwd)
|
|
warnings = [str(w.message) for w in recorder.warnings]
|
|
|
|
expected = ['tests may fail, unable to change CWD to: ' + bad_dir]
|
|
self.assertEqual(warnings, expected)
|
|
|
|
# Tests for change_cwd()
|
|
|
|
def test_change_cwd__chdir_warning(self):
|
|
"""Check the warning message when os.chdir() fails."""
|
|
path = TESTFN + '_does_not_exist'
|
|
with support.check_warnings() as recorder:
|
|
with support.change_cwd(path=path, quiet=True):
|
|
pass
|
|
messages = [str(w.message) for w in recorder.warnings]
|
|
self.assertEqual(messages, ['tests may fail, unable to change CWD to: ' + path])
|
|
|
|
# Tests for temp_cwd()
|
|
|
|
def test_temp_cwd(self):
|
|
here = os.getcwd()
|
|
with support.temp_cwd(name=TESTFN):
|
|
self.assertEqual(os.path.basename(os.getcwd()), TESTFN)
|
|
self.assertFalse(os.path.exists(TESTFN))
|
|
self.assertTrue(os.path.basename(os.getcwd()), here)
|
|
|
|
|
|
def test_temp_cwd__name_none(self):
|
|
"""Test passing None to temp_cwd()."""
|
|
original_cwd = os.getcwd()
|
|
with support.temp_cwd(name=None) as new_cwd:
|
|
self.assertNotEqual(new_cwd, original_cwd)
|
|
self.assertTrue(os.path.isdir(new_cwd))
|
|
self.assertEqual(os.getcwd(), new_cwd)
|
|
self.assertEqual(os.getcwd(), original_cwd)
|
|
|
|
def test_sortdict(self):
|
|
self.assertEqual(support.sortdict({3:3, 2:2, 1:1}), "{1: 1, 2: 2, 3: 3}")
|
|
|
|
def test_make_bad_fd(self):
|
|
fd = support.make_bad_fd()
|
|
with self.assertRaises(OSError) as cm:
|
|
os.write(fd, b"foo")
|
|
self.assertEqual(cm.exception.errno, errno.EBADF)
|
|
|
|
def test_check_syntax_error(self):
|
|
support.check_syntax_error(self, "def class")
|
|
self.assertRaises(AssertionError, support.check_syntax_error, self, "1")
|
|
|
|
def test_CleanImport(self):
|
|
import importlib
|
|
with support.CleanImport("asyncore"):
|
|
importlib.import_module("asyncore")
|
|
|
|
def test_DirsOnSysPath(self):
|
|
with support.DirsOnSysPath('foo', 'bar'):
|
|
self.assertIn("foo", sys.path)
|
|
self.assertIn("bar", sys.path)
|
|
self.assertNotIn("foo", sys.path)
|
|
self.assertNotIn("bar", sys.path)
|
|
|
|
def test_captured_stdout(self):
|
|
with support.captured_stdout() as stdout:
|
|
print("hello")
|
|
self.assertEqual(stdout.getvalue(), "hello\n")
|
|
|
|
def test_captured_stderr(self):
|
|
with support.captured_stderr() as stderr:
|
|
print("hello", file=sys.stderr)
|
|
self.assertEqual(stderr.getvalue(), "hello\n")
|
|
|
|
def test_captured_stdin(self):
|
|
with support.captured_stdin() as stdin:
|
|
stdin.write('hello\n')
|
|
stdin.seek(0)
|
|
# call test code that consumes from sys.stdin
|
|
captured = input()
|
|
self.assertEqual(captured, "hello")
|
|
|
|
def test_gc_collect(self):
|
|
support.gc_collect()
|
|
|
|
def test_python_is_optimized(self):
|
|
self.assertIsInstance(support.python_is_optimized(), bool)
|
|
|
|
def test_swap_attr(self):
|
|
class Obj:
|
|
x = 1
|
|
obj = Obj()
|
|
with support.swap_attr(obj, "x", 5):
|
|
self.assertEqual(obj.x, 5)
|
|
self.assertEqual(obj.x, 1)
|
|
|
|
def test_swap_item(self):
|
|
D = {"item":1}
|
|
with support.swap_item(D, "item", 5):
|
|
self.assertEqual(D["item"], 5)
|
|
self.assertEqual(D["item"], 1)
|
|
|
|
# XXX -follows a list of untested API
|
|
# make_legacy_pyc
|
|
# is_resource_enabled
|
|
# requires
|
|
# fcmp
|
|
# umaks
|
|
# findfile
|
|
# check_warnings
|
|
# EnvironmentVarGuard
|
|
# TransientResource
|
|
# transient_internet
|
|
# run_with_locale
|
|
# set_memlimit
|
|
# bigmemtest
|
|
# precisionbigmemtest
|
|
# bigaddrspacetest
|
|
# requires_resource
|
|
# run_doctest
|
|
# threading_cleanup
|
|
# reap_threads
|
|
# reap_children
|
|
# strip_python_stderr
|
|
# args_from_interpreter_flags
|
|
# can_symlink
|
|
# skip_unless_symlink
|
|
# SuppressCrashReport
|
|
|
|
|
|
def test_main():
|
|
tests = [TestSupport]
|
|
support.run_unittest(*tests)
|
|
|
|
if __name__ == '__main__':
|
|
test_main()
|