tahoe-lafs/misc/build_helpers/test-with-fake-dists.py

71 lines
2.7 KiB
Python

#!/usr/bin/env python
import StringIO, glob, os, platform, shutil, subprocess, sys, tarfile, zipfile
import pkg_resources
def test():
# We put a "fakedependency-1.0.0.egg" package and a
# "fakedependency-9.9.99.tar.gz" into a directory, but the latter is
# booby-trapped so it will raise an exception when you try to build it.
#
# Then we run
#
# python setup.py --fakedependency -v test -s buildtest.test_build_with_fake_dist
#
# which imports fakedependency and passes if fakedependency.__version__ == '1.0.0'.
#
# The goal is to turn red if the build system tries to build the
# source dist when it could have used the binary dist.
#
# Note that for this test to make sense, Tahoe-LAFS needs to be asking
# for a version of fakedependency which can be satisfied by 1.0.0.
# The --fakedependency option to setup.py arranges that.
fake_distdir = 'tahoe-deps'
fake_distname = "fakedependency"
fake_sdistversion = "9.9.99"
fake_bdistversion = "1.0.0"
sdist_setup = "raise Exception('Aha I caught you trying to build me. I am a fakedependency 9.9.99 sdist and you should be satisfied with a bdist.')"
testsuite = "buildtest.test_build_with_fake_dist"
dist_dirname = os.path.join(os.getcwd(), fake_distdir)
try:
os.makedirs(dist_dirname)
except OSError:
# probably already exists
pass
bdist_egg_name = os.path.join(dist_dirname, '%s-%s-py%s.%s-%s.egg' % (fake_distname, fake_bdistversion, platform.python_version_tuple()[0], platform.python_version_tuple()[1], pkg_resources.get_supported_platform()))
try:
bdist_egg = zipfile.ZipFile(bdist_egg_name, 'w')
bdist_egg.writestr('fakedependency/__init__.py', '__version__ = "%s"\n' % (fake_bdistversion,))
bdist_egg.close()
sdist_name = os.path.join(dist_dirname, '%s-%s.tar' % (fake_distname, fake_sdistversion))
sdist = tarfile.open(sdist_name, 'w:gz')
sdist.errorlevel = 2
tarinfo = tarfile.TarInfo('setup.py')
tarinfo.errorlevel = 2
tarinfo.size = len(sdist_setup)
sdist.addfile(tarinfo, StringIO.StringIO(sdist_setup))
sdist.close()
sys.exit(subprocess.call([sys.executable, "setup.py", "--fakedependency", "-v", "test", "-s", testsuite],
env=os.environ))
finally:
os.remove(bdist_egg_name)
os.remove(sdist_name)
cleanup()
def cleanup():
shutil.rmtree('build')
shutil.rmtree('support')
shutil.rmtree(os.path.join('src', 'allmydata_tahoe.egg-info'))
os.remove(os.path.join('bin', 'tahoe'))
os.remove(os.path.join('bin', 'tahoe.pyscript'))
if __name__ == '__main__':
test()