misc/build_helpers/test-dont-use-too-old-dep.py

   1 #!/usr/bin/env python
   2
   3 import StringIO, os, platform, shutil, subprocess, sys, tarfile, zipfile
   4 import pkg_resources
   5
   6 def test():
   7     # We put a "fakedependency-0.9.9.egg" package and a
   8     # "fakedependency-1.0.0.tar.gz" into a directory, but the former is
   9     # booby-trapped so it will raise an exception when you try to import it.
  10     #
  11     # Then we run
  12     #
  13     #   python setup.py --fakedependency -v test -s buildtest.test_build_with_fake_dist
  14     #
  15     # which requires "fakedependency >= 1.0.0", imports fakedependency
  16     # and passes if fakedependency.__version__ == '1.0.0'.
  17     #
  18     # The goal is to turn red if the build system tries to use the
  19     # source dist when it could have used the binary dist.
  20     #
  21     # Note that for this test to make sense, Tahoe-LAFS needs to be asking
  22     # for a version of fakedependency which can be satisfied by 1.0.0.
  23     # The --fakedependency option to setup.py arranges that.
  24
  25     fake_distdir = 'tahoe-deps'
  26     fake_distname = "fakedependency"
  27     fake_sdistversion = "1.0.0"
  28     fake_bdistversion = "0.9.9"
  29
  30     bdist_init = "raise Exception('Aha I caught you trying to import me. I am a fakedependency 0.9.9 package and you should not be satisfied with something < 1.0.0.')"
  31     sdist_setup = "import distutils.core\ndistutils.core.setup(name='fakedependency', version='1.0.0', packages=['fakedependency'])"
  32     sdist_init = "__version__ = '%s'" % (fake_sdistversion,)
  33
  34     testsuite = "buildtest.test_build_with_fake_dist"
  35
  36     dist_dirname = os.path.join(os.getcwd(), fake_distdir)
  37
  38     try:
  39         os.makedirs(dist_dirname)
  40     except OSError:
  41         # probably already exists
  42         pass
  43
  44     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()))
  45     try:
  46         bdist_egg = zipfile.ZipFile(bdist_egg_name, 'w')
  47         bdist_egg.writestr('fakedependency/__init__.py', bdist_init)
  48         bdist_egg.close()
  49
  50         sdist_name = os.path.join(dist_dirname, '%s-%s.tar' % (fake_distname, fake_sdistversion))
  51         sdist = tarfile.open(sdist_name, 'w:gz')
  52         sdist.errorlevel = 2
  53         tarinfo = tarfile.TarInfo('setup.py')
  54         tarinfo.errorlevel = 2
  55         tarinfo.size = len(sdist_setup)
  56         sdist.addfile(tarinfo, StringIO.StringIO(sdist_setup))
  57         tarinfo = tarfile.TarInfo('fakedependency/__init__.py')
  58         tarinfo.errorlevel = 2
  59         tarinfo.size = len(sdist_init)
  60         sdist.addfile(tarinfo, StringIO.StringIO(sdist_init))
  61         sdist.close()
  62
  63         sys.exit(subprocess.call([sys.executable, "setup.py", "--fakedependency", "-v", "test", "-s", testsuite],
  64                                  env=os.environ))
  65     finally:
  66         os.remove(bdist_egg_name)
  67         os.remove(sdist_name)
  68         cleanup()
  69
  70 def cleanup():
  71     egg_info = os.path.join('src', 'allmydata_tahoe.egg-info')
  72     bin_tahoe = os.path.join('bin', 'tahoe')
  73     bin_tahoe_pyscript = os.path.join('bin', 'tahoe.pyscript')
  74
  75     if os.path.exists('build'):
  76         shutil.rmtree('build')
  77     if os.path.exists('support'):
  78         shutil.rmtree('support')
  79     if os.path.exists(egg_info):
  80         shutil.rmtree(egg_info)
  81     if os.path.exists(bin_tahoe):
  82         os.remove(bin_tahoe)
  83     if os.path.exists(bin_tahoe_pyscript):
  84         os.remove(bin_tahoe_pyscript)
  85
  86 if __name__ == '__main__':
  87     test()