mirror of
https://github.com/python/cpython.git
synced 2025-11-08 05:39:34 +00:00
bf816223df
We don't need to create a temporary buffered binary or text file object just to create an empty file. Replace also os.fdopen(handle).close() by os.close(handle).
122 lines
3.9 KiB
Python
122 lines
3.9 KiB
Python
"""Tests for distutils.command.build_py."""
|
|
|
|
import os
|
|
import sys
|
|
import io
|
|
import unittest
|
|
|
|
from distutils.command.build_py import build_py
|
|
from distutils.core import Distribution
|
|
from distutils.errors import DistutilsFileError
|
|
|
|
from distutils.tests import support
|
|
from test.support import run_unittest, create_empty_file
|
|
|
|
|
|
class BuildPyTestCase(support.TempdirManager,
|
|
support.LoggingSilencer,
|
|
unittest.TestCase):
|
|
|
|
def test_package_data(self):
|
|
sources = self.mkdtemp()
|
|
f = open(os.path.join(sources, "__init__.py"), "w")
|
|
try:
|
|
f.write("# Pretend this is a package.")
|
|
finally:
|
|
f.close()
|
|
f = open(os.path.join(sources, "README.txt"), "w")
|
|
try:
|
|
f.write("Info about this package")
|
|
finally:
|
|
f.close()
|
|
|
|
destination = self.mkdtemp()
|
|
|
|
dist = Distribution({"packages": ["pkg"],
|
|
"package_dir": {"pkg": sources}})
|
|
# script_name need not exist, it just need to be initialized
|
|
dist.script_name = os.path.join(sources, "setup.py")
|
|
dist.command_obj["build"] = support.DummyCommand(
|
|
force=0,
|
|
build_lib=destination)
|
|
dist.packages = ["pkg"]
|
|
dist.package_data = {"pkg": ["README.txt"]}
|
|
dist.package_dir = {"pkg": sources}
|
|
|
|
cmd = build_py(dist)
|
|
cmd.compile = 1
|
|
cmd.ensure_finalized()
|
|
self.assertEqual(cmd.package_data, dist.package_data)
|
|
|
|
cmd.run()
|
|
|
|
# This makes sure the list of outputs includes byte-compiled
|
|
# files for Python modules but not for package data files
|
|
# (there shouldn't *be* byte-code files for those!).
|
|
#
|
|
self.assertEqual(len(cmd.get_outputs()), 3)
|
|
pkgdest = os.path.join(destination, "pkg")
|
|
files = os.listdir(pkgdest)
|
|
self.assertIn("__init__.py", files)
|
|
self.assertIn("README.txt", files)
|
|
# XXX even with -O, distutils writes pyc, not pyo; bug?
|
|
if sys.dont_write_bytecode:
|
|
self.assertNotIn("__init__.pyc", files)
|
|
else:
|
|
self.assertIn("__init__.pyc", files)
|
|
|
|
def test_empty_package_dir(self):
|
|
# See SF 1668596/1720897.
|
|
cwd = os.getcwd()
|
|
|
|
# create the distribution files.
|
|
sources = self.mkdtemp()
|
|
create_empty_file(os.path.join(sources, "__init__.py"))
|
|
|
|
testdir = os.path.join(sources, "doc")
|
|
os.mkdir(testdir)
|
|
create_empty_file(os.path.join(testdir, "testfile"))
|
|
|
|
os.chdir(sources)
|
|
old_stdout = sys.stdout
|
|
sys.stdout = io.StringIO()
|
|
|
|
try:
|
|
dist = Distribution({"packages": ["pkg"],
|
|
"package_dir": {"pkg": ""},
|
|
"package_data": {"pkg": ["doc/*"]}})
|
|
# script_name need not exist, it just need to be initialized
|
|
dist.script_name = os.path.join(sources, "setup.py")
|
|
dist.script_args = ["build"]
|
|
dist.parse_command_line()
|
|
|
|
try:
|
|
dist.run_commands()
|
|
except DistutilsFileError:
|
|
self.fail("failed package_data test when package_dir is ''")
|
|
finally:
|
|
# Restore state.
|
|
os.chdir(cwd)
|
|
sys.stdout = old_stdout
|
|
|
|
def test_dont_write_bytecode(self):
|
|
# makes sure byte_compile is not used
|
|
pkg_dir, dist = self.create_dist()
|
|
cmd = build_py(dist)
|
|
cmd.compile = 1
|
|
cmd.optimize = 1
|
|
|
|
old_dont_write_bytecode = sys.dont_write_bytecode
|
|
sys.dont_write_bytecode = True
|
|
try:
|
|
cmd.byte_compile([])
|
|
finally:
|
|
sys.dont_write_bytecode = old_dont_write_bytecode
|
|
|
|
self.assertIn('byte-compiling is disabled', self.logs[0][1])
|
|
|
|
def test_suite():
|
|
return unittest.makeSuite(BuildPyTestCase)
|
|
|
|
if __name__ == "__main__":
|
|
run_unittest(test_suite())
|