221 lines
6.7 KiB
Python
221 lines
6.7 KiB
Python
"""Tests for distutils.command.sdist."""
|
|
import os
|
|
import unittest
|
|
import shutil
|
|
import zipfile
|
|
from os.path import join
|
|
import sys
|
|
|
|
from distutils.command.sdist import sdist
|
|
from distutils.core import Distribution
|
|
from distutils.tests.test_config import PyPIRCCommandTestCase
|
|
from distutils.errors import DistutilsExecError
|
|
from distutils.spawn import find_executable
|
|
|
|
SETUP_PY = """
|
|
from distutils.core import setup
|
|
import somecode
|
|
|
|
setup(name='fake')
|
|
"""
|
|
|
|
MANIFEST_IN = """
|
|
recursive-include somecode *
|
|
"""
|
|
|
|
class sdistTestCase(PyPIRCCommandTestCase):
|
|
|
|
def setUp(self):
|
|
super(sdistTestCase, self).setUp()
|
|
self.old_path = os.getcwd()
|
|
self.temp_pkg = os.path.join(self.mkdtemp(), 'temppkg')
|
|
self.tmp_dir = self.mkdtemp()
|
|
|
|
def tearDown(self):
|
|
os.chdir(self.old_path)
|
|
super(sdistTestCase, self).tearDown()
|
|
|
|
def _init_tmp_pkg(self):
|
|
if os.path.exists(self.temp_pkg):
|
|
shutil.rmtree(self.temp_pkg)
|
|
os.mkdir(self.temp_pkg)
|
|
os.mkdir(join(self.temp_pkg, 'somecode'))
|
|
os.mkdir(join(self.temp_pkg, 'dist'))
|
|
# creating a MANIFEST, a package, and a README
|
|
self._write(join(self.temp_pkg, 'MANIFEST.in'), MANIFEST_IN)
|
|
self._write(join(self.temp_pkg, 'README'), 'xxx')
|
|
self._write(join(self.temp_pkg, 'somecode', '__init__.py'), '#')
|
|
self._write(join(self.temp_pkg, 'setup.py'), SETUP_PY)
|
|
os.chdir(self.temp_pkg)
|
|
|
|
def _write(self, path, content):
|
|
f = open(path, 'w')
|
|
try:
|
|
f.write(content)
|
|
finally:
|
|
f.close()
|
|
|
|
def test_prune_file_list(self):
|
|
# this test creates a package with some vcs dirs in it
|
|
# and launch sdist to make sure they get pruned
|
|
# on all systems
|
|
self._init_tmp_pkg()
|
|
|
|
# creating VCS directories with some files in them
|
|
os.mkdir(join(self.temp_pkg, 'somecode', '.svn'))
|
|
self._write(join(self.temp_pkg, 'somecode', '.svn', 'ok.py'), 'xxx')
|
|
|
|
os.mkdir(join(self.temp_pkg, 'somecode', '.hg'))
|
|
self._write(join(self.temp_pkg, 'somecode', '.hg',
|
|
'ok'), 'xxx')
|
|
|
|
os.mkdir(join(self.temp_pkg, 'somecode', '.git'))
|
|
self._write(join(self.temp_pkg, 'somecode', '.git',
|
|
'ok'), 'xxx')
|
|
|
|
# now building a sdist
|
|
dist = Distribution()
|
|
dist.script_name = 'setup.py'
|
|
dist.metadata.name = 'fake'
|
|
dist.metadata.version = '1.0'
|
|
dist.metadata.url = 'http://xxx'
|
|
dist.metadata.author = dist.metadata.author_email = 'xxx'
|
|
dist.packages = ['somecode']
|
|
dist.include_package_data = True
|
|
cmd = sdist(dist)
|
|
cmd.manifest = 'MANIFEST'
|
|
cmd.template = 'MANIFEST.in'
|
|
cmd.dist_dir = 'dist'
|
|
|
|
# zip is available universally
|
|
# (tar might not be installed under win32)
|
|
cmd.formats = ['zip']
|
|
cmd.run()
|
|
|
|
# now let's check what we have
|
|
dist_folder = join(self.temp_pkg, 'dist')
|
|
files = os.listdir(dist_folder)
|
|
self.assertEquals(files, ['fake-1.0.zip'])
|
|
|
|
zip_file = zipfile.ZipFile(join(dist_folder, 'fake-1.0.zip'))
|
|
try:
|
|
content = zip_file.namelist()
|
|
finally:
|
|
zip_file.close()
|
|
|
|
# making sure everything has been pruned correctly
|
|
self.assertEquals(len(content), 4)
|
|
|
|
def test_make_distribution(self):
|
|
|
|
# check if tar and gzip are installed
|
|
if (find_executable('tar') is None or
|
|
find_executable('gzip') is None):
|
|
return
|
|
|
|
self._init_tmp_pkg()
|
|
|
|
# now building a sdist
|
|
dist = Distribution()
|
|
dist.script_name = 'setup.py'
|
|
dist.metadata.name = 'fake'
|
|
dist.metadata.version = '1.0'
|
|
dist.metadata.url = 'http://xxx'
|
|
dist.metadata.author = dist.metadata.author_email = 'xxx'
|
|
dist.packages = ['somecode']
|
|
dist.include_package_data = True
|
|
cmd = sdist(dist)
|
|
cmd.manifest = 'MANIFEST'
|
|
cmd.template = 'MANIFEST.in'
|
|
cmd.dist_dir = 'dist'
|
|
|
|
# creating a gztar then a tar
|
|
cmd.formats = ['gztar', 'tar']
|
|
cmd.run()
|
|
|
|
# making sure we have two files
|
|
dist_folder = join(self.temp_pkg, 'dist')
|
|
result = os.listdir(dist_folder)
|
|
result.sort()
|
|
self.assertEquals(result,
|
|
['fake-1.0.tar', 'fake-1.0.tar.gz'] )
|
|
|
|
os.remove(join(dist_folder, 'fake-1.0.tar'))
|
|
os.remove(join(dist_folder, 'fake-1.0.tar.gz'))
|
|
|
|
# now trying a tar then a gztar
|
|
cmd.formats = ['tar', 'gztar']
|
|
cmd.run()
|
|
|
|
result = os.listdir(dist_folder)
|
|
result.sort()
|
|
self.assertEquals(result,
|
|
['fake-1.0.tar', 'fake-1.0.tar.gz'])
|
|
|
|
def get_cmd(self, metadata=None):
|
|
"""Returns a cmd"""
|
|
if metadata is None:
|
|
metadata = {'name': 'fake', 'version': '1.0',
|
|
'url': 'xxx', 'author': 'xxx',
|
|
'author_email': 'xxx'}
|
|
dist = Distribution(metadata)
|
|
dist.script_name = 'setup.py'
|
|
dist.packages = ['somecode']
|
|
dist.include_package_data = True
|
|
cmd = sdist(dist)
|
|
cmd.dist_dir = 'dist'
|
|
def _warn(*args):
|
|
pass
|
|
cmd.warn = _warn
|
|
return dist, cmd
|
|
|
|
def test_get_file_list(self):
|
|
# make sure MANIFEST is recalculated
|
|
dist, cmd = self.get_cmd()
|
|
|
|
os.chdir(self.tmp_dir)
|
|
|
|
# filling data_files by pointing files in package_data
|
|
os.mkdir(os.path.join(self.tmp_dir, 'somecode'))
|
|
self.write_file((self.tmp_dir, 'somecode', '__init__.py'), '#')
|
|
self.write_file((self.tmp_dir, 'somecode', 'one.py'), '#')
|
|
cmd.ensure_finalized()
|
|
cmd.run()
|
|
|
|
f = open(cmd.manifest)
|
|
try:
|
|
manifest = [line.strip() for line in f.read().split('\n')
|
|
if line.strip() != '']
|
|
finally:
|
|
f.close()
|
|
|
|
self.assertEquals(len(manifest), 2)
|
|
|
|
# adding a file
|
|
self.write_file((self.tmp_dir, 'somecode', 'two.py'), '#')
|
|
|
|
# make sure build_py is reinitinialized, like a fresh run
|
|
build_py = dist.get_command_obj('build_py')
|
|
build_py.finalized = False
|
|
build_py.ensure_finalized()
|
|
|
|
cmd.run()
|
|
|
|
f = open(cmd.manifest)
|
|
try:
|
|
manifest2 = [line.strip() for line in f.read().split('\n')
|
|
if line.strip() != '']
|
|
finally:
|
|
f.close()
|
|
|
|
# do we have the new file in MANIFEST ?
|
|
self.assertEquals(len(manifest2), 3)
|
|
self.assert_('two.py' in manifest2[-1])
|
|
|
|
|
|
def test_suite():
|
|
return unittest.makeSuite(sdistTestCase)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(defaultTest="test_suite")
|